github/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-10 04:12:44 +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"; import { modelsStatusCommand } from "./list.status-command.js";
const defaultResolveAgentModelPrimaryImpl = mocks.resolveAgentModelPrimary.getMockImplementation(); const defaultResolveEnvApiKeyImpl:
const defaultResolveAgentModelFallbacksOverrideImpl = | ((provider: string) => { apiKey: string; source: string } | null)
mocks.resolveAgentModelFallbacksOverride.getMockImplementation(); | undefined = mocks.resolveEnvApiKey.getMockImplementation();
const defaultResolveEnvApiKeyImpl = mocks.resolveEnvApiKey.getMockImplementation();
const runtime = { const runtime = {
log: vi.fn(), log: vi.fn(),
@@ -161,14 +160,12 @@ async function withAgentScopeOverrides<T>(
if (originalPrimary) { if (originalPrimary) {
mocks.resolveAgentModelPrimary.mockImplementation(originalPrimary); mocks.resolveAgentModelPrimary.mockImplementation(originalPrimary);
} else { } else {
mocks.resolveAgentModelPrimary.mockImplementation(defaultResolveAgentModelPrimaryImpl); mocks.resolveAgentModelPrimary.mockReturnValue(undefined);
} }
if (originalFallbacks) { if (originalFallbacks) {
mocks.resolveAgentModelFallbacksOverride.mockImplementation(originalFallbacks); mocks.resolveAgentModelFallbacksOverride.mockImplementation(originalFallbacks);
} else { } else {
mocks.resolveAgentModelFallbacksOverride.mockImplementation( mocks.resolveAgentModelFallbacksOverride.mockReturnValue(undefined);
defaultResolveAgentModelFallbacksOverrideImpl,
);
} }
if (originalAgentDir) { if (originalAgentDir) {
mocks.resolveAgentDir.mockImplementation(originalAgentDir); mocks.resolveAgentDir.mockImplementation(originalAgentDir);
@@ -276,8 +273,10 @@ describe("modelsStatusCommand auth overview", () => {
mocks.store.profiles = originalProfiles; mocks.store.profiles = originalProfiles;
if (originalEnvImpl) { if (originalEnvImpl) {
mocks.resolveEnvApiKey.mockImplementation(originalEnvImpl); mocks.resolveEnvApiKey.mockImplementation(originalEnvImpl);
} else { } else if (defaultResolveEnvApiKeyImpl) {
mocks.resolveEnvApiKey.mockImplementation(defaultResolveEnvApiKeyImpl); mocks.resolveEnvApiKey.mockImplementation(defaultResolveEnvApiKeyImpl);
} else {
mocks.resolveEnvApiKey.mockImplementation(() => null);
} }
} }
}); });