From 677fc0694019175cfbbe30769ca6e6028f691a79 Mon Sep 17 00:00:00 2001
From: Cursor Agent <cursoragent@cursor.com>
Date: Sun, 15 Feb 2026 02:13:39 +0000
Subject: [PATCH] Cover reconnect behavior when cleanup and onError all fail

Co-authored-by: Eric Allam <eric@trigger.dev>
---
 packages/ai/src/chatTransport.test.ts | 37 +++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/packages/ai/src/chatTransport.test.ts b/packages/ai/src/chatTransport.test.ts
index 8faa266ef..cba7d672a 100644
--- a/packages/ai/src/chatTransport.test.ts
+++ b/packages/ai/src/chatTransport.test.ts
@@ -2494,6 +2494,43 @@ describe("TriggerChatTransport", function () {
     }
   );
 
+  it(
+    "preserves reconnect root failures when cleanup set/delete and onError callbacks all fail",
+    async function () {
+      const runStore = new FailingCleanupSetAndDeleteRunStore();
+      runStore.set({
+        chatId: "chat-reconnect-cleanup-all-failure",
+        runId: "run_reconnect_cleanup_all_failure",
+        publicAccessToken: "pk_reconnect_cleanup_all_failure",
+        streamKey: "chat-stream",
+        lastEventId: "100-0",
+        isActive: true,
+      });
+
+      const transport = new TriggerChatTransport({
+        task: "chat-task",
+        stream: "chat-stream",
+        accessToken: "pk_trigger",
+        runStore,
+        onError: async function onError() {
+          throw new Error("onError failed");
+        },
+      });
+
+      (transport as any).fetchRunStream = async function fetchRunStream() {
+        throw new Error("reconnect root cause");
+      };
+
+      const stream = await transport.reconnectToStream({
+        chatId: "chat-reconnect-cleanup-all-failure",
+      });
+
+      expect(stream).toBeNull();
+      expect(runStore.setCalls).toContain("chat-reconnect-cleanup-all-failure");
+      expect(runStore.deleteCalls).toContain("chat-reconnect-cleanup-all-failure");
+    }
+  );
+
   it("normalizes non-Error reconnect failures before reporting onError", async function () {
     const errors: TriggerChatTransportError[] = [];
     const runStore = new InMemoryTriggerChatRunStore();