From d073ec42cd7fabd1004f6959628743817a4cb0e8 Mon Sep 17 00:00:00 2001
From: Gustavo Madeira Santana <gumadeiras@gmail.com>
Date: Wed, 18 Mar 2026 01:20:04 +0000
Subject: [PATCH] Tests: reuse embedded runner harness imports

---
 src/agents/pi-embedded-runner/compact.hooks.test.ts   |  9 +++++++--
 .../run.overflow-compaction.test.ts                   | 11 +++++++----
 2 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/src/agents/pi-embedded-runner/compact.hooks.test.ts b/src/agents/pi-embedded-runner/compact.hooks.test.ts
index c5806609c0d..1a97501959e 100644
--- a/src/agents/pi-embedded-runner/compact.hooks.test.ts
+++ b/src/agents/pi-embedded-runner/compact.hooks.test.ts
@@ -1,5 +1,5 @@
 import { getApiProvider, unregisterApiProviders } from "@mariozechner/pi-ai";
-import { beforeEach, describe, expect, it, vi } from "vitest";
+import { beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
 import { getCustomApiRegistrySourceId } from "../custom-api-registry.js";
 import {
   contextEngineCompactMock,
@@ -13,6 +13,7 @@ import {
   resolveMemorySearchConfigMock,
   resolveModelMock,
   resolveSessionAgentIdMock,
+  resetCompactHooksHarnessMocks,
   sanitizeSessionHistoryMock,
   sessionAbortCompactionMock,
   sessionCompactImpl,
@@ -103,13 +104,17 @@ const sessionHook = (action: string): SessionHookEvent | undefined =>
     return event?.type === "session" && event.action === action;
   })?.[0] as SessionHookEvent | undefined;
 
-beforeEach(async () => {
+beforeAll(async () => {
   const loaded = await loadCompactHooksHarness();
   compactEmbeddedPiSessionDirect = loaded.compactEmbeddedPiSessionDirect;
   compactEmbeddedPiSession = loaded.compactEmbeddedPiSession;
   onSessionTranscriptUpdate = loaded.onSessionTranscriptUpdate;
 });
 
+beforeEach(() => {
+  resetCompactHooksHarnessMocks();
+});
+
 describe("compactEmbeddedPiSessionDirect hooks", () => {
   beforeEach(() => {
     ensureRuntimePluginsLoaded.mockReset();
diff --git a/src/agents/pi-embedded-runner/run.overflow-compaction.test.ts b/src/agents/pi-embedded-runner/run.overflow-compaction.test.ts
index 75a9ab6e034..1f5f0b6de35 100644
--- a/src/agents/pi-embedded-runner/run.overflow-compaction.test.ts
+++ b/src/agents/pi-embedded-runner/run.overflow-compaction.test.ts
@@ -1,4 +1,4 @@
-import { beforeEach, describe, expect, it } from "vitest";
+import { beforeAll, beforeEach, describe, expect, it } from "vitest";
 import {
   makeAttemptResult,
   makeCompactionSuccess,
@@ -16,6 +16,7 @@ import {
   mockedContextEngine,
   mockedCompactDirect,
   mockedRunEmbeddedAttempt,
+  resetRunOverflowCompactionHarnessMocks,
   mockedSessionLikelyHasOversizedToolResults,
   mockedTruncateOversizedToolResultsInSession,
   overflowBaseRunParams,
@@ -24,10 +25,12 @@ import {
 let runEmbeddedPiAgent: typeof import("./run.js").runEmbeddedPiAgent;
 
 describe("runEmbeddedPiAgent overflow compaction trigger routing", () => {
+  beforeAll(async () => {
+    ({ runEmbeddedPiAgent } = await loadRunOverflowCompactionHarness());
+  });
+
   beforeEach(() => {
-    return loadRunOverflowCompactionHarness().then((loaded) => {
-      runEmbeddedPiAgent = loaded.runEmbeddedPiAgent;
-    });
+    resetRunOverflowCompactionHarnessMocks();
   });
 
   beforeEach(() => {