From 3f86972e46a9583c73edea4534a3d4ae06ff2c2d Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Fri, 3 Apr 2026 20:40:17 +0100
Subject: [PATCH] test: trim sandbox and gateway partial mocks

---
 src/commands/sandbox-explain.test.ts                      | 4 ++--
 src/gateway/openai-http.image-budget.test.ts              | 5 +++--
 src/gateway/server-methods/chat.abort-persistence.test.ts | 5 +++--
 3 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/commands/sandbox-explain.test.ts b/src/commands/sandbox-explain.test.ts
index 349f0eb2f27..b0c141c7c47 100644
--- a/src/commands/sandbox-explain.test.ts
+++ b/src/commands/sandbox-explain.test.ts
@@ -5,8 +5,8 @@ const SANDBOX_EXPLAIN_TEST_TIMEOUT_MS = process.platform === "win32" ? 45_000 :
 
 let mockCfg: unknown = {};
 
-vi.mock("../config/config.js", async (importOriginal) => {
-  const actual = await importOriginal<typeof import("../config/config.js")>();
+vi.mock("../config/config.js", async () => {
+  const actual = await vi.importActual<typeof import("../config/config.js")>("../config/config.js");
   return {
     ...actual,
     loadConfig: vi.fn().mockImplementation(() => mockCfg),
diff --git a/src/gateway/openai-http.image-budget.test.ts b/src/gateway/openai-http.image-budget.test.ts
index fcc7e2049ae..8fcccacaa91 100644
--- a/src/gateway/openai-http.image-budget.test.ts
+++ b/src/gateway/openai-http.image-budget.test.ts
@@ -2,8 +2,9 @@ import { beforeEach, describe, expect, it, vi } from "vitest";
 
 const extractImageContentFromSourceMock = vi.fn();
 
-vi.mock("../media/input-files.js", async (importOriginal) => {
-  const actual = await importOriginal<typeof import("../media/input-files.js")>();
+vi.mock("../media/input-files.js", async () => {
+  const actual =
+    await vi.importActual<typeof import("../media/input-files.js")>("../media/input-files.js");
   return {
     ...actual,
     extractImageContentFromSource: (...args: unknown[]) =>
diff --git a/src/gateway/server-methods/chat.abort-persistence.test.ts b/src/gateway/server-methods/chat.abort-persistence.test.ts
index e11b2dc08cb..7238abe6225 100644
--- a/src/gateway/server-methods/chat.abort-persistence.test.ts
+++ b/src/gateway/server-methods/chat.abort-persistence.test.ts
@@ -18,8 +18,9 @@ const sessionEntryState = vi.hoisted(() => ({
   sessionId: "",
 }));
 
-vi.mock("../session-utils.js", async (importOriginal) => {
-  const original = await importOriginal<typeof import("../session-utils.js")>();
+vi.mock("../session-utils.js", async () => {
+  const original =
+    await vi.importActual<typeof import("../session-utils.js")>("../session-utils.js");
   return {
     ...original,
     loadSessionEntry: () => ({