diff --git a/src/agents/model-fallback.run-embedded.e2e.test.ts b/src/agents/model-fallback.run-embedded.e2e.test.ts index 5464ea78bfc..353755ae152 100644 --- a/src/agents/model-fallback.run-embedded.e2e.test.ts +++ b/src/agents/model-fallback.run-embedded.e2e.test.ts @@ -23,16 +23,18 @@ const { computeBackoffMock, sleepWithAbortMock } = vi.hoisted(() => ({ sleepWithAbortMock: vi.fn(async (_ms: number, _abortSignal?: AbortSignal) => undefined), })); -vi.mock("./pi-embedded-runner/run/attempt.js", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("./pi-embedded-runner/run/attempt.js", async () => { + const actual = await vi.importActual( + "./pi-embedded-runner/run/attempt.js", + ); return { ...actual, runEmbeddedAttempt: (params: unknown) => runEmbeddedAttemptMock(params), }; }); -vi.mock("../infra/backoff.js", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("../infra/backoff.js", async () => { + const actual = await vi.importActual("../infra/backoff.js"); return { ...actual, computeBackoff: ( @@ -43,8 +45,8 @@ vi.mock("../infra/backoff.js", async (importOriginal) => { }; }); -vi.mock("./models-config.js", async (importOriginal) => { - const mod = await importOriginal(); +vi.mock("./models-config.js", async () => { + const mod = await vi.importActual("./models-config.js"); return { ...mod, ensureOpenClawModelsJson: vi.fn(async () => ({ wrote: false })), @@ -68,8 +70,10 @@ const installRunEmbeddedMocks = () => { resolveModelAsync: async (provider: string, modelId: string) => createResolvedEmbeddedRunnerModel(provider, modelId), })); - vi.doMock("../plugins/provider-runtime.js", async (importOriginal) => { - const actual = await importOriginal(); + vi.doMock("../plugins/provider-runtime.js", async () => { + const actual = await vi.importActual( + "../plugins/provider-runtime.js", + ); return { ...actual, prepareProviderRuntimeAuth: vi.fn(async () => undefined), diff --git a/src/agents/pi-embedded-runner.e2e.test.ts b/src/agents/pi-embedded-runner.e2e.test.ts index cc94a2e9d49..270fdba830f 100644 --- a/src/agents/pi-embedded-runner.e2e.test.ts +++ b/src/agents/pi-embedded-runner.e2e.test.ts @@ -20,8 +20,8 @@ const disposeSessionMcpRuntimeMock = vi.fn<(sessionId: string) => Promise> }); let refreshRuntimeAuthOnFirstPromptError = false; -vi.mock("@mariozechner/pi-ai", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("@mariozechner/pi-ai", async () => { + const actual = await vi.importActual("@mariozechner/pi-ai"); const buildAssistantMessage = (model: { api: string; provider: string; id: string }) => ({ role: "assistant" as const, @@ -101,8 +101,10 @@ const installRunEmbeddedMocks = () => { vi.doMock("./pi-bundle-mcp-tools.js", () => ({ disposeSessionMcpRuntime: (sessionId: string) => disposeSessionMcpRuntimeMock(sessionId), })); - vi.doMock("./pi-embedded-runner/model.js", async (importOriginal) => { - const actual = await importOriginal(); + vi.doMock("./pi-embedded-runner/model.js", async () => { + const actual = await vi.importActual( + "./pi-embedded-runner/model.js", + ); return { ...actual, resolveModelAsync: async (provider: string, modelId: string) => @@ -119,15 +121,17 @@ const installRunEmbeddedMocks = () => { stopRuntimeAuthRefreshTimer: vi.fn(), }), })); - vi.doMock("../plugins/provider-runtime.js", async (importOriginal) => { - const actual = await importOriginal(); + vi.doMock("../plugins/provider-runtime.js", async () => { + const actual = await vi.importActual( + "../plugins/provider-runtime.js", + ); return { ...actual, prepareProviderRuntimeAuth: vi.fn(async () => undefined), }; }); - vi.doMock("./models-config.js", async (importOriginal) => { - const mod = await importOriginal(); + vi.doMock("./models-config.js", async () => { + const mod = await vi.importActual("./models-config.js"); return { ...mod, ensureOpenClawModelsJson: vi.fn(async () => ({ wrote: false })), diff --git a/src/agents/pi-embedded-runner.run-embedded-pi-agent.auth-profile-rotation.e2e.test.ts b/src/agents/pi-embedded-runner.run-embedded-pi-agent.auth-profile-rotation.e2e.test.ts index 48d9ebd6a65..006a7fdcaeb 100644 --- a/src/agents/pi-embedded-runner.run-embedded-pi-agent.auth-profile-rotation.e2e.test.ts +++ b/src/agents/pi-embedded-runner.run-embedded-pi-agent.auth-profile-rotation.e2e.test.ts @@ -59,8 +59,10 @@ const installRunEmbeddedMocks = () => { vi.doMock("./pi-embedded-runner/run/attempt.js", () => ({ runEmbeddedAttempt: (params: unknown) => runEmbeddedAttemptMock(params), })); - vi.doMock("../plugins/provider-runtime.js", async (importOriginal) => { - const actual = await importOriginal(); + vi.doMock("../plugins/provider-runtime.js", async () => { + const actual = await vi.importActual( + "../plugins/provider-runtime.js", + ); return { ...actual, prepareProviderRuntimeAuth: async (params: { @@ -92,8 +94,8 @@ const installRunEmbeddedMocks = () => { throw new Error("compact should not run in auth profile rotation tests"); }), })); - vi.doMock("./models-config.js", async (importOriginal) => { - const mod = await importOriginal(); + vi.doMock("./models-config.js", async () => { + const mod = await vi.importActual("./models-config.js"); return { ...mod, ensureOpenClawModelsJson: vi.fn(async () => ({ wrote: false })), diff --git a/src/agents/sandbox-agent-config.agent-specific-sandbox-config.e2e.test.ts b/src/agents/sandbox-agent-config.agent-specific-sandbox-config.e2e.test.ts index 80660fb14dd..a4ad9c481f8 100644 --- a/src/agents/sandbox-agent-config.agent-specific-sandbox-config.e2e.test.ts +++ b/src/agents/sandbox-agent-config.agent-specific-sandbox-config.e2e.test.ts @@ -12,8 +12,8 @@ type SpawnCall = { const spawnCalls: SpawnCall[] = []; -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: (command: string, args: string[]) => { @@ -42,8 +42,8 @@ vi.mock("node:child_process", async (importOriginal) => { }; }); -vi.mock("./skills.js", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("./skills.js", async () => { + const actual = await vi.importActual("./skills.js"); return { ...actual, syncSkillsToWorkspace: vi.fn(async () => undefined), diff --git a/src/commands/onboard-channels.e2e.test.ts b/src/commands/onboard-channels.e2e.test.ts index a94b0ccb80e..a1cbda461d0 100644 --- a/src/commands/onboard-channels.e2e.test.ts +++ b/src/commands/onboard-channels.e2e.test.ts @@ -575,8 +575,10 @@ vi.mock("../channel-web.js", () => ({ loginWeb: vi.fn(async () => {}), })); -vi.mock("../channels/plugins/catalog.js", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("../channels/plugins/catalog.js", async () => { + const actual = await vi.importActual( + "../channels/plugins/catalog.js", + ); return { ...actual, listChannelPluginCatalogEntries: ((...args) => { @@ -589,8 +591,10 @@ vi.mock("../channels/plugins/catalog.js", async (importOriginal) => { }; }); -vi.mock("../plugins/manifest-registry.js", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("../plugins/manifest-registry.js", async () => { + const actual = await vi.importActual( + "../plugins/manifest-registry.js", + ); return { ...actual, loadPluginManifestRegistry: manifestRegistryMocks.loadPluginManifestRegistry, @@ -606,8 +610,8 @@ vi.mock("./onboard-helpers.js", () => ({ detectBinary: vi.fn(async () => false), })); -vi.mock("./channel-setup/plugin-install.js", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("./channel-setup/plugin-install.js", async () => { + const actual = await vi.importActual("./channel-setup/plugin-install.js"); return { ...(actual as Record), ensureChannelSetupPluginInstalled: vi.fn(async ({ cfg }: { cfg: OpenClawConfig }) => ({