diff --git a/src/commands/models.list.e2e.test.ts b/src/commands/models.list.e2e.test.ts index 71bb10fab8c..07ccd001a52 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>, @@ -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",