From 07522218491653962ec217bd5f6ca8f1d4fbbc02 Mon Sep 17 00:00:00 2001
From: Dev Randalpura <devrandalpura@google.com>
Date: Fri, 6 Mar 2026 13:26:08 -0500
Subject: [PATCH] Updated tests for ShellToolMessage

---
 .../messages/ShellToolMessage.test.tsx        |  4 ++--
 .../ShellToolMessage.test.tsx.snap            | 24 ++++++-------------
 2 files changed, 9 insertions(+), 19 deletions(-)

diff --git a/packages/cli/src/ui/components/messages/ShellToolMessage.test.tsx b/packages/cli/src/ui/components/messages/ShellToolMessage.test.tsx
index 233f905760..8dc6dac738 100644
--- a/packages/cli/src/ui/components/messages/ShellToolMessage.test.tsx
+++ b/packages/cli/src/ui/components/messages/ShellToolMessage.test.tsx
@@ -181,7 +181,7 @@ describe('<ShellToolMessage />', () => {
       [
         'respects availableTerminalHeight when it is smaller than ACTIVE_SHELL_MAX_LINES',
         10,
-        8,
+        3, // 10 - 1 (static) - 6 (ASB reserved) = 3
         false,
       ],
       [
@@ -193,7 +193,7 @@ describe('<ShellToolMessage />', () => {
       [
         'uses full availableTerminalHeight when focused in alternate buffer mode',
         100,
-        98, // 100 - 2
+        93, // 100 - 1 (static) - 6 (ASB reserved) = 93
         true,
       ],
       [
diff --git a/packages/cli/src/ui/components/messages/__snapshots__/ShellToolMessage.test.tsx.snap b/packages/cli/src/ui/components/messages/__snapshots__/ShellToolMessage.test.tsx.snap
index b51d7c435b..c4950b2c33 100644
--- a/packages/cli/src/ui/components/messages/__snapshots__/ShellToolMessage.test.tsx.snap
+++ b/packages/cli/src/ui/components/messages/__snapshots__/ShellToolMessage.test.tsx.snap
@@ -133,11 +133,6 @@ exports[`<ShellToolMessage /> > Height Constraints > respects availableTerminalH
 "╭──────────────────────────────────────────────────────────────────────────────╮
 │ ⊶  Shell Command A shell command                                             │
 │                                                                              │
-│ Line 93                                                                      │
-│ Line 94                                                                      │
-│ Line 95                                                                      │
-│ Line 96                                                                      │
-│ Line 97                                                                      │
 │ Line 98                                                                      │
 │ Line 99                                                                      │
 │ Line 100                                                                   █ │
@@ -192,18 +187,13 @@ exports[`<ShellToolMessage /> > Height Constraints > uses full availableTerminal
 "╭──────────────────────────────────────────────────────────────────────────────╮
 │ ⊶  Shell Command A shell command                      (Shift+Tab to unfocus) │
 │                                                                              │
-│ Line 3                                                                       │
-│ Line 4                                                                       │
-│ Line 5                                                                     █ │
-│ Line 6                                                                     █ │
-│ Line 7                                                                     █ │
-│ Line 8                                                                     █ │
-│ Line 9                                                                     █ │
-│ Line 10                                                                    █ │
-│ Line 11                                                                    █ │
-│ Line 12                                                                    █ │
-│ Line 13                                                                    █ │
-│ Line 14                                                                    █ │
+│ Line 8                                                                       │
+│ Line 9                                                                       │
+│ Line 10                                                                      │
+│ Line 11                                                                      │
+│ Line 12                                                                      │
+│ Line 13                                                                      │
+│ Line 14                                                                    ▄ │
 │ Line 15                                                                    █ │
 │ Line 16                                                                    █ │
 │ Line 17                                                                    █ │