test: speed up sanitize session history policy smoke

Author: steipete

src/agents/pi-embedded-runner.sanitize-session-history.policy.test.ts

@@ -1,4 +1,4 @@
-import { beforeEach, describe, expect, it, vi } from "vitest";
+import { beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
 import {
   loadSanitizeSessionHistoryWithCleanMocks,
   makeMockSessionManager,
@@ -14,19 +14,30 @@ vi.mock("./pi-embedded-helpers.js", async () => ({
   sanitizeSessionMessagesImages: vi.fn(async (msgs) => msgs),
 }));
 
+vi.mock("../plugins/provider-runtime.js", () => ({
+  resolveProviderRuntimePlugin: vi.fn(() => undefined),
+  sanitizeProviderReplayHistoryWithPlugin: vi.fn(() => undefined),
+  validateProviderReplayTurnsWithPlugin: vi.fn(() => undefined),
+}));
+
 let sanitizeSessionHistory: SanitizeSessionHistoryHarness["sanitizeSessionHistory"];
 let mockedHelpers: SanitizeSessionHistoryHarness["mockedHelpers"];
 
 describe("sanitizeSessionHistory e2e smoke", () => {
   const mockSessionManager = makeMockSessionManager();
   const mockMessages = makeSimpleUserMessages();
 
-  beforeEach(async () => {
+  beforeAll(async () => {
     const harness = await loadSanitizeSessionHistoryWithCleanMocks();
     sanitizeSessionHistory = harness.sanitizeSessionHistory;
     mockedHelpers = harness.mockedHelpers;
   });
 
+  beforeEach(() => {
+    vi.clearAllMocks();
+    vi.mocked(mockedHelpers.sanitizeSessionMessagesImages).mockImplementation(async (msgs) => msgs);
+  });
+
   it("passes simple user-only history through for google model APIs", async () => {
     vi.mocked(mockedHelpers.isGoogleModelApi).mockReturnValue(true);