test(core): reduce mock reset overhead in targeted suites

This commit is contained in:
Peter Steinberger
2026-02-22 08:40:24 +00:00
parent 2a66c8d676
commit ccc00d874c
8 changed files with 19 additions and 12 deletions

View File

@@ -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);
}
}
});