From 2fdeb7af9675013686cfae2a91b4235afeacd7d8 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Wed, 8 Apr 2026 11:20:00 +0100
Subject: [PATCH] test: stub image provider discovery in generation tool tests

---
 .../pi-embedded-runner.compaction-safety-timeout.test.ts  | 8 +++++++-
 src/agents/tools/image-generate-tool.test.ts              | 1 +
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/agents/pi-embedded-runner.compaction-safety-timeout.test.ts b/src/agents/pi-embedded-runner.compaction-safety-timeout.test.ts
index 1898373cfc95..4cedf36bed9b 100644
--- a/src/agents/pi-embedded-runner.compaction-safety-timeout.test.ts
+++ b/src/agents/pi-embedded-runner.compaction-safety-timeout.test.ts
@@ -1,4 +1,4 @@
-import { afterEach, describe, expect, it, vi } from "vitest";
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
 import {
   compactWithSafetyTimeout,
   EMBEDDED_COMPACTION_TIMEOUT_MS,
@@ -6,7 +6,13 @@ import {
 } from "./pi-embedded-runner/compaction-safety-timeout.js";
 
 describe("compactWithSafetyTimeout", () => {
+  beforeEach(() => {
+    vi.useRealTimers();
+    vi.clearAllTimers();
+  });
+
   afterEach(() => {
+    vi.clearAllTimers();
     vi.useRealTimers();
   });
 
diff --git a/src/agents/tools/image-generate-tool.test.ts b/src/agents/tools/image-generate-tool.test.ts
index 82d10f4b8bf9..751e84b98bf6 100644
--- a/src/agents/tools/image-generate-tool.test.ts
+++ b/src/agents/tools/image-generate-tool.test.ts
@@ -564,6 +564,7 @@ describe("createImageGenerateTool", () => {
   });
 
   it("ignores non-finite mediaMaxMb when loading reference images", async () => {
+    stubImageGenerationProviders();
     stubEditedImageFlow({ width: 3200, height: 1800 });
     const tool = requireImageGenerateTool(
       createImageGenerateTool({