mirror of
https://github.com/openclaw/openclaw.git
synced 2026-05-08 06:21:26 +00:00
test(core): reduce mock reset overhead in targeted suites
This commit is contained in:
@@ -118,6 +118,11 @@ 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 runtime = {
|
||||
log: vi.fn(),
|
||||
error: vi.fn(),
|
||||
@@ -156,12 +161,14 @@ async function withAgentScopeOverrides<T>(
|
||||
if (originalPrimary) {
|
||||
mocks.resolveAgentModelPrimary.mockImplementation(originalPrimary);
|
||||
} else {
|
||||
mocks.resolveAgentModelPrimary.mockReset();
|
||||
mocks.resolveAgentModelPrimary.mockImplementation(defaultResolveAgentModelPrimaryImpl);
|
||||
}
|
||||
if (originalFallbacks) {
|
||||
mocks.resolveAgentModelFallbacksOverride.mockImplementation(originalFallbacks);
|
||||
} else {
|
||||
mocks.resolveAgentModelFallbacksOverride.mockReset();
|
||||
mocks.resolveAgentModelFallbacksOverride.mockImplementation(
|
||||
defaultResolveAgentModelFallbacksOverrideImpl,
|
||||
);
|
||||
}
|
||||
if (originalAgentDir) {
|
||||
mocks.resolveAgentDir.mockImplementation(originalAgentDir);
|
||||
@@ -270,7 +277,7 @@ describe("modelsStatusCommand auth overview", () => {
|
||||
if (originalEnvImpl) {
|
||||
mocks.resolveEnvApiKey.mockImplementation(originalEnvImpl);
|
||||
} else {
|
||||
mocks.resolveEnvApiKey.mockReset();
|
||||
mocks.resolveEnvApiKey.mockImplementation(defaultResolveEnvApiKeyImpl);
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user