test(agents): preserve provider hook mock exports

giodl73-repo · giodl73-repo · commit bb8fdb65b7bd · 2026-05-25T07:42:38.000-07:00
diff --git a/src/agents/pi-embedded-runner.openai-tool-id-preservation.test.ts b/src/agents/pi-embedded-runner.openai-tool-id-preservation.test.ts
@@ -17,23 +17,25 @@ vi.mock(
   "../plugins/provider-runtime.js",
   async () => await createSanitizeSessionHistoryProviderRuntimeMock(),
 );
-vi.mock("../plugins/provider-hook-runtime.js", () =>
-  createSanitizeSessionHistoryProviderHookRuntimeMock({
-    resolveProviderRuntimePlugin: vi.fn(({ provider }: { provider?: string }) =>
-      provider === "openai"
-        ? {
-            buildReplayPolicy: (context?: { modelApi?: string }) => ({
-              sanitizeMode: "images-only",
-              sanitizeToolCallIds: context?.modelApi === "openai-completions",
-              ...(context?.modelApi === "openai-completions" ? { toolCallIdMode: "strict" } : {}),
-              applyAssistantFirstOrderingFix: false,
-              validateGeminiTurns: false,
-              validateAnthropicTurns: false,
-            }),
-          }
-        : undefined,
-    ),
-  }),
+vi.mock(
+  "../plugins/provider-hook-runtime.js",
+  async () =>
+    await createSanitizeSessionHistoryProviderHookRuntimeMock({
+      resolveProviderRuntimePlugin: vi.fn(({ provider }: { provider?: string }) =>
+        provider === "openai"
+          ? {
+              buildReplayPolicy: (context?: { modelApi?: string }) => ({
+                sanitizeMode: "images-only",
+                sanitizeToolCallIds: context?.modelApi === "openai-completions",
+                ...(context?.modelApi === "openai-completions" ? { toolCallIdMode: "strict" } : {}),
+                applyAssistantFirstOrderingFix: false,
+                validateGeminiTurns: false,
+                validateAnthropicTurns: false,
+              }),
+            }
+          : undefined,
+      ),
+    }),
 );
 
 describe("sanitizeSessionHistory openai tool id preservation", () => {
diff --git a/src/agents/pi-embedded-runner.sanitize-session-history.policy.test.ts b/src/agents/pi-embedded-runner.sanitize-session-history.policy.test.ts
@@ -21,8 +21,9 @@ vi.mock(
   "../plugins/provider-runtime.js",
   async () => await createSanitizeSessionHistoryProviderRuntimeMock(),
 );
-vi.mock("../plugins/provider-hook-runtime.js", () =>
-  createSanitizeSessionHistoryProviderHookRuntimeMock(),
+vi.mock(
+  "../plugins/provider-hook-runtime.js",
+  async () => await createSanitizeSessionHistoryProviderHookRuntimeMock(),
 );
 
 let sanitizeSessionHistory: SanitizeSessionHistoryHarness["sanitizeSessionHistory"];
diff --git a/src/agents/pi-embedded-runner.sanitize-session-history.test-harness.ts b/src/agents/pi-embedded-runner.sanitize-session-history.test-harness.ts
@@ -84,11 +84,15 @@ export async function createSanitizeSessionHistoryProviderRuntimeMock(
   };
 }
 
-export function createSanitizeSessionHistoryProviderHookRuntimeMock(
+export async function createSanitizeSessionHistoryProviderHookRuntimeMock(
   extra: Record<string, unknown> = {},
 ) {
   const clearProviderRuntimePluginCacheForTest = vi.fn();
+  const actual = await vi.importActual<typeof import("../plugins/provider-hook-runtime.js")>(
+    "../plugins/provider-hook-runtime.js",
+  );
   return {
+    ...actual,
     clearProviderRuntimePluginCacheForTest,
     resolveProviderRuntimePlugin: vi.fn(() => undefined),
     resolveProviderHookPlugin: vi.fn(() => undefined),
