mirror of
https://github.com/moltbot/moltbot.git
synced 2026-05-07 07:58:36 +00:00
test: fix models list e2e static catalog mock
This commit is contained in:
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user