test: fix models list e2e static catalog mock

This commit is contained in:
Peter Steinberger
2026-04-24 10:21:45 +01:00
parent f672535e0c
commit 3385d10ee5

View File

@@ -21,6 +21,7 @@ const loadModelCatalog = vi.fn(async () => []);
const loadProviderCatalogModelsForList = vi.fn<() => Promise<Array<Record<string, unknown>>>>(
async () => [],
);
const hasProviderStaticCatalogForFilter = vi.fn().mockResolvedValue(false);
const shouldSuppressBuiltInModel = vi.fn().mockReturnValue(false);
const modelRegistryState = {
models: [] as Array<Record<string, unknown>>,
@@ -75,6 +76,7 @@ vi.mock("./models/list.runtime.js", () => {
resolveEnvApiKey,
resolveAwsSdkEnvVarName,
hasUsableCustomProviderApiKey,
hasProviderStaticCatalogForFilter,
loadModelCatalog,
loadProviderCatalogModelsForList,
discoverAuthStorage: () => ({}) as unknown,
@@ -141,6 +143,8 @@ beforeEach(() => {
loadModelCatalog.mockResolvedValue([]);
loadProviderCatalogModelsForList.mockReset();
loadProviderCatalogModelsForList.mockResolvedValue([]);
hasProviderStaticCatalogForFilter.mockReset();
hasProviderStaticCatalogForFilter.mockResolvedValue(false);
shouldSuppressBuiltInModel.mockReset();
shouldSuppressBuiltInModel.mockReturnValue(false);
readConfigFileSnapshotForWrite.mockClear();
@@ -348,13 +352,14 @@ describe("models list/status", () => {
it("models list all includes unauthenticated provider catalog rows", async () => {
setDefaultZaiRegistry({ available: false });
hasProviderStaticCatalogForFilter.mockResolvedValueOnce(true);
loadProviderCatalogModelsForList.mockResolvedValueOnce([MOONSHOT_MODEL]);
const runtime = makeRuntime();
await modelsListCommand({ all: true, provider: "moonshot", json: true }, runtime);
const payload = parseJsonLog(runtime);
expect(loadModelCatalog).toHaveBeenCalledTimes(1);
expect(loadModelCatalog).not.toHaveBeenCalled();
expect(payload.models).toEqual([
expect.objectContaining({
key: "moonshot/kimi-k2.6",