From 50a22245f8d3d9851a1a70e74934f79b518be0dd Mon Sep 17 00:00:00 2001
From: Mahima Shanware <mshanware@google.com>
Date: Fri, 27 Mar 2026 17:15:43 +0000
Subject: [PATCH] fix(cli): update DefaultAppLayout mock to include btwState

---
 packages/cli/src/ui/layouts/DefaultAppLayout.test.tsx | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/packages/cli/src/ui/layouts/DefaultAppLayout.test.tsx b/packages/cli/src/ui/layouts/DefaultAppLayout.test.tsx
index 6be2424194..6bfd238b06 100644
--- a/packages/cli/src/ui/layouts/DefaultAppLayout.test.tsx
+++ b/packages/cli/src/ui/layouts/DefaultAppLayout.test.tsx
@@ -38,6 +38,13 @@ const mockUIState = {
   availableTerminalHeight: 20,
   activePtyId: null,
   isBackgroundTaskVisible: true,
+  btwState: {
+    isActive: false,
+    query: '',
+    response: '',
+    isStreaming: false,
+    error: null,
+  },
 } as unknown as UIState;
 
 vi.mock('../contexts/UIStateContext.js', () => ({