107 lines
3.4 KiB
TypeScript
107 lines
3.4 KiB
TypeScript
import { describe, expect, it, vi } from "vitest";
|
|
import type { OpenClawConfig } from "../config/config.js";
|
|
import { resetLogger, setLoggerOverride } from "../logging/logger.js";
|
|
import { __setModelCatalogImportForTest, loadModelCatalog } from "./model-catalog.js";
|
|
import {
|
|
installModelCatalogTestHooks,
|
|
mockCatalogImportFailThenRecover,
|
|
type PiSdkModule,
|
|
} from "./model-catalog.test-harness.js";
|
|
|
|
describe("loadModelCatalog", () => {
|
|
installModelCatalogTestHooks();
|
|
|
|
it("retries after import failure without poisoning the cache", async () => {
|
|
setLoggerOverride({ level: "silent", consoleLevel: "warn" });
|
|
const warnSpy = vi.spyOn(console, "warn").mockImplementation(() => {});
|
|
try {
|
|
const getCallCount = mockCatalogImportFailThenRecover();
|
|
|
|
const cfg = {} as OpenClawConfig;
|
|
const first = await loadModelCatalog({ config: cfg });
|
|
expect(first).toEqual([]);
|
|
|
|
const second = await loadModelCatalog({ config: cfg });
|
|
expect(second).toEqual([{ id: "gpt-4.1", name: "GPT-4.1", provider: "openai" }]);
|
|
expect(getCallCount()).toBe(2);
|
|
expect(warnSpy).toHaveBeenCalledTimes(1);
|
|
} finally {
|
|
setLoggerOverride(null);
|
|
resetLogger();
|
|
}
|
|
});
|
|
|
|
it("returns partial results on discovery errors", async () => {
|
|
setLoggerOverride({ level: "silent", consoleLevel: "warn" });
|
|
const warnSpy = vi.spyOn(console, "warn").mockImplementation(() => {});
|
|
try {
|
|
__setModelCatalogImportForTest(
|
|
async () =>
|
|
({
|
|
AuthStorage: class {},
|
|
ModelRegistry: class {
|
|
getAll() {
|
|
return [
|
|
{ id: "gpt-4.1", name: "GPT-4.1", provider: "openai" },
|
|
{
|
|
get id() {
|
|
throw new Error("boom");
|
|
},
|
|
provider: "openai",
|
|
name: "bad",
|
|
},
|
|
];
|
|
}
|
|
},
|
|
}) as unknown as PiSdkModule,
|
|
);
|
|
|
|
const result = await loadModelCatalog({ config: {} as OpenClawConfig });
|
|
expect(result).toEqual([{ id: "gpt-4.1", name: "GPT-4.1", provider: "openai" }]);
|
|
expect(warnSpy).toHaveBeenCalledTimes(1);
|
|
} finally {
|
|
setLoggerOverride(null);
|
|
resetLogger();
|
|
}
|
|
});
|
|
|
|
it("adds openai-codex/gpt-5.3-codex-spark when base gpt-5.3-codex exists", async () => {
|
|
__setModelCatalogImportForTest(
|
|
async () =>
|
|
({
|
|
AuthStorage: class {},
|
|
ModelRegistry: class {
|
|
getAll() {
|
|
return [
|
|
{
|
|
id: "gpt-5.3-codex",
|
|
provider: "openai-codex",
|
|
name: "GPT-5.3 Codex",
|
|
reasoning: true,
|
|
contextWindow: 200000,
|
|
input: ["text"],
|
|
},
|
|
{
|
|
id: "gpt-5.2-codex",
|
|
provider: "openai-codex",
|
|
name: "GPT-5.2 Codex",
|
|
},
|
|
];
|
|
}
|
|
},
|
|
}) as unknown as PiSdkModule,
|
|
);
|
|
|
|
const result = await loadModelCatalog({ config: {} as OpenClawConfig });
|
|
expect(result).toContainEqual(
|
|
expect.objectContaining({
|
|
provider: "openai-codex",
|
|
id: "gpt-5.3-codex-spark",
|
|
}),
|
|
);
|
|
const spark = result.find((entry) => entry.id === "gpt-5.3-codex-spark");
|
|
expect(spark?.name).toBe("gpt-5.3-codex-spark");
|
|
expect(spark?.reasoning).toBe(true);
|
|
});
|
|
});
|