chore: Add integration tests for proxy support in the Atlas API

Uses a variant devtools-shared/devtools-proxy-support http proxy
server for testing.

Author: kmruiz

package-lock.json

@@ -50,11 +50,11 @@
     "@types/yargs-parser": "^21.0.3",
     "@vitest/coverage-v8": "^3.2.4",
     "ai": "^4.3.17",
+    "duplexpair": "^1.0.2",
     "eslint": "^9.30.1",
     "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",

package.json

@@ -21,7 +21,7 @@ export interface ApiClientOptions {
     userAgent?: string;
 }
 
-interface AccessToken {
+export interface AccessToken {
     access_token: string;
     expires_at?: number;
 }

src/common/atlas/apiClient.ts

@@ -0,0 +1,77 @@
+import { afterEach, beforeEach, describe, expect, it } from "vitest";
+import type { AccessToken } from "../../../src/common/atlas/apiClient.js";
+import { ApiClient } from "../../../src/common/atlas/apiClient.js";
+import { HTTPServerProxyTestSetup } from "../fixtures/httpsServerProxyTest.js";
+
+describe("ApiClient integration test", () => {
+    describe("oauth authentication proxy", () => {
+        let apiClient: ApiClient;
+        let proxyTestSetup: HTTPServerProxyTestSetup;
+
+        beforeEach(async () => {
+            process.env.NODE_TLS_REJECT_UNAUTHORIZED = "0";
+            proxyTestSetup = new HTTPServerProxyTestSetup();
+            await proxyTestSetup.listen();
+
+            process.env.HTTP_PROXY = `https://localhost:${proxyTestSetup.httpsProxyPort}/`;
+            apiClient = new ApiClient({
+                baseUrl: `https://localhost:${proxyTestSetup.httpsServerPort}/`,
+                credentials: {
+                    clientId: "test-client-id",
+                    clientSecret: "test-client-secret",
+                },
+                userAgent: "test-user-agent",
+            });
+        });
+
+        function withToken(accessToken: string, expired: boolean) {
+            const apiClientMut = apiClient as unknown as { accessToken: AccessToken };
+            const expireAt = expired ? Date.now() - 100000 : Date.now() + 10000;
+
+            apiClientMut.accessToken = {
+                access_token: accessToken,
+                expires_at: expireAt,
+            };
+        }
+
+        afterEach(async () => {
+            delete process.env.NODE_TLS_REJECT_UNAUTHORIZED;
+            delete process.env.HTTP_PROXY;
+
+            await apiClient.close();
+            await proxyTestSetup.teardown();
+        });
+
+        it("should send the oauth request through a proxy if configured", async () => {
+            await apiClient.validateAccessToken();
+            expect(proxyTestSetup.getRequestedUrls()).toEqual([
+                `http://localhost:${proxyTestSetup.httpsServerPort}/api/oauth/token`,
+            ]);
+        });
+
+        it("should send the oauth revoke request through a proxy if configured", async () => {
+            withToken("my non expired token", false);
+            await apiClient.close();
+            expect(proxyTestSetup.getRequestedUrls()).toEqual([
+                `http://localhost:${proxyTestSetup.httpsServerPort}/api/oauth/revoke`,
+            ]);
+        });
+
+        it("should make an atlas call when the token is not expired", async () => {
+            withToken("my not expired", false);
+            await apiClient.listOrganizations();
+            expect(proxyTestSetup.getRequestedUrls()).toEqual([
+                `http://localhost:${proxyTestSetup.httpsServerPort}/api/atlas/v2/orgs`,
+            ]);
+        });
+
+        it("should request a new token and an atlas call when the token is expired", async () => {
+            withToken("my expired", true);
+            await apiClient.listOrganizations();
+            expect(proxyTestSetup.getRequestedUrls()).toEqual([
+                `http://localhost:${proxyTestSetup.httpsServerPort}/api/oauth/token`,
+                `http://localhost:${proxyTestSetup.httpsServerPort}/api/atlas/v2/orgs`,
+            ]);
+        });
+    });
+});