From fc5db6c2ab4c48f0484da5ead4ed1569e029da7e Mon Sep 17 00:00:00 2001
From: mkorwel <matt.korwel@gmail.com>
Date: Fri, 17 Apr 2026 23:20:11 +0000
Subject: [PATCH] test: fix ShowMoreLinesLayout tests

---
 .../cli/src/ui/components/ShowMoreLinesLayout.test.tsx | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/packages/cli/src/ui/components/ShowMoreLinesLayout.test.tsx b/packages/cli/src/ui/components/ShowMoreLinesLayout.test.tsx
index 78fd94a5e5..a3ac6f3440 100644
--- a/packages/cli/src/ui/components/ShowMoreLinesLayout.test.tsx
+++ b/packages/cli/src/ui/components/ShowMoreLinesLayout.test.tsx
@@ -9,6 +9,8 @@ import { render } from '../../test-utils/render.js';
 import { ShowMoreLines } from './ShowMoreLines.js';
 import { useOverflowState } from '../contexts/OverflowContext.js';
 import { useAlternateBuffer } from '../hooks/useAlternateBuffer.js';
+import { StreamingContext } from '../contexts/StreamingContext.js';
+import { StreamingState } from '../types.js';
 
 import type React from 'react';
 
@@ -42,7 +44,9 @@ describe('ShowMoreLines layout and padding', () => {
     const TestComponent = () => (
       <Box flexDirection="column">
         <Text>Top</Text>
-        <ShowMoreLines constrainHeight={true} isOverflowing={true} />
+        <StreamingContext.Provider value={StreamingState.Idle}>
+          <ShowMoreLines constrainHeight={true} isOverflowing={true} />
+        </StreamingContext.Provider>
         <Text>Bottom</Text>
       </Box>
     );
@@ -74,7 +78,9 @@ describe('ShowMoreLines layout and padding', () => {
     const TestComponent = () => (
       <Box flexDirection="column">
         <Text>Top</Text>
-        <ShowMoreLines constrainHeight={true} isOverflowing={true} />
+        <StreamingContext.Provider value={StreamingState.Idle}>
+          <ShowMoreLines constrainHeight={true} isOverflowing={true} />
+        </StreamingContext.Provider>
         <Text>Bottom</Text>
       </Box>
     );