chore: [broken] Add a test to verify that the http-proxy is called.

Right now the code is broken because there is some issue with the
promise returned by the agent. I'm still investigating it.

Author: kmruiz

package-lock.json

@@ -16,7 +16,7 @@
   },
   "type": "module",
   "scripts": {
-    "start": "node dist/index.js --transport http --loggers stderr mcp",
+    "start": "node dist/index.js --transport stdio --loggers stderr mcp",
     "prepare": "npm run build",
     "build:clean": "rm -rf dist",
     "build:compile": "tsc --project tsconfig.build.json",
@@ -43,6 +43,7 @@
     "@modelcontextprotocol/inspector": "^0.16.0",
     "@redocly/cli": "^1.34.4",
     "@types/express": "^5.0.1",
+    "@types/http-proxy": "^1.17.16",
     "@types/node": "^24.0.12",
     "@types/proper-lockfile": "^4.1.4",
     "@types/simple-oauth2": "^5.0.7",
@@ -53,6 +54,7 @@
     "eslint-config-prettier": "^10.1.5",
     "eslint-plugin-prettier": "^5.5.1",
     "globals": "^16.3.0",
+    "http-proxy": "^1.18.1",
     "mongodb-runner": "^5.9.2",
     "ollama-ai-provider": "^1.2.0",
     "openapi-types": "^12.1.3",
@@ -63,14 +65,15 @@
     "tsx": "^4.20.3",
     "typescript": "^5.8.3",
     "typescript-eslint": "^8.36.0",
-    "vitest": "^3.2.4",
     "uuid": "^11.1.0",
+    "vitest": "^3.2.4",
     "yaml": "^2.8.0"
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.15.0",
     "@mongodb-js/device-id": "^0.3.1",
     "@mongodb-js/devtools-connect": "^3.9.2",
+    "@mongodb-js/devtools-proxy-support": "^0.5.1",
     "@mongosh/service-provider-node-driver": "^3.10.2",
     "@vitest/eslint-plugin": "^1.3.4",
     "bson": "^6.10.4",

package.json

@@ -6,6 +6,8 @@ import { paths, operations } from "./openapi.js";
 import { CommonProperties, TelemetryEvent } from "../../telemetry/types.js";
 import { packageInfo } from "../packageInfo.js";
 import logger, { LogId } from "../logger.js";
+import { createFetch, useOrCreateAgent } from "@mongodb-js/devtools-proxy-support";
+import HTTPS from "https";
 
 const ATLAS_API_VERSION = "2025-03-12";
 
@@ -29,11 +31,25 @@ export class ApiClient {
             clientSecret: string;
         };
     };
+
+    private static customFetch = createFetch({
+        useEnvironmentVariableProxies: true,
+    }) as unknown as typeof fetch;
+
+    private static customAgent = useOrCreateAgent({
+        useEnvironmentVariableProxies: true,
+    });
+
     private client: Client<paths>;
     private oauth2Client?: ClientCredentials;
     private accessToken?: AccessToken;
 
+    private ensureAgentIsInitialized = async () => {
+        await ApiClient.customAgent?.initialize?.();
+    };
+
     private getAccessToken = async () => {
+        //        await this.ensureAgentIsInitialized();
         if (this.oauth2Client && (!this.accessToken || this.accessToken.expired())) {
             this.accessToken = await this.oauth2Client.getToken({});
         }
@@ -72,6 +88,7 @@ export class ApiClient {
                 "User-Agent": this.options.userAgent,
                 Accept: `application/vnd.atlas.${ATLAS_API_VERSION}+json`,
             },
+            fetch: ApiClient.customFetch,
         });
         if (this.options.credentials?.clientId && this.options.credentials?.clientSecret) {
             this.oauth2Client = new ClientCredentials({
@@ -88,6 +105,7 @@ export class ApiClient {
                     headers: {
                         "User-Agent": this.options.userAgent,
                     },
+                    agent: ApiClient.customAgent,
                 },
             });
             this.client.use(this.authMiddleware);

src/common/atlas/apiClient.ts

@@ -1,6 +1,7 @@
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
 import { ApiClient } from "../../../src/common/atlas/apiClient.js";
 import { CommonProperties, TelemetryEvent, TelemetryResult } from "../../../src/telemetry/types.js";
+import httpProxy from "http-proxy";
 
 describe("ApiClient", () => {
     let apiClient: ApiClient;
@@ -94,6 +95,44 @@ describe("ApiClient", () => {
         });
     });
 
+    describe("oauth authentication proxy", () => {
+        let apiClient: ApiClient;
+        let proxyServer: httpProxy;
+        let requests: unknown[];
+
+        beforeEach(() => {
+            process.env.HTTP_PROXY = "localhost:8888";
+            apiClient = new ApiClient({
+                baseUrl: "https://httpbin.org",
+                credentials: {
+                    clientId: "test-client-id",
+                    clientSecret: "test-client-secret",
+                },
+                userAgent: "test-user-agent",
+            });
+
+            proxyServer = httpProxy.createProxyServer({ target: "https://api.test.com" });
+            proxyServer.on("proxyReq", (proxyReq, req, res, options) => {
+                requests.push(req);
+            });
+
+            requests = [];
+        });
+
+        afterEach(async () => {
+            delete process.env.HTTP_PROXY;
+
+            await proxyServer.close();
+            //await apiClient.close();
+        });
+
+        it("should send the oauth request through a proxy if configured", async () => {
+            await apiClient.validateAccessToken();
+            console.log(requests);
+            expect(true).toBeFalsy();
+        });
+    });
+
     describe("sendEvents", () => {
         it("should send events to authenticated endpoint when token is available and valid", async () => {
             const mockFetch = vi.spyOn(global, "fetch");