liuyu/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-06-06 05:51:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: clean model cache state env

Browse Source
This commit is contained in:
Shakker
2026-06-05 17:04:19 +01:00
parent 6ce71737e5
commit 9235c25d33
1 changed files with 34 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
src/agents/embedded-agent-runner/model.test.ts
View File

@@ -3,7 +3,8 @@ import fs from "node:fs";
import os from "node:os";
import path from "node:path";
import { MAX_TIMER_TIMEOUT_MS } from "@openclaw/normalization-core/number-coercion";
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
import { beforeEach, describe, expect, it, vi } from "vitest";
import { withEnvAsync } from "../../test-utils/env.js";
import { discoverAuthStorage, discoverModels } from "../agent-model-discovery.js";
import {
clearRuntimeAuthProfileStoreSnapshots,
@@ -188,10 +189,6 @@ beforeEach(() => {
resolveBundledStaticCatalogModelMock.mockReset();
});
afterEach(() => {
vi.unstubAllEnvs();
});
function createRuntimeHooks() {
// Runtime hooks emulate provider plugin model discovery, transport
// normalization, and OpenRouter capability loading without plugin imports.
@@ -412,39 +409,44 @@ describe("resolveModel", () => {
it("invalidates agent discovery stores when implicit main auth changes without config", async () => {
const rootDir = fs.mkdtempSync(path.join(os.tmpdir(), "openclaw-model-cache-state-"));
vi.stubEnv("OPENCLAW_STATE_DIR", rootDir);
const agentDir = path.join(rootDir, "agents", "worker", "agent");
const mainAgentDir = path.join(rootDir, "agents", "main", "agent");
fs.mkdirSync(agentDir, { recursive: true });
fs.mkdirSync(mainAgentDir, { recursive: true });
mockDiscoveredModel(discoverModels, {
provider: "openai",
modelId: "gpt-5.5",
templateModel: {
provider: "openai",
...makeModel("gpt-5.5"),
},
});
try {
await withEnvAsync({ OPENCLAW_STATE_DIR: rootDir }, async () => {
mockDiscoveredModel(discoverModels, {
provider: "openai",
modelId: "gpt-5.5",
templateModel: {
provider: "openai",
...makeModel("gpt-5.5"),
},
});
const first = await resolveModelAsync("openai", "gpt-5.5", agentDir, undefined, {
runtimeHooks: createRuntimeHooks(),
});
saveAuthProfileStore(
{
version: 1,
profiles: { "openai:default": { type: "api_key", provider: "openai", key: "one" } },
},
mainAgentDir,
{ filterExternalAuthProfiles: false, syncExternalCli: false },
);
const second = await resolveModelAsync("openai", "gpt-5.5", agentDir, undefined, {
runtimeHooks: createRuntimeHooks(),
});
const first = await resolveModelAsync("openai", "gpt-5.5", agentDir, undefined, {
runtimeHooks: createRuntimeHooks(),
});
saveAuthProfileStore(
{
version: 1,
profiles: { "openai:default": { type: "api_key", provider: "openai", key: "one" } },
},
mainAgentDir,
{ filterExternalAuthProfiles: false, syncExternalCli: false },
);
const second = await resolveModelAsync("openai", "gpt-5.5", agentDir, undefined, {
runtimeHooks: createRuntimeHooks(),
});
expectResolvedModel(first);
expectResolvedModel(second);
expect(discoverAuthStorage).toHaveBeenCalledTimes(2);
expect(discoverModels).toHaveBeenCalledTimes(2);
expectResolvedModel(first);
expectResolvedModel(second);
expect(discoverAuthStorage).toHaveBeenCalledTimes(2);
expect(discoverModels).toHaveBeenCalledTimes(2);
});
} finally {
fs.rmSync(rootDir, { recursive: true, force: true });
}
});
it("does not cache agent discovery stores while runtime auth snapshots are active", async () => {