Tests: reuse embedded runner harness imports

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();

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(() => {