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

chore(test): harden models status mock restoration

Browse Source
This commit is contained in:
Vignesh Natarajan
2026-02-22 00:53:13 -08:00
parent cf570d3b44
commit a4c107ee11
1 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/commands/models/list.status.e2e.test.ts
View File

@@ -118,10 +118,9 @@ vi.mock("../../config/config.js", async (importOriginal) => {
import { modelsStatusCommand } from "./list.status-command.js";
const defaultResolveAgentModelPrimaryImpl = mocks.resolveAgentModelPrimary.getMockImplementation();
const defaultResolveAgentModelFallbacksOverrideImpl =
mocks.resolveAgentModelFallbacksOverride.getMockImplementation();
const defaultResolveEnvApiKeyImpl = mocks.resolveEnvApiKey.getMockImplementation();
const defaultResolveEnvApiKeyImpl:
| ((provider: string) => { apiKey: string; source: string } | null)
| undefined = mocks.resolveEnvApiKey.getMockImplementation();
const runtime = {
log: vi.fn(),
@@ -161,14 +160,12 @@ async function withAgentScopeOverrides<T>(
if (originalPrimary) {
mocks.resolveAgentModelPrimary.mockImplementation(originalPrimary);
} else {
mocks.resolveAgentModelPrimary.mockImplementation(defaultResolveAgentModelPrimaryImpl);
mocks.resolveAgentModelPrimary.mockReturnValue(undefined);
}
if (originalFallbacks) {
mocks.resolveAgentModelFallbacksOverride.mockImplementation(originalFallbacks);
} else {
mocks.resolveAgentModelFallbacksOverride.mockImplementation(
defaultResolveAgentModelFallbacksOverrideImpl,
);
mocks.resolveAgentModelFallbacksOverride.mockReturnValue(undefined);
}
if (originalAgentDir) {
mocks.resolveAgentDir.mockImplementation(originalAgentDir);
@@ -276,8 +273,10 @@ describe("modelsStatusCommand auth overview", () => {
mocks.store.profiles = originalProfiles;
if (originalEnvImpl) {
mocks.resolveEnvApiKey.mockImplementation(originalEnvImpl);
} else {
} else if (defaultResolveEnvApiKeyImpl) {
mocks.resolveEnvApiKey.mockImplementation(defaultResolveEnvApiKeyImpl);
} else {
mocks.resolveEnvApiKey.mockImplementation(() => null);
}
}
});