vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-03-30 03:11:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: isolate zai probe target env in alias coverage

Browse Source
This commit is contained in:
Peter Steinberger
2026-03-28 02:00:18 +00:00
parent 8aace2b448
commit 269f461b2e
1 changed files with 59 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

95
src/commands/models/list.probe.targets.test.ts
View File

@@ -82,6 +82,27 @@ async function withClearedAnthropicEnv<T>(fn: () => Promise<T>): Promise<T> {
}
}
async function withClearedZaiEnv<T>(fn: () => Promise<T>): Promise<T> {
const previousZai = process.env.ZAI_API_KEY;
const previousLegacyZai = process.env.Z_AI_API_KEY;
delete process.env.ZAI_API_KEY;
delete process.env.Z_AI_API_KEY;
try {
return await fn();
} finally {
if (previousZai === undefined) {
delete process.env.ZAI_API_KEY;
} else {
process.env.ZAI_API_KEY = previousZai;
}
if (previousLegacyZai === undefined) {
delete process.env.Z_AI_API_KEY;
} else {
process.env.Z_AI_API_KEY = previousLegacyZai;
}
}
}
async function buildAnthropicPlanFromModelsJsonApiKey(apiKey: string) {
return await buildProbeTargets({
cfg: {
@@ -223,46 +244,48 @@ describe("buildProbeTargets reason codes", () => {
});
it("matches canonical providers against alias-valued catalog probe models", async () => {
mockStore = {
version: 1,
profiles: {},
order: {},
};
loadModelCatalogMock.mockResolvedValueOnce([
{ provider: "z.ai", id: "glm-4.7", name: "GLM-4.7" },
]);
await withClearedZaiEnv(async () => {
mockStore = {
version: 1,
profiles: {},
order: {},
};
loadModelCatalogMock.mockResolvedValueOnce([
{ provider: "z.ai", id: "glm-4.7", name: "GLM-4.7" },
]);
const plan = await buildProbeTargets({
cfg: {
models: {
providers: {
zai: {
baseUrl: "https://api.z.ai/v1",
api: "openai-responses",
apiKey: "sk-zai-test", // pragma: allowlist secret
models: [],
const plan = await buildProbeTargets({
cfg: {
models: {
providers: {
zai: {
baseUrl: "https://api.z.ai/v1",
api: "openai-responses",
apiKey: "sk-zai-test", // pragma: allowlist secret
models: [],
},
},
},
} as OpenClawConfig,
providers: ["zai"],
modelCandidates: [],
options: {
timeoutMs: 5_000,
concurrency: 1,
maxTokens: 16,
},
} as OpenClawConfig,
providers: ["zai"],
modelCandidates: [],
options: {
timeoutMs: 5_000,
concurrency: 1,
maxTokens: 16,
},
});
});
expect(plan.results).toEqual([]);
expect(plan.targets).toHaveLength(1);
expect(plan.targets[0]).toEqual(
expect.objectContaining({
provider: "zai",
model: { provider: "zai", model: "glm-4.7" },
source: "models.json",
label: "models.json",
}),
);
expect(plan.results).toEqual([]);
expect(plan.targets).toHaveLength(1);
expect(plan.targets[0]).toEqual(
expect.objectContaining({
provider: "zai",
model: { provider: "zai", model: "glm-4.7" },
source: "models.json",
label: "models.json",
}),
);
});
});
});