From d3fe591853d092e743c85756a233ff467bbc97fe Mon Sep 17 00:00:00 2001 From: Shakker Date: Fri, 24 Apr 2026 05:15:23 +0100 Subject: [PATCH] test: update models list static fast path expectation --- src/commands/models.list.e2e.test.ts | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/src/commands/models.list.e2e.test.ts b/src/commands/models.list.e2e.test.ts index 441c0675a79..5fbbb88f12b 100644 --- a/src/commands/models.list.e2e.test.ts +++ b/src/commands/models.list.e2e.test.ts @@ -21,6 +21,7 @@ const loadModelCatalog = vi.fn(async () => []); const loadProviderCatalogModelsForList = vi.fn<() => Promise>>>( async () => [], ); +const hasProviderStaticCatalogForFilter = vi.fn().mockResolvedValue(false); const shouldSuppressBuiltInModel = vi.fn().mockReturnValue(false); const modelRegistryState = { models: [] as Array>, @@ -103,6 +104,14 @@ vi.mock("./models/list.runtime.js", () => { }; }); +vi.mock("./models/list.provider-catalog.js", async (importOriginal) => { + const actual = await importOriginal(); + return { + ...actual, + hasProviderStaticCatalogForFilter, + }; +}); + vi.mock("../agents/model-suppression.js", () => ({ shouldSuppressBuiltInModel, })); @@ -152,6 +161,8 @@ beforeEach(() => { loadModelCatalog.mockResolvedValue([]); loadProviderCatalogModelsForList.mockReset(); loadProviderCatalogModelsForList.mockResolvedValue([]); + hasProviderStaticCatalogForFilter.mockReset(); + hasProviderStaticCatalogForFilter.mockResolvedValue(false); shouldSuppressBuiltInModel.mockReset(); shouldSuppressBuiltInModel.mockReturnValue(false); readConfigFileSnapshotForWrite.mockClear(); @@ -359,13 +370,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",