From 13bcc9f70d1321d759ceac01aaab46f628933c4f Mon Sep 17 00:00:00 2001
From: James Long <longster@gmail.com>
Date: Wed, 8 Jul 2026 14:08:12 +0000
Subject: [PATCH] test(tui): cover streaming shell state

---
 packages/tui/test/cli/tui/data.test.tsx | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/packages/tui/test/cli/tui/data.test.tsx b/packages/tui/test/cli/tui/data.test.tsx
index c116f43178..94da8151f2 100644
--- a/packages/tui/test/cli/tui/data.test.tsx
+++ b/packages/tui/test/cli/tui/data.test.tsx
@@ -1770,6 +1770,14 @@ test("settles pending tools when a live failure arrives", async () => {
         name: "bash",
       },
     })
+    await wait(() => {
+      const assistant = sync.session.message.get("session-1", "msg_explicit_assistant_9")
+      return (
+        assistant?.type === "assistant" &&
+        assistant.content[0]?.type === "tool" &&
+        assistant.content[0].state.status === "streaming"
+      )
+    })
     emitEvent(events, {
       id: "evt_called_1",
       created: 0,