diff --git a/packages/memory-host-sdk/src/host/qmd-process.test.ts b/packages/memory-host-sdk/src/host/qmd-process.test.ts index d8adadc7250..f706f525af3 100644 --- a/packages/memory-host-sdk/src/host/qmd-process.test.ts +++ b/packages/memory-host-sdk/src/host/qmd-process.test.ts @@ -6,8 +6,8 @@ import { afterEach, beforeEach, describe, expect, it, vi } from "vitest"; const spawnMock = vi.hoisted(() => vi.fn()); -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/acp/control-plane/manager.test.ts b/src/acp/control-plane/manager.test.ts index 1f09a8cee5b..ae75d6a9da2 100644 --- a/src/acp/control-plane/manager.test.ts +++ b/src/acp/control-plane/manager.test.ts @@ -25,8 +25,9 @@ vi.mock("../runtime/session-meta.js", () => ({ upsertAcpSessionMeta: (params: unknown) => hoisted.upsertAcpSessionMetaMock(params), })); -vi.mock("../runtime/registry.js", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("../runtime/registry.js", async () => { + const actual = + await vi.importActual("../runtime/registry.js"); return { ...actual, requireAcpRuntimeBackend: (backendId?: string) => diff --git a/src/channels/plugins/session-conversation.bundled-fallback.test.ts b/src/channels/plugins/session-conversation.bundled-fallback.test.ts index 26ac8628394..514750d883a 100644 --- a/src/channels/plugins/session-conversation.bundled-fallback.test.ts +++ b/src/channels/plugins/session-conversation.bundled-fallback.test.ts @@ -21,8 +21,10 @@ vi.mock("../../plugin-sdk/facade-runtime.js", () => ({ : null, })); -vi.mock("../../plugins/bundled-plugin-metadata.js", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("../../plugins/bundled-plugin-metadata.js", async () => { + const actual = await vi.importActual( + "../../plugins/bundled-plugin-metadata.js", + ); return { ...actual, resolveBundledPluginPublicSurfacePath: ({ dirName }: { dirName: string }) => diff --git a/src/cli/config-cli.test.ts b/src/cli/config-cli.test.ts index ca4b32db86d..576e88761d8 100644 --- a/src/cli/config-cli.test.ts +++ b/src/cli/config-cli.test.ts @@ -42,8 +42,11 @@ const mockLog = defaultRuntime.log; const mockError = defaultRuntime.error; const mockExit = defaultRuntime.exit; -vi.mock("../runtime.js", async (importOriginal) => { - return mockRuntimeModule(importOriginal, defaultRuntime); +vi.mock("../runtime.js", async () => { + return mockRuntimeModule( + () => vi.importActual("../runtime.js"), + defaultRuntime, + ); }); function buildSnapshot(params: { diff --git a/src/infra/path-env.test.ts b/src/infra/path-env.test.ts index 617897c45ce..56665fd9e0a 100644 --- a/src/infra/path-env.test.ts +++ b/src/infra/path-env.test.ts @@ -11,8 +11,8 @@ const abs = (p: string) => path.resolve(p); const setDir = (p: string) => state.dirs.add(abs(p)); const setExe = (p: string) => state.executables.add(abs(p)); -vi.mock("node:fs", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("node:fs", async () => { + const actual = await vi.importActual("node:fs"); const pathMod = await import("node:path"); const absInMock = (p: string) => pathMod.resolve(p); diff --git a/src/logging/config.test.ts b/src/logging/config.test.ts index f11c9217575..b8566bd34e9 100644 --- a/src/logging/config.test.ts +++ b/src/logging/config.test.ts @@ -3,8 +3,8 @@ import { readLoggingConfig } from "./config.js"; const loadConfigMock = vi.hoisted(() => vi.fn()); -vi.mock("../config/config.js", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("../config/config.js", async () => { + const actual = await vi.importActual("../config/config.js"); return { ...actual, loadConfig: () => loadConfigMock(), diff --git a/src/media-understanding/apply.test.ts b/src/media-understanding/apply.test.ts index b87644e7b81..edd05872a68 100644 --- a/src/media-understanding/apply.test.ts +++ b/src/media-understanding/apply.test.ts @@ -263,8 +263,9 @@ describe("applyMediaUnderstanding", () => { vi.doMock("../process/exec.js", () => ({ runExec: runExecMock, })); - vi.doMock("./provider-registry.js", async (importOriginal) => { - const actual = await importOriginal(); + vi.doMock("./provider-registry.js", async () => { + const actual = + await vi.importActual("./provider-registry.js"); const registryProviders = createRegistryMediaProviders(); return { ...actual, diff --git a/src/media-understanding/shared.test.ts b/src/media-understanding/shared.test.ts index 83e9dbc4555..c13f3f1f987 100644 --- a/src/media-understanding/shared.test.ts +++ b/src/media-understanding/shared.test.ts @@ -4,8 +4,10 @@ const { fetchWithSsrFGuardMock } = vi.hoisted(() => ({ fetchWithSsrFGuardMock: vi.fn(), })); -vi.mock("../infra/net/fetch-guard.js", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("../infra/net/fetch-guard.js", async () => { + const actual = await vi.importActual( + "../infra/net/fetch-guard.js", + ); return { ...actual, fetchWithSsrFGuard: fetchWithSsrFGuardMock, diff --git a/src/plugin-activation-boundary.test.ts b/src/plugin-activation-boundary.test.ts index 62e9c52bc77..64929687839 100644 --- a/src/plugin-activation-boundary.test.ts +++ b/src/plugin-activation-boundary.test.ts @@ -2,8 +2,10 @@ import { beforeEach, describe, expect, it, vi } from "vitest"; const loadBundledPluginPublicSurfaceModuleSync = vi.hoisted(() => vi.fn()); -vi.mock("./plugin-sdk/facade-runtime.js", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("./plugin-sdk/facade-runtime.js", async () => { + const actual = await vi.importActual( + "./plugin-sdk/facade-runtime.js", + ); return { ...actual, loadBundledPluginPublicSurfaceModuleSync, diff --git a/src/plugins/conversation-binding.test.ts b/src/plugins/conversation-binding.test.ts index 09d9b71a82b..0aeb04f502c 100644 --- a/src/plugins/conversation-binding.test.ts +++ b/src/plugins/conversation-binding.test.ts @@ -379,8 +379,9 @@ async function expectResolutionDoesNotWait(params: { describe("plugin conversation binding approvals", () => { beforeEach(async () => { vi.resetModules(); - vi.doMock("../infra/home-dir.js", async (importOriginal) => { - const actual = await importOriginal(); + vi.doMock("../infra/home-dir.js", async () => { + const actual = + await vi.importActual("../infra/home-dir.js"); return { ...actual, expandHomePrefix: (value: string) => { diff --git a/src/process/exec.windows.test.ts b/src/process/exec.windows.test.ts index 202e4372976..d34b54c06c1 100644 --- a/src/process/exec.windows.test.ts +++ b/src/process/exec.windows.test.ts @@ -6,12 +6,15 @@ import { afterEach, beforeEach, describe, expect, it, vi } from "vitest"; const spawnMock = vi.hoisted(() => vi.fn()); const execFileMock = vi.hoisted(() => vi.fn()); -vi.mock("node:child_process", async (importOriginal) => { +vi.mock("node:child_process", async () => { const { mockNodeBuiltinModule } = await import("../../test/helpers/node-builtin-mocks.js"); - return mockNodeBuiltinModule(importOriginal, { - spawn: spawnMock, - execFile: execFileMock, - }); + return mockNodeBuiltinModule( + () => vi.importActual("node:child_process"), + { + spawn: spawnMock, + execFile: execFileMock, + }, + ); }); let runCommandWithTimeout: typeof import("./exec.js").runCommandWithTimeout; diff --git a/src/process/kill-tree.test.ts b/src/process/kill-tree.test.ts index 3aba1c97a29..aba5355cd9d 100644 --- a/src/process/kill-tree.test.ts +++ b/src/process/kill-tree.test.ts @@ -4,11 +4,14 @@ const { spawnMock } = vi.hoisted(() => ({ spawnMock: vi.fn(), })); -vi.mock("node:child_process", async (importOriginal) => { +vi.mock("node:child_process", async () => { const { mockNodeBuiltinModule } = await import("../../test/helpers/node-builtin-mocks.js"); - return mockNodeBuiltinModule(importOriginal, { - spawn: (...args: unknown[]) => spawnMock(...args), - }); + return mockNodeBuiltinModule( + () => vi.importActual("node:child_process"), + { + spawn: (...args: unknown[]) => spawnMock(...args), + }, + ); }); let killProcessTree: typeof import("./kill-tree.js").killProcessTree; diff --git a/src/security/windows-acl.test.ts b/src/security/windows-acl.test.ts index 369f97a8cb8..5c8e9726471 100644 --- a/src/security/windows-acl.test.ts +++ b/src/security/windows-acl.test.ts @@ -3,10 +3,10 @@ import type { WindowsAclEntry, WindowsAclSummary } from "./windows-acl.js"; const MOCK_USERNAME = "MockUser"; -vi.mock("node:os", async (importOriginal) => { +vi.mock("node:os", async () => { const { mockNodeBuiltinModule } = await import("../../test/helpers/node-builtin-mocks.js"); return mockNodeBuiltinModule( - importOriginal, + () => vi.importActual("node:os"), { userInfo: () => ({ username: MOCK_USERNAME }) }, { mirrorToDefault: true }, ); diff --git a/src/tts/tts.test.ts b/src/tts/tts.test.ts index 1244d3c4e38..2bc9618f4c8 100644 --- a/src/tts/tts.test.ts +++ b/src/tts/tts.test.ts @@ -3,8 +3,10 @@ import { beforeEach, describe, expect, it, vi } from "vitest"; const loadBundledPluginPublicSurfaceModuleSync = vi.hoisted(() => vi.fn()); const loadActivatedBundledPluginPublicSurfaceModuleSync = vi.hoisted(() => vi.fn()); -vi.mock("../plugin-sdk/facade-runtime.js", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("../plugin-sdk/facade-runtime.js", async () => { + const actual = await vi.importActual( + "../plugin-sdk/facade-runtime.js", + ); return { ...actual, loadActivatedBundledPluginPublicSurfaceModuleSync,