From bc30277683fc91b4fa39bc109666430fade8d2fd Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Tue, 18 Aug 2026 10:27:32 -0700
Subject: [PATCH] perf(test): narrow harness differential mock (#125911)

---
 ...om-config.turn-model-selection-differential.test.ts | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

diff --git a/src/auto-reply/reply/dispatch-from-config.turn-model-selection-differential.test.ts b/src/auto-reply/reply/dispatch-from-config.turn-model-selection-differential.test.ts
index f7539e444ace..012db2412af9 100644
--- a/src/auto-reply/reply/dispatch-from-config.turn-model-selection-differential.test.ts
+++ b/src/auto-reply/reply/dispatch-from-config.turn-model-selection-differential.test.ts
@@ -20,13 +20,9 @@ import { buildTestCtx } from "./test-ctx.js";
 
 const selectAgentHarnessMock = vi.hoisted(() => vi.fn());
 
-vi.mock("../../agents/harness/selection.js", async (importOriginal) => {
-  const actual = await importOriginal<typeof import("../../agents/harness/selection.js")>();
-  return {
-    ...actual,
-    selectAgentHarness: (...args: unknown[]) => selectAgentHarnessMock(...args),
-  };
-});
+vi.mock("../../agents/harness/selection.js", () => ({
+  selectAgentHarness: (...args: unknown[]) => selectAgentHarnessMock(...args),
+}));
 
 const { resolveVisibleRepliesPolicy } = await import("./dispatch-from-config.harness-defaults.js");