diff --git a/src/agents/sandbox/ssh.spawn-env.test.ts b/src/agents/sandbox/ssh.spawn-env.test.ts index cdb23c9c9a0..17837fa55d0 100644 --- a/src/agents/sandbox/ssh.spawn-env.test.ts +++ b/src/agents/sandbox/ssh.spawn-env.test.ts @@ -24,8 +24,8 @@ function createMockChildProcess(): MockChildProcess { return child; } -vi.mock("node:child_process", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("node:child_process", async () => { + const actual = await vi.importActual("node:child_process"); return { ...actual, spawn: spawnMock, diff --git a/src/agents/skills-install.download.test.ts b/src/agents/skills-install.download.test.ts index 8439f62bf74..52a95208a77 100644 --- a/src/agents/skills-install.download.test.ts +++ b/src/agents/skills-install.download.test.ts @@ -21,8 +21,8 @@ vi.mock("../infra/net/fetch-guard.js", () => ({ fetchWithSsrFGuard: (...args: unknown[]) => fetchWithSsrFGuardMock(...args), })); -vi.mock("./skills.js", async (importOriginal) => ({ - ...(await importOriginal()), +vi.mock("./skills.js", async () => ({ + ...(await vi.importActual("./skills.js")), hasBinary: (bin: string) => hasBinaryMock(bin), })); diff --git a/src/agents/tools-effective-inventory.test.ts b/src/agents/tools-effective-inventory.test.ts index 1c94f23f1fa..5758c33d237 100644 --- a/src/agents/tools-effective-inventory.test.ts +++ b/src/agents/tools-effective-inventory.test.ts @@ -9,8 +9,8 @@ async function loadHarness(options?: { resolvedModelCompat?: Record; }) { vi.resetModules(); - vi.doMock("./agent-scope.js", async (importOriginal) => { - const actual = await importOriginal(); + vi.doMock("./agent-scope.js", async () => { + const actual = await vi.importActual("./agent-scope.js"); return { ...actual, resolveSessionAgentId: () => "main",