From bec4401afd91e003d3355fbb3dcba6690070d89c Mon Sep 17 00:00:00 2001
From: Cursor Agent <cursoragent@cursor.com>
Date: Sun, 15 Feb 2026 02:55:08 +0000
Subject: [PATCH] Assert full run-state transition snapshots on stream
 completion

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

diff --git a/packages/ai/src/chatTransport.test.ts b/packages/ai/src/chatTransport.test.ts
index 28479f3f7..40a8214b2 100644
--- a/packages/ai/src/chatTransport.test.ts
+++ b/packages/ai/src/chatTransport.test.ts
@@ -2989,6 +2989,31 @@ describe("TriggerChatTransport", function () {
       return trackedRunStore.deleteCalls.includes("chat-cleanup");
     });
 
+    expect(trackedRunStore.setSnapshots).toHaveLength(4);
+    expect(trackedRunStore.setSnapshots[0]).toMatchObject({
+      chatId: "chat-cleanup",
+      runId: "run_cleanup",
+      isActive: true,
+      lastEventId: undefined,
+    });
+    expect(trackedRunStore.setSnapshots[1]).toMatchObject({
+      chatId: "chat-cleanup",
+      runId: "run_cleanup",
+      isActive: true,
+      lastEventId: "1-0",
+    });
+    expect(trackedRunStore.setSnapshots[2]).toMatchObject({
+      chatId: "chat-cleanup",
+      runId: "run_cleanup",
+      isActive: true,
+      lastEventId: "2-0",
+    });
+    expect(trackedRunStore.setSnapshots[3]).toMatchObject({
+      chatId: "chat-cleanup",
+      runId: "run_cleanup",
+      isActive: false,
+      lastEventId: "2-0",
+    });
     expect(trackedRunStore.get("chat-cleanup")).toBeUndefined();
   });