mirror of https://github.com/openclaw/openclaw.git
88 lines
2.4 KiB
TypeScript
88 lines
2.4 KiB
TypeScript
import { beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
|
|
import type { OpenClawConfig } from "../config/config.js";
|
|
|
|
const ensureOpenClawModelsJsonMock = vi.fn<
|
|
(config: unknown, agentDir: unknown) => Promise<{ agentDir: string; wrote: boolean }>
|
|
>(async () => ({ agentDir: "/tmp/agent", wrote: false }));
|
|
const resolveModelMock = vi.fn<
|
|
(
|
|
provider: unknown,
|
|
modelId: unknown,
|
|
agentDir: unknown,
|
|
cfg: unknown,
|
|
options?: unknown,
|
|
) => { model: { id: string; provider: string; api: string } }
|
|
>(() => ({
|
|
model: {
|
|
id: "gpt-5.4",
|
|
provider: "openai-codex",
|
|
api: "openai-codex-responses",
|
|
},
|
|
}));
|
|
|
|
vi.mock("../agents/agent-paths.js", () => ({
|
|
resolveOpenClawAgentDir: () => "/tmp/agent",
|
|
}));
|
|
|
|
vi.mock("../agents/models-config.js", () => ({
|
|
ensureOpenClawModelsJson: (config: unknown, agentDir: unknown) =>
|
|
ensureOpenClawModelsJsonMock(config, agentDir),
|
|
}));
|
|
|
|
vi.mock("../agents/pi-embedded-runner/model.js", () => ({
|
|
resolveModel: (
|
|
provider: unknown,
|
|
modelId: unknown,
|
|
agentDir: unknown,
|
|
cfg: unknown,
|
|
options?: unknown,
|
|
) => resolveModelMock(provider, modelId, agentDir, cfg, options),
|
|
}));
|
|
|
|
let prewarmConfiguredPrimaryModel: typeof import("./server-startup.js").__testing.prewarmConfiguredPrimaryModel;
|
|
|
|
describe("gateway startup primary model warmup", () => {
|
|
beforeAll(async () => {
|
|
({
|
|
__testing: { prewarmConfiguredPrimaryModel },
|
|
} = await import("./server-startup.js"));
|
|
});
|
|
|
|
beforeEach(() => {
|
|
ensureOpenClawModelsJsonMock.mockClear();
|
|
resolveModelMock.mockClear();
|
|
});
|
|
|
|
it("prewarms an explicit configured primary model", async () => {
|
|
const cfg = {
|
|
agents: {
|
|
defaults: {
|
|
model: {
|
|
primary: "openai-codex/gpt-5.4",
|
|
},
|
|
},
|
|
},
|
|
} as OpenClawConfig;
|
|
|
|
await prewarmConfiguredPrimaryModel({
|
|
cfg,
|
|
log: { warn: vi.fn() },
|
|
});
|
|
|
|
expect(ensureOpenClawModelsJsonMock).toHaveBeenCalledWith(cfg, "/tmp/agent");
|
|
expect(resolveModelMock).toHaveBeenCalledWith("openai-codex", "gpt-5.4", "/tmp/agent", cfg, {
|
|
skipProviderRuntimeHooks: true,
|
|
});
|
|
});
|
|
|
|
it("skips warmup when no explicit primary model is configured", async () => {
|
|
await prewarmConfiguredPrimaryModel({
|
|
cfg: {} as OpenClawConfig,
|
|
log: { warn: vi.fn() },
|
|
});
|
|
|
|
expect(ensureOpenClawModelsJsonMock).not.toHaveBeenCalled();
|
|
expect(resolveModelMock).not.toHaveBeenCalled();
|
|
});
|
|
});
|