chore: add test for token refresh

Author: kmruiz

tests/integration/common/connectionManager.oidc.test.ts

@@ -7,21 +7,21 @@ import {
     getServerVersion,
     MongoDBIntegrationTestCase,
 } from "../tools/mongodb/mongodbHelpers.js";
-import { defaultTestConfig, responseAsText } from "../helpers.js";
+import { defaultTestConfig, responseAsText, timeout } from "../helpers.js";
 import {
     ConnectionManager,
     ConnectionStateConnected,
     ConnectionStateConnecting,
 } from "../../../src/common/connectionManager.js";
 import { setupDriverConfig, UserConfig } from "../../../src/common/config.js";
-import fs from "fs";
 import path from "path";
 import type { OIDCMockProviderConfig } from "@mongodb-js/oidc-mock-provider";
 import { OIDCMockProvider } from "@mongodb-js/oidc-mock-provider";
 
 const DEFAULT_TIMEOUT = 10000;
 
-describe("ConnectionManager OIDC Tests", async () => {
+// OIDC is only supported on Linux servers
+describe.skipIf(process.platform !== "linux")("ConnectionManager OIDC Tests", async () => {
     function setParameter(param: string): ["--setParameter", string] {
         return ["--setParameter", param];
     }
@@ -55,7 +55,7 @@ describe("ConnectionManager OIDC Tests", async () => {
         getTokenPayload = ((metadata) => {
             tokenFetches++;
             return {
-                expires_in: 3600,
+                expires_in: 1,
                 payload: {
                     // Define the user information stored inside the access tokens
                     groups: [`${metadata.client_id}-group`],
@@ -100,6 +100,8 @@ describe("ConnectionManager OIDC Tests", async () => {
             ...defaultTestConfig,
             oidcRedirectURi: "http://localhost:0/",
             authenticationMechanism: "MONGODB-OIDC",
+            maxIdleTimeMS: "1",
+            minPoolSize: "0",
             username: "testuser",
             browser: fetchBrowserFixture,
             ...args?.additionalConfig,
@@ -111,7 +113,6 @@ describe("ConnectionManager OIDC Tests", async () => {
                 function oidcIt(name: string, cb: Parameters<OidcIt>[1]): void {
                     /* eslint-disable @vitest/expect-expect */
                     it(name, { timeout: DEFAULT_TIMEOUT }, async (context) => {
-                        context.skip(process.platform !== "linux", "OIDC is only supported on Linux");
                         context.skip(
                             await isCommunityServer(integration),
                             "OIDC is not supported in MongoDB Community"
@@ -193,6 +194,18 @@ describe("ConnectionManager OIDC Tests", async () => {
                 const databases = listDbResult.databases as unknown[];
                 expect(databases.length).toBeGreaterThan(0);
             });
+
+            it("can refresh a token once expired", async ({ signal }, integration) => {
+                const state = await ConnectionManager.waitUntil<ConnectionStateConnected>(
+                    "connected",
+                    integration.mcpServer().session.connectionManager,
+                    signal
+                );
+
+                await timeout(2000);
+                await state.serviceProvider.listDatabases("admin");
+                expect(tokenFetches).toBeGreaterThan(1);
+            });
         });
     }