diff --git a/src/agents/models-config.falls-back-default-baseurl-token-exchange-fails.e2e.test.ts b/src/agents/models-config.falls-back-default-baseurl-token-exchange-fails.e2e.test.ts index ee0e4580d..8458f492f 100644 --- a/src/agents/models-config.falls-back-default-baseurl-token-exchange-fails.e2e.test.ts +++ b/src/agents/models-config.falls-back-default-baseurl-token-exchange-fails.e2e.test.ts @@ -2,6 +2,7 @@ import fs from "node:fs/promises"; import path from "node:path"; import { describe, expect, it, vi } from "vitest"; import { DEFAULT_COPILOT_API_BASE_URL } from "../providers/github-copilot-token.js"; +import { captureEnv } from "../test-utils/env.js"; import { installModelsConfigTestHooks, withModelsTempHome as withTempHome, @@ -13,7 +14,7 @@ installModelsConfigTestHooks({ restoreFetch: true }); describe("models-config", () => { it("falls back to default baseUrl when token exchange fails", async () => { await withTempHome(async () => { - const previous = process.env.COPILOT_GITHUB_TOKEN; + const envSnapshot = captureEnv(["COPILOT_GITHUB_TOKEN"]); process.env.COPILOT_GITHUB_TOKEN = "gh-token"; const fetchMock = vi.fn().mockResolvedValue({ ok: false, @@ -33,20 +34,14 @@ describe("models-config", () => { expect(parsed.providers["github-copilot"]?.baseUrl).toBe(DEFAULT_COPILOT_API_BASE_URL); } finally { - if (previous === undefined) { - delete process.env.COPILOT_GITHUB_TOKEN; - } else { - process.env.COPILOT_GITHUB_TOKEN = previous; - } + envSnapshot.restore(); } }); }); it("uses agentDir override auth profiles for copilot injection", async () => { await withTempHome(async (home) => { - const previous = process.env.COPILOT_GITHUB_TOKEN; - const previousGh = process.env.GH_TOKEN; - const previousGithub = process.env.GITHUB_TOKEN; + const envSnapshot = captureEnv(["COPILOT_GITHUB_TOKEN", "GH_TOKEN", "GITHUB_TOKEN"]); delete process.env.COPILOT_GITHUB_TOKEN; delete process.env.GH_TOKEN; delete process.env.GITHUB_TOKEN; @@ -91,21 +86,7 @@ describe("models-config", () => { expect(parsed.providers["github-copilot"]?.baseUrl).toBe("https://api.copilot.example"); } finally { - if (previous === undefined) { - delete process.env.COPILOT_GITHUB_TOKEN; - } else { - process.env.COPILOT_GITHUB_TOKEN = previous; - } - if (previousGh === undefined) { - delete process.env.GH_TOKEN; - } else { - process.env.GH_TOKEN = previousGh; - } - if (previousGithub === undefined) { - delete process.env.GITHUB_TOKEN; - } else { - process.env.GITHUB_TOKEN = previousGithub; - } + envSnapshot.restore(); } }); });