github/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-08 07:11:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Tests: update pi SDK mocks

Browse Source
This commit is contained in:
Mario Zechner
2026-01-31 05:23:37 +01:00
parent 310eed825e
commit 9b1a6b30d9
3 changed files with 58 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
src/commands/models.list.test.ts
View File

@@ -13,8 +13,22 @@ const resolveProfileUnusableUntilForDisplay = vi.fn().mockReturnValue(null);
const resolveEnvApiKey = vi.fn().mockReturnValue(undefined);
const resolveAwsSdkEnvVarName = vi.fn().mockReturnValue(undefined);
const getCustomProviderApiKey = vi.fn().mockReturnValue(undefined);
const discoverAuthStorage = vi.fn().mockReturnValue({});
const discoverModels = vi.fn();
const modelRegistryState = {
models: [] as Array<Record<string, unknown>>,
available: [] as Array<Record<string, unknown>>,
};
class AuthStorage {}
class ModelRegistry {
getAll() {
return modelRegistryState.models;
}
getAvailable() {
return modelRegistryState.available;
}
}
vi.mock("../config/config.js", () => ({
CONFIG_PATH: "/tmp/openclaw.json",
@@ -45,8 +59,8 @@ vi.mock("../agents/model-auth.js", () => ({
}));
vi.mock("@mariozechner/pi-coding-agent", () => ({
discoverAuthStorage,
discoverModels,
AuthStorage,
ModelRegistry,
}));
function makeRuntime() {
@@ -99,10 +113,8 @@ describe("models list/status", () => {
contextWindow: 128000,
};
discoverModels.mockReturnValue({
getAll: () => [model],
getAvailable: () => [model],
});
modelRegistryState.models = [model];
modelRegistryState.available = [model];
const { modelsListCommand } = await import("./models/list.js");
await modelsListCommand({ json: true }, runtime);
@@ -127,10 +139,8 @@ describe("models list/status", () => {
contextWindow: 128000,
};
discoverModels.mockReturnValue({
getAll: () => [model],
getAvailable: () => [model],
});
modelRegistryState.models = [model];
modelRegistryState.available = [model];
const { modelsListCommand } = await import("./models/list.js");
await modelsListCommand({ plain: true }, runtime);
@@ -164,10 +174,8 @@ describe("models list/status", () => {
},
];
discoverModels.mockReturnValue({
getAll: () => models,
getAvailable: () => models,
});
modelRegistryState.models = models;
modelRegistryState.available = models;
const { modelsListCommand } = await import("./models/list.js");
await modelsListCommand({ all: true, provider: "z.ai", json: true }, runtime);
@@ -203,10 +211,8 @@ describe("models list/status", () => {
},
];
discoverModels.mockReturnValue({
getAll: () => models,
getAvailable: () => models,
});
modelRegistryState.models = models;
modelRegistryState.available = models;
const { modelsListCommand } = await import("./models/list.js");
await modelsListCommand({ all: true, provider: "Z.AI", json: true }, runtime);
@@ -242,10 +248,8 @@ describe("models list/status", () => {
},
];
discoverModels.mockReturnValue({
getAll: () => models,
getAvailable: () => models,
});
modelRegistryState.models = models;
modelRegistryState.available = models;
const { modelsListCommand } = await import("./models/list.js");
await modelsListCommand({ all: true, provider: "z-ai", json: true }, runtime);
@@ -271,10 +275,8 @@ describe("models list/status", () => {
contextWindow: 128000,
};
discoverModels.mockReturnValue({
getAll: () => [model],
getAvailable: () => [],
});
modelRegistryState.models = [model];
modelRegistryState.available = [];
const { modelsListCommand } = await import("./models/list.js");
await modelsListCommand({ all: true, json: true }, runtime);