test(agents): make grok api key test hermetic

This commit is contained in:
Peter Steinberger
2026-02-09 16:33:32 +00:00
parent b8a5f94f25
commit d25e96637c

View File

@@ -1,6 +1,30 @@
import { describe, expect, it } from "vitest"; import { describe, expect, it } from "vitest";
import { __testing } from "./web-search.js"; import { __testing } from "./web-search.js";
function withEnv<T>(env: Record<string, string | undefined>, fn: () => T): T {
const prev: Record<string, string | undefined> = {};
for (const [key, value] of Object.entries(env)) {
prev[key] = process.env[key];
if (value === undefined) {
// Make tests hermetic even on machines with real keys set.
delete process.env[key];
} else {
process.env[key] = value;
}
}
try {
return fn();
} finally {
for (const [key, value] of Object.entries(prev)) {
if (value === undefined) {
delete process.env[key];
} else {
process.env[key] = value;
}
}
}
}
const { const {
inferPerplexityBaseUrlFromApiKey, inferPerplexityBaseUrlFromApiKey,
resolvePerplexityBaseUrl, resolvePerplexityBaseUrl,
@@ -110,18 +134,10 @@ describe("web_search grok config resolution", () => {
}); });
it("returns undefined when no apiKey is available", () => { it("returns undefined when no apiKey is available", () => {
const previous = process.env.XAI_API_KEY; withEnv({ XAI_API_KEY: undefined }, () => {
try {
delete process.env.XAI_API_KEY;
expect(resolveGrokApiKey({})).toBeUndefined(); expect(resolveGrokApiKey({})).toBeUndefined();
expect(resolveGrokApiKey(undefined)).toBeUndefined(); expect(resolveGrokApiKey(undefined)).toBeUndefined();
} finally { });
if (previous === undefined) {
delete process.env.XAI_API_KEY;
} else {
process.env.XAI_API_KEY = previous;
}
}
}); });
it("uses default model when not specified", () => { it("uses default model when not specified", () => {