diff --git a/src/agents/models-config.providers.minimax.test.ts b/src/agents/models-config.providers.minimax.test.ts index 7832e483bc..94b7994a6c 100644 --- a/src/agents/models-config.providers.minimax.test.ts +++ b/src/agents/models-config.providers.minimax.test.ts @@ -2,12 +2,13 @@ import { mkdtempSync } from "node:fs"; import { tmpdir } from "node:os"; import { join } from "node:path"; import { describe, expect, it } from "vitest"; +import { captureEnv } from "../test-utils/env.js"; import { resolveImplicitProviders } from "./models-config.providers.js"; describe("MiniMax implicit provider (#15275)", () => { it("should use anthropic-messages API for API-key provider", async () => { const agentDir = mkdtempSync(join(tmpdir(), "openclaw-test-")); - const previous = process.env.MINIMAX_API_KEY; + const envSnapshot = captureEnv(["MINIMAX_API_KEY"]); process.env.MINIMAX_API_KEY = "test-key"; try { @@ -16,11 +17,7 @@ describe("MiniMax implicit provider (#15275)", () => { expect(providers?.minimax?.api).toBe("anthropic-messages"); expect(providers?.minimax?.baseUrl).toBe("https://api.minimax.io/anthropic"); } finally { - if (previous === undefined) { - delete process.env.MINIMAX_API_KEY; - } else { - process.env.MINIMAX_API_KEY = previous; - } + envSnapshot.restore(); } }); }); diff --git a/src/agents/models-config.providers.nvidia.test.ts b/src/agents/models-config.providers.nvidia.test.ts index 42a46ebe4a..a9920a3cba 100644 --- a/src/agents/models-config.providers.nvidia.test.ts +++ b/src/agents/models-config.providers.nvidia.test.ts @@ -2,13 +2,14 @@ import { mkdtempSync } from "node:fs"; import { tmpdir } from "node:os"; import { join } from "node:path"; import { describe, expect, it } from "vitest"; +import { captureEnv } from "../test-utils/env.js"; import { resolveApiKeyForProvider } from "./model-auth.js"; import { buildNvidiaProvider, resolveImplicitProviders } from "./models-config.providers.js"; describe("NVIDIA provider", () => { it("should include nvidia when NVIDIA_API_KEY is configured", async () => { const agentDir = mkdtempSync(join(tmpdir(), "openclaw-test-")); - const previous = process.env.NVIDIA_API_KEY; + const envSnapshot = captureEnv(["NVIDIA_API_KEY"]); process.env.NVIDIA_API_KEY = "test-key"; try { @@ -16,17 +17,13 @@ describe("NVIDIA provider", () => { expect(providers?.nvidia).toBeDefined(); expect(providers?.nvidia?.models?.length).toBeGreaterThan(0); } finally { - if (previous === undefined) { - delete process.env.NVIDIA_API_KEY; - } else { - process.env.NVIDIA_API_KEY = previous; - } + envSnapshot.restore(); } }); it("resolves the nvidia api key value from env", async () => { const agentDir = mkdtempSync(join(tmpdir(), "openclaw-test-")); - const previous = process.env.NVIDIA_API_KEY; + const envSnapshot = captureEnv(["NVIDIA_API_KEY"]); process.env.NVIDIA_API_KEY = "nvidia-test-api-key"; try { @@ -39,11 +36,7 @@ describe("NVIDIA provider", () => { expect(auth.mode).toBe("api-key"); expect(auth.source).toContain("NVIDIA_API_KEY"); } finally { - if (previous === undefined) { - delete process.env.NVIDIA_API_KEY; - } else { - process.env.NVIDIA_API_KEY = previous; - } + envSnapshot.restore(); } });