From 2b79de486bc630e19d90ddd5fc48da1930ba6ab6 Mon Sep 17 00:00:00 2001
From: joshavant <830519+joshavant@users.noreply.github.com>
Date: Sun, 2 Aug 2026 22:20:14 -0500
Subject: [PATCH] test: isolate image tool from delivery queue runtime

---
 src/agents/tools/image-generate-tool.test.ts | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/agents/tools/image-generate-tool.test.ts b/src/agents/tools/image-generate-tool.test.ts
index e6512611b3f0..2fe72839032f 100644
--- a/src/agents/tools/image-generate-tool.test.ts
+++ b/src/agents/tools/image-generate-tool.test.ts
@@ -24,9 +24,14 @@ const taskRuntimeMocks = vi.hoisted(() => ({
 const sessionAccessorMocks = vi.hoisted(() => ({
   loadSessionEntryReadOnly: vi.fn(),
 }));
+const subagentAnnounceDeliveryMocks = vi.hoisted(() => ({
+  deliverSubagentAnnouncement: vi.fn(),
+  loadRequesterSessionEntry: vi.fn(),
+}));
 
 vi.mock("../../tasks/runtime-internal.js", () => taskRuntimeInternalMocks);
 vi.mock("../../tasks/detached-task-runtime.js", () => taskRuntimeMocks);
+vi.mock("../subagent-announce-delivery.js", () => subagentAnnounceDeliveryMocks);
 vi.mock("../../config/sessions/session-accessor.js", async (importOriginal) => ({
   ...(await importOriginal<typeof import("../../config/sessions/session-accessor.js")>()),
   loadSessionEntryReadOnly: sessionAccessorMocks.loadSessionEntryReadOnly,
@@ -372,6 +377,14 @@ describe("createImageGenerateTool", () => {
     taskRuntimeMocks.completeTaskRunByRunId.mockReset();
     taskRuntimeMocks.failTaskRunByRunId.mockReset();
     sessionAccessorMocks.loadSessionEntryReadOnly.mockReset();
+    subagentAnnounceDeliveryMocks.deliverSubagentAnnouncement.mockReset();
+    subagentAnnounceDeliveryMocks.deliverSubagentAnnouncement.mockResolvedValue({
+      delivered: true,
+      path: "direct",
+      disposition: "delivered",
+    });
+    subagentAnnounceDeliveryMocks.loadRequesterSessionEntry.mockReset();
+    subagentAnnounceDeliveryMocks.loadRequesterSessionEntry.mockReturnValue({ entry: undefined });
     taskRuntimeInternalMocks.listTasksForOwnerKey.mockReset();
     taskRuntimeInternalMocks.listTasksForOwnerKey.mockReturnValue([]);
     taskRuntimeInternalMocks.listFreshTasksForOwnerKey.mockReset();