@machina.ai/cell-cli 1.36.0-rc1 → 1.38.1-rc2

package/dist/src/ui/components/InputPrompt.test.js

@@ -1,17 +1,18 @@
-import { jsx as _jsx } from "react/jsx-runtime";
+import { jsx as _jsx, Fragment as _Fragment, jsxs as _jsxs } from "react/jsx-runtime";
 /**
  * @license
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { renderWithProviders } from '../../test-utils/render.js';
+import { renderWithProviders, cleanup } from '../../test-utils/render.js';
 import { createMockSettings } from '../../test-utils/settings.js';
 import { makeFakeConfig } from '@google/gemini-cli-core';
 import { waitFor } from '../../test-utils/async.js';
-import { act, useState } from 'react';
+import { act, useState, useMemo } from 'react';
 import { InputPrompt, tryTogglePasteExpansion, } from './InputPrompt.js';
+import { InputContext } from '../contexts/InputContext.js';
 import { calculateTransformationsForLine, calculateTransformedLine, } from './shared/text-buffer.js';
-import { ApprovalMode, debugLogger, } from '@google/gemini-cli-core';
+import { ApprovalMode, debugLogger, coreEvents, } from '@google/gemini-cli-core';
 import * as path from 'node:path';
 import { CommandKind, } from '../commands/types.js';
 import { describe, it, expect, beforeEach, vi, afterEach } from 'vitest';
@@ -31,7 +32,9 @@ import { terminalCapabilityManager } from '../utils/terminalCapabilityManager.js
 import { isLowColorDepth } from '../utils/terminalUtils.js';
 import { cpLen } from '../utils/textUtils.js';
 import { defaultKeyMatchers, Command } from '../key/keyMatchers.js';
+import { useKeypress } from '../hooks/useKeypress.js';
 import { appEvents, AppEvent, TransientMessageType, } from '../../utils/events.js';
+import '../../test-utils/customMatchers.js';
 vi.mock('../hooks/useShellHistory.js');
 vi.mock('../hooks/useCommandCompletion.js');
 vi.mock('../hooks/useInputHistory.js');
@@ -52,6 +55,8 @@ vi.mock('ink', async (importOriginal) => {
 });
 afterEach(() => {
     vi.restoreAllMocks();
+    vi.useRealTimers();
+    cleanup();
 });
 const mockSlashCommands = [
     {
@@ -112,6 +117,26 @@ const mockSlashCommands = [
         action: vi.fn(),
     },
 ];
+const TestInputPrompt = (props) => {
+    const contextValue = useMemo(() => ({
+        buffer: props.buffer,
+        userMessages: props.userMessages,
+        shellModeActive: props.shellModeActive,
+        copyModeEnabled: props.copyModeEnabled,
+        showEscapePrompt: props.showEscapePrompt || false,
+        inputWidth: props.inputWidth,
+        suggestionsWidth: props.suggestionsWidth,
+    }), [
+        props.buffer,
+        props.userMessages,
+        props.shellModeActive,
+        props.copyModeEnabled,
+        props.showEscapePrompt,
+        props.inputWidth,
+        props.suggestionsWidth,
+    ]);
+    return (_jsx(InputContext.Provider, { value: contextValue, children: _jsx(InputPrompt, { ...props }) }));
+};
 describe('InputPrompt', () => {
     let props;
     let mockShellHistory;
@@ -120,6 +145,15 @@ describe('InputPrompt', () => {
     let mockReverseSearchCompletion;
     let mockBuffer;
     let mockCommandContext;
+    const GlobalEscapeHandler = ({ onEscape }) => {
+        useKeypress((key) => {
+            if (key.name !== 'escape')
+                return false;
+            onEscape();
+            return true;
+        }, { isActive: true, priority: false });
+        return null;
+    };
     const mockedUseShellHistory = vi.mocked(useShellHistory);
     const mockedUseCommandCompletion = vi.mocked(useCommandCompletion);
     const mockedUseInputHistory = vi.mocked(useInputHistory);
@@ -134,9 +168,11 @@ describe('InputPrompt', () => {
         setCleanUiDetailsVisible: mockSetCleanUiDetailsVisible,
         toggleCleanUiDetailsVisible: mockToggleCleanUiDetailsVisible,
         revealCleanUiDetailsTemporarily: mockRevealCleanUiDetailsTemporarily,
+        addMessage: vi.fn(),
     };
     beforeEach(() => {
         vi.resetAllMocks();
+        coreEvents.removeAllListeners();
         vi.spyOn(terminalCapabilityManager, 'isKittyProtocolEnabled').mockReturnValue(true);
         mockCommandContext = createMockCommandContext();
         mockBuffer = {
@@ -145,7 +181,7 @@ describe('InputPrompt', () => {
             lines: [''],
             setText: vi.fn((newText, cursorPosition) => {
                 mockBuffer.text = newText;
-                mockBuffer.lines = [newText];
+                mockBuffer.lines = newText.split('\n');
                 let col = 0;
                 if (typeof cursorPosition === 'number') {
                     col = cursorPosition;
@@ -157,16 +193,25 @@ describe('InputPrompt', () => {
                     col = newText.length;
                 }
                 mockBuffer.cursor = [0, col];
-                mockBuffer.viewportVisualLines = [newText];
-                mockBuffer.allVisualLines = [newText];
-                mockBuffer.visualToLogicalMap = [[0, 0]];
+                mockBuffer.allVisualLines = newText.split('\n');
+                mockBuffer.viewportVisualLines = newText.split('\n');
+                mockBuffer.visualToLogicalMap = newText
+                    .split('\n')
+                    .map((_, i) => [i, 0]);
                 mockBuffer.visualCursor = [0, col];
+                mockBuffer.visualScrollRow = 0;
+                mockBuffer.viewportHeight = 10;
+                mockBuffer.visualToTransformedMap = newText
+                    .split('\n')
+                    .map((_, i) => i);
+                mockBuffer.transformationsByLine = newText.split('\n').map(() => []);
             }),
             replaceRangeByOffset: vi.fn(),
             viewportVisualLines: [''],
             allVisualLines: [''],
             visualCursor: [0, 0],
             visualScrollRow: 0,
+            viewportHeight: 10,
             handleInput: vi.fn((key) => {
                 if (defaultKeyMatchers[Command.CLEAR_INPUT](key)) {
                     if (mockBuffer.text.length > 0) {
@@ -280,6 +325,7 @@ describe('InputPrompt', () => {
         });
         vi.mocked(clipboardy.read).mockResolvedValue('');
         props = {
+            onQueueMessage: vi.fn(),
             buffer: mockBuffer,
             onSubmit: vi.fn(),
             userMessages: [],
@@ -289,6 +335,7 @@ describe('InputPrompt', () => {
                 getTargetDir: () => path.join('test', 'project', 'src'),
                 getVimMode: () => false,
                 getUseBackgroundColor: () => true,
+                getUseTerminalBuffer: () => false,
                 getTerminalBackground: () => undefined,
                 getWorkspaceContext: () => ({
                     getDirectories: () => ['/test/project/src'],
@@ -309,7 +356,7 @@ describe('InputPrompt', () => {
     });
     it('should call shellHistory.getPreviousCommand on up arrow in shell mode', async () => {
         props.shellModeActive = true;
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -320,7 +367,7 @@ describe('InputPrompt', () => {
     });
     it('should call shellHistory.getNextCommand on down arrow in shell mode', async () => {
         props.shellModeActive = true;
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -332,7 +379,7 @@ describe('InputPrompt', () => {
     it('should set the buffer text when a shell history command is retrieved', async () => {
         props.shellModeActive = true;
         vi.mocked(mockShellHistory.getPreviousCommand).mockReturnValue('previous command');
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -347,7 +394,7 @@ describe('InputPrompt', () => {
     it('should call shellHistory.addCommandToHistory on submit in shell mode', async () => {
         props.shellModeActive = true;
         props.buffer.setText('ls -l');
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -371,7 +418,7 @@ describe('InputPrompt', () => {
             ],
             activeSuggestionIndex: 0,
         });
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         // Press Enter without navigating — should dismiss suggestions and fall
@@ -398,7 +445,7 @@ describe('InputPrompt', () => {
             ],
             activeSuggestionIndex: 1,
         });
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         // Press ArrowDown to navigate, then Enter to accept
@@ -417,7 +464,7 @@ describe('InputPrompt', () => {
     });
     it('should NOT call shell history methods when not in shell mode', async () => {
         props.buffer.setText('some text');
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -442,7 +489,7 @@ describe('InputPrompt', () => {
             mockBuffer.allVisualLines = ['line 1', 'line 2'];
             mockBuffer.visualCursor = [0, 5]; // First line, not at start
             mockBuffer.visualScrollRow = 0;
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiActions,
             });
             await act(async () => {
@@ -458,7 +505,7 @@ describe('InputPrompt', () => {
             mockBuffer.allVisualLines = ['line 1', 'line 2'];
             mockBuffer.visualCursor = [0, 0]; // First line, at start
             mockBuffer.visualScrollRow = 0;
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiActions,
             });
             await act(async () => {
@@ -474,7 +521,7 @@ describe('InputPrompt', () => {
             mockBuffer.allVisualLines = ['line 1', 'line 2'];
             mockBuffer.visualCursor = [1, 0]; // Last line, not at end
             mockBuffer.visualScrollRow = 0;
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiActions,
             });
             await act(async () => {
@@ -490,7 +537,7 @@ describe('InputPrompt', () => {
             mockBuffer.allVisualLines = ['line 1', 'line 2'];
             mockBuffer.visualCursor = [1, 6]; // Last line, at end ("line 2" is length 6)
             mockBuffer.visualScrollRow = 0;
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiActions,
             });
             await act(async () => {
@@ -513,7 +560,7 @@ describe('InputPrompt', () => {
             ],
         });
         props.buffer.setText('/mem');
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         // Test up arrow
@@ -538,7 +585,7 @@ describe('InputPrompt', () => {
             ],
         });
         props.buffer.setText('/mem');
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         // Test down arrow
@@ -559,7 +606,7 @@ describe('InputPrompt', () => {
             showSuggestions: false,
         });
         props.buffer.setText('some text');
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -584,7 +631,7 @@ describe('InputPrompt', () => {
     });
     it('should clear the buffer and reset completion on Ctrl+C', async () => {
         mockBuffer.text = 'some text';
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -605,7 +652,7 @@ describe('InputPrompt', () => {
         it('should handle Ctrl+V when clipboard has an image', async () => {
             vi.mocked(clipboardUtils.clipboardHasImage).mockResolvedValue(true);
             vi.mocked(clipboardUtils.saveClipboardImage).mockResolvedValue('/test/.cell-cli-clipboard/clipboard-123.png');
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             // Send Ctrl+V
             await act(async () => {
                 stdin.write('\x16'); // Ctrl+V
@@ -620,7 +667,7 @@ describe('InputPrompt', () => {
         });
         it('should not insert anything when clipboard has no image', async () => {
             vi.mocked(clipboardUtils.clipboardHasImage).mockResolvedValue(false);
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x16'); // Ctrl+V
             });
@@ -634,7 +681,7 @@ describe('InputPrompt', () => {
         it('should handle image save failure gracefully', async () => {
             vi.mocked(clipboardUtils.clipboardHasImage).mockResolvedValue(true);
             vi.mocked(clipboardUtils.saveClipboardImage).mockResolvedValue(null);
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x16'); // Ctrl+V
             });
@@ -654,7 +701,7 @@ describe('InputPrompt', () => {
             vi.mocked(mockBuffer.getOffset).mockReturnValue(5);
             mockBuffer.lines = ['Hello world'];
             mockBuffer.replaceRangeByOffset = vi.fn();
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x16'); // Ctrl+V
             });
@@ -675,7 +722,7 @@ describe('InputPrompt', () => {
                 .spyOn(debugLogger, 'error')
                 .mockImplementation(() => { });
             vi.mocked(clipboardUtils.clipboardHasImage).mockRejectedValue(new Error('Clipboard error'));
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x16'); // Ctrl+V
             });
@@ -692,7 +739,7 @@ describe('InputPrompt', () => {
             vi.mocked(clipboardUtils.clipboardHasImage).mockResolvedValue(false);
             vi.mocked(clipboardy.read).mockResolvedValue('pasted text');
             vi.mocked(mockBuffer.replaceRangeByOffset).mockClear();
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x16'); // Ctrl+V
             });
@@ -707,7 +754,7 @@ describe('InputPrompt', () => {
             const settings = createMockSettings({
                 experimental: { useOSC52Paste: true },
             });
-            const { stdout, stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), { settings });
+            const { stdout, stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), { settings });
             const writeSpy = vi.spyOn(stdout, 'write');
             await act(async () => {
                 stdin.write('\x16'); // Ctrl+V
@@ -759,7 +806,7 @@ describe('InputPrompt', () => {
             activeSuggestionIndex: activeIndex,
         });
         props.buffer.setText(bufferText);
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => stdin.write('\t'));
@@ -774,7 +821,7 @@ describe('InputPrompt', () => {
             activeSuggestionIndex: 0,
         });
         props.buffer.setText('/mem');
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -803,7 +850,7 @@ describe('InputPrompt', () => {
             activeSuggestionIndex: 0,
         });
         props.buffer.setText('/?');
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -812,9 +859,55 @@ describe('InputPrompt', () => {
         await waitFor(() => expect(mockCommandCompletion.handleAutocomplete).toHaveBeenCalledWith(0));
         unmount();
     });
+    it('queues a message when Tab is pressed during generation', async () => {
+        props.buffer.setText('A new prompt');
+        props.streamingState = StreamingState.Responding;
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
+            uiActions,
+        });
+        await act(async () => {
+            stdin.write('\t');
+        });
+        await waitFor(() => {
+            expect(props.onQueueMessage).toHaveBeenCalledWith('A new prompt');
+            expect(props.buffer.text).toBe('');
+        });
+        unmount();
+    });
+    it('shows an error when attempting to queue a slash command', async () => {
+        props.buffer.setText('/clear');
+        props.streamingState = StreamingState.Responding;
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
+            uiActions,
+        });
+        await act(async () => {
+            stdin.write('\t');
+        });
+        await waitFor(() => {
+            expect(props.setQueueErrorMessage).toHaveBeenCalledWith('Slash commands cannot be queued');
+            expect(props.onQueueMessage).not.toHaveBeenCalled();
+        });
+        unmount();
+    });
+    it('shows an error when attempting to queue a shell command', async () => {
+        props.shellModeActive = true;
+        props.buffer.setText('ls');
+        props.streamingState = StreamingState.Responding;
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
+            uiActions,
+        });
+        await act(async () => {
+            stdin.write('\t');
+        });
+        await waitFor(() => {
+            expect(props.setQueueErrorMessage).toHaveBeenCalledWith('Shell commands cannot be queued');
+            expect(props.onQueueMessage).not.toHaveBeenCalled();
+        });
+        unmount();
+    });
     it('should not submit on Enter when the buffer is empty or only contains whitespace', async () => {
         props.buffer.setText('   '); // Set buffer to whitespace
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -832,7 +925,7 @@ describe('InputPrompt', () => {
             isPerfectMatch: true,
         });
         props.buffer.setText('/clear');
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -853,7 +946,7 @@ describe('InputPrompt', () => {
             isPerfectMatch: true,
         });
         props.buffer.text = '/review';
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -876,7 +969,7 @@ describe('InputPrompt', () => {
             isPerfectMatch: true, // /review is a perfect match
         });
         props.buffer.text = '/review';
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -897,7 +990,7 @@ describe('InputPrompt', () => {
             isPerfectMatch: false, // Added explicit isPerfectMatch false
         });
         props.buffer.setText('/clear');
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -916,7 +1009,7 @@ describe('InputPrompt', () => {
             completionMode: CompletionMode.AT,
         });
         props.buffer.text = '@file.txt';
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -938,7 +1031,7 @@ describe('InputPrompt', () => {
             completionMode: CompletionMode.AT,
         });
         props.buffer.text = '@file.txt';
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -978,7 +1071,7 @@ describe('InputPrompt', () => {
         props.buffer.setText('/ab');
         props.buffer.lines = ['/ab'];
         props.buffer.cursor = [0, 3];
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -1012,7 +1105,7 @@ describe('InputPrompt', () => {
         props.buffer.setText('/sh');
         props.buffer.lines = ['/sh'];
         props.buffer.cursor = [0, 3];
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -1045,7 +1138,7 @@ describe('InputPrompt', () => {
         props.buffer.setText('/ab');
         props.buffer.lines = ['/ab'];
         props.buffer.cursor = [0, 3];
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -1070,7 +1163,7 @@ describe('InputPrompt', () => {
         props.buffer.setText('/ab');
         props.buffer.lines = ['/ab'];
         props.buffer.cursor = [0, 3];
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -1101,7 +1194,7 @@ describe('InputPrompt', () => {
         props.buffer.setText('/find');
         props.buffer.lines = ['/find'];
         props.buffer.cursor = [0, 5];
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -1143,7 +1236,7 @@ describe('InputPrompt', () => {
         props.buffer.setText('/mcp auth ');
         props.buffer.lines = ['/mcp auth '];
         props.buffer.cursor = [0, 10];
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -1190,7 +1283,7 @@ describe('InputPrompt', () => {
         props.buffer.setText('/extensions enable ');
         props.buffer.lines = ['/extensions enable '];
         props.buffer.cursor = [0, 19];
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -1232,7 +1325,7 @@ describe('InputPrompt', () => {
         props.buffer.setText('/chat resu');
         props.buffer.lines = ['/chat resu'];
         props.buffer.cursor = [0, 10];
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -1253,7 +1346,7 @@ describe('InputPrompt', () => {
             activeSuggestionIndex: 0,
         });
         props.buffer.setText('@src/components/');
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -1268,7 +1361,7 @@ describe('InputPrompt', () => {
         mockBuffer.text = 'first line\\';
         mockBuffer.cursor = [0, 11];
         mockBuffer.lines = ['first line\\'];
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -1285,7 +1378,7 @@ describe('InputPrompt', () => {
         await act(async () => {
             props.buffer.setText('some text to clear');
         });
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -1300,7 +1393,7 @@ describe('InputPrompt', () => {
     });
     it('should render correctly in plan mode', async () => {
         props.approvalMode = ApprovalMode.PLAN;
-        const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
         await waitFor(() => {
             const frame = stdout.lastFrameRaw();
             // In plan mode it uses '>' but with success color.
@@ -1313,7 +1406,7 @@ describe('InputPrompt', () => {
     });
     it('should NOT clear the buffer on Ctrl+C if it is empty', async () => {
         props.buffer.text = '';
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -1325,7 +1418,7 @@ describe('InputPrompt', () => {
         unmount();
     });
     it('should call setBannerVisible(false) when clear screen key is pressed', async () => {
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             uiActions,
         });
         await act(async () => {
@@ -1344,7 +1437,7 @@ describe('InputPrompt', () => {
             vi.restoreAllMocks();
         });
         it('should render with background color by default', async () => {
-            const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await waitFor(() => {
                 const frame = stdout.lastFrameRaw();
                 expect(frame).toContain('▀');
@@ -1361,7 +1454,7 @@ describe('InputPrompt', () => {
             { color: '#fff', name: '#fff' },
         ])('should render with safe grey background but NO side borders in 8-bit mode when background is $name', async ({ color }) => {
             vi.mocked(isLowColorDepth).mockReturnValue(true);
-            const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiState: {
                     terminalBackgroundColor: color,
                 },
@@ -1385,7 +1478,7 @@ describe('InputPrompt', () => {
         });
         it('should NOT render with background color but SHOULD render horizontal lines when color depth is < 24 and background is NOT black', async () => {
             vi.mocked(isLowColorDepth).mockReturnValue(true);
-            const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiState: {
                     terminalBackgroundColor: '#333333',
                 },
@@ -1403,7 +1496,7 @@ describe('InputPrompt', () => {
         });
         it('should handle 4-bit color mode (16 colors) as low color depth', async () => {
             vi.mocked(isLowColorDepth).mockReturnValue(true);
-            const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiState: {
                     terminalBackgroundColor: 'black',
                 },
@@ -1417,7 +1510,7 @@ describe('InputPrompt', () => {
         });
         it('should render horizontal lines (but NO background) in 8-bit mode when background is blue', async () => {
             vi.mocked(isLowColorDepth).mockReturnValue(true);
-            const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiState: {
                     terminalBackgroundColor: 'blue',
                 },
@@ -1437,7 +1530,7 @@ describe('InputPrompt', () => {
         });
         it('should render with plain borders when useBackgroundColor is false', async () => {
             props.config.getUseBackgroundColor = () => false;
-            const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await waitFor(() => {
                 const frame = stdout.lastFrameRaw();
                 expect(frame).not.toContain('▀');
@@ -1539,7 +1632,7 @@ describe('InputPrompt', () => {
                     ? [{ label: 'suggestion', value: 'suggestion' }]
                     : [],
             });
-            const { unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiActions,
             });
             await waitFor(() => {
@@ -1576,7 +1669,7 @@ describe('InputPrompt', () => {
             },
         ])('$name', async ({ vimHandled, expectBufferHandleInput }) => {
             props.vimHandleInput = vi.fn().mockReturnValue(vimHandled);
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => stdin.write('i'));
             await waitFor(() => {
                 expect(props.vimHandleInput).toHaveBeenCalled();
@@ -1593,7 +1686,7 @@ describe('InputPrompt', () => {
     describe('unfocused paste', () => {
         it('should handle bracketed paste when not focused', async () => {
             props.focus = false;
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x1B[200~pasted text\x1B[201~');
             });
@@ -1607,7 +1700,7 @@ describe('InputPrompt', () => {
         });
         it('should ignore regular keypresses when not focused', async () => {
             props.focus = false;
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('a');
             });
@@ -1672,10 +1765,11 @@ describe('InputPrompt', () => {
             ])('should display cursor correctly $name', async ({ text, visualCursor }) => {
                 mockBuffer.text = text;
                 mockBuffer.lines = [text];
+                mockBuffer.allVisualLines = [text];
                 mockBuffer.viewportVisualLines = [text];
                 mockBuffer.visualCursor = visualCursor;
                 props.config.getUseBackgroundColor = () => false;
-                const renderResult = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+                const renderResult = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
                 await renderResult.waitUntilReady();
                 await expect(renderResult).toMatchSvgSnapshot();
                 renderResult.unmount();
@@ -1714,11 +1808,12 @@ describe('InputPrompt', () => {
             ])('should display cursor correctly $name in a multiline block', async ({ text, visualCursor, visualToLogicalMap }) => {
                 mockBuffer.text = text;
                 mockBuffer.lines = text.split('\n');
+                mockBuffer.allVisualLines = text.split('\n');
                 mockBuffer.viewportVisualLines = text.split('\n');
                 mockBuffer.visualCursor = visualCursor;
                 mockBuffer.visualToLogicalMap = visualToLogicalMap;
                 props.config.getUseBackgroundColor = () => false;
-                const renderResult = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+                const renderResult = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
                 await renderResult.waitUntilReady();
                 await expect(renderResult).toMatchSvgSnapshot();
                 renderResult.unmount();
@@ -1727,6 +1822,7 @@ describe('InputPrompt', () => {
                 const text = 'first line\n\nthird line';
                 mockBuffer.text = text;
                 mockBuffer.lines = text.split('\n');
+                mockBuffer.allVisualLines = text.split('\n');
                 mockBuffer.viewportVisualLines = text.split('\n');
                 mockBuffer.visualCursor = [1, 0]; // cursor on the blank line
                 mockBuffer.visualToLogicalMap = [
@@ -1735,18 +1831,102 @@ describe('InputPrompt', () => {
                     [2, 0],
                 ];
                 props.config.getUseBackgroundColor = () => false;
-                const renderResult = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+                const renderResult = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
                 await renderResult.waitUntilReady();
                 await expect(renderResult).toMatchSvgSnapshot();
                 renderResult.unmount();
             });
         });
     });
+    describe('scrolling large inputs', () => {
+        it('should correctly render scrolling down and up for large inputs', async () => {
+            const lines = Array.from({ length: 50 }).map((_, i) => `testline ${i}`);
+            // Since we need to test how the React component tree responds to TextBuffer state changes,
+            // we must provide a fake TextBuffer implementation that triggers re-renders like the real one.
+            const TestWrapper = () => {
+                const [bufferState, setBufferState] = useState({
+                    text: lines.join('\n'),
+                    lines,
+                    allVisualLines: lines,
+                    viewportVisualLines: lines.slice(0, 10),
+                    visualToLogicalMap: lines.map((_, i) => [i, 0]),
+                    visualCursor: [0, 0],
+                    visualScrollRow: 0,
+                    viewportHeight: 10,
+                });
+                const fakeBuffer = {
+                    ...mockBuffer,
+                    ...bufferState,
+                    handleInput: vi.fn().mockImplementation((key) => {
+                        let newRow = bufferState.visualCursor[0];
+                        let newScroll = bufferState.visualScrollRow;
+                        if (key.name === 'down') {
+                            newRow = Math.min(49, newRow + 1);
+                            if (newRow >= newScroll + 10)
+                                newScroll++;
+                        }
+                        else if (key.name === 'up') {
+                            newRow = Math.max(0, newRow - 1);
+                            if (newRow < newScroll)
+                                newScroll--;
+                        }
+                        setBufferState({
+                            ...bufferState,
+                            visualCursor: [newRow, 0],
+                            visualScrollRow: newScroll,
+                            viewportVisualLines: lines.slice(newScroll, newScroll + 10),
+                        });
+                        return true;
+                    }),
+                };
+                const inputState = {
+                    buffer: fakeBuffer,
+                    userMessages: [],
+                    shellModeActive: false,
+                    showEscapePrompt: false,
+                    copyModeEnabled: false,
+                    inputWidth: 80,
+                    suggestionsWidth: 80,
+                };
+                return (_jsx(InputContext.Provider, { value: inputState, children: _jsx(InputPrompt, { ...props }) }));
+            };
+            const { stdout, unmount, stdin } = await renderWithProviders(_jsx(TestWrapper, {}), {
+                uiActions,
+            });
+            // Verify initial render
+            await waitFor(() => {
+                expect(stdout.lastFrame()).toContain('testline 0');
+                expect(stdout.lastFrame()).not.toContain('testline 49');
+            });
+            // Move cursor to bottom
+            for (let i = 0; i < 49; i++) {
+                act(() => {
+                    stdin.write('\x1b[B'); // Arrow Down
+                });
+            }
+            await waitFor(() => {
+                expect(stdout.lastFrame()).toContain('testline 49');
+                expect(stdout.lastFrame()).not.toContain('testline 0');
+            });
+            // Move cursor back to top
+            for (let i = 0; i < 49; i++) {
+                act(() => {
+                    stdin.write('\x1b[A'); // Arrow Up
+                });
+            }
+            await waitFor(() => {
+                expect(stdout.lastFrame()).toContain('testline 0');
+                expect(stdout.lastFrame()).not.toContain('testline 49');
+            });
+            unmount();
+        });
+    });
     describe('multiline rendering', () => {
         it('should correctly render multiline input including blank lines', async () => {
             const text = 'hello\n\nworld';
             mockBuffer.text = text;
             mockBuffer.lines = text.split('\n');
+            mockBuffer.allVisualLines = text.split('\n');
             mockBuffer.viewportVisualLines = text.split('\n');
             mockBuffer.allVisualLines = text.split('\n');
             mockBuffer.visualCursor = [2, 5]; // cursor at the end of "world"
@@ -1757,7 +1937,7 @@ describe('InputPrompt', () => {
                 [2, 0],
             ];
             props.config.getUseBackgroundColor = () => false;
-            const renderResult = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const renderResult = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await renderResult.waitUntilReady();
             await expect(renderResult).toMatchSvgSnapshot();
             renderResult.unmount();
@@ -1778,7 +1958,7 @@ describe('InputPrompt', () => {
                 pastedText: 'This\r\nis\r\na\r\nmultiline\r\npaste.',
             },
         ])('should handle multiline paste $description', async ({ pastedText }) => {
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             // Simulate a bracketed paste event from the terminal
             await act(async () => {
                 stdin.write(`\x1b[200~${pastedText}\x1b[201~`);
@@ -1799,7 +1979,7 @@ describe('InputPrompt', () => {
             vi.mocked(clipboardUtils.clipboardHasImage).mockResolvedValue(false);
             const largeText = '1\n2\n3\n4\n5\n6';
             vi.mocked(clipboardy.read).mockResolvedValue(largeText);
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x16'); // Ctrl+V
             });
@@ -1812,7 +1992,7 @@ describe('InputPrompt', () => {
             vi.mocked(clipboardUtils.clipboardHasImage).mockResolvedValue(false);
             const largeText = 'a'.repeat(501);
             vi.mocked(clipboardy.read).mockResolvedValue(largeText);
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x16'); // Ctrl+V
             });
@@ -1825,7 +2005,7 @@ describe('InputPrompt', () => {
             vi.mocked(clipboardUtils.clipboardHasImage).mockResolvedValue(false);
             const smallText = 'hello world';
             vi.mocked(clipboardy.read).mockResolvedValue(smallText);
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x16'); // Ctrl+V
             });
@@ -1840,7 +2020,7 @@ describe('InputPrompt', () => {
             const id = '[Pasted Text: 6 lines]';
             mockBuffer.text = `Check this: ${id}`;
             mockBuffer.pastedContent = { [id]: largeText };
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\r'); // Enter
             });
@@ -1865,7 +2045,7 @@ describe('InputPrompt', () => {
         it('should prevent auto-submission immediately after an unsafe paste', async () => {
             // isTerminalPasteTrusted will be false due to beforeEach setup.
             props.buffer.text = 'some command';
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 await vi.runAllTimersAsync();
             });
@@ -1894,7 +2074,7 @@ describe('InputPrompt', () => {
                 completionMode: CompletionMode.AT,
             });
             props.buffer.text = '@file.txt';
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             // Simulate an unsafe paste of a perfect match
             await act(async () => {
                 stdin.write(`\x1b[200~@file.txt\x1b[201~`);
@@ -1912,7 +2092,7 @@ describe('InputPrompt', () => {
         it('should allow submission after unsafe paste protection timeout', async () => {
             // isTerminalPasteTrusted will be false due to beforeEach setup.
             props.buffer.text = 'pasted text';
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 await vi.runAllTimersAsync();
             });
@@ -1949,7 +2129,7 @@ describe('InputPrompt', () => {
         ])('should allow immediate submission for a trusted paste ($name)', async ({ setup }) => {
             setup();
             props.buffer.text = 'pasted command';
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 await vi.runAllTimersAsync();
             });
@@ -1974,7 +2154,7 @@ describe('InputPrompt', () => {
         it('should not interfere with normal Enter key submission when no recent paste', async () => {
             // Set up buffer with text before rendering to ensure submission works
             props.buffer.text = 'normal command';
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 await vi.runAllTimersAsync();
             });
@@ -1997,7 +2177,7 @@ describe('InputPrompt', () => {
             const onEscapePromptChange = vi.fn();
             props.onEscapePromptChange = onEscapePromptChange;
             props.buffer.setText('text to clear');
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x03');
                 vi.advanceTimersByTime(100);
@@ -2011,7 +2191,7 @@ describe('InputPrompt', () => {
             props.onEscapePromptChange = onEscapePromptChange;
             props.buffer.setText('');
             vi.mocked(props.buffer.setText).mockClear();
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiState: {
                     history: [{ id: 1, type: 'user', text: 'test' }],
                 },
@@ -2030,7 +2210,7 @@ describe('InputPrompt', () => {
             props.onEscapePromptChange = onEscapePromptChange;
             props.buffer.setText('some text');
             vi.mocked(props.buffer.setText).mockClear();
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x1B\x1B');
                 vi.advanceTimersByTime(100);
@@ -2043,7 +2223,7 @@ describe('InputPrompt', () => {
             const onEscapePromptChange = vi.fn();
             props.onEscapePromptChange = onEscapePromptChange;
             props.buffer.setText('some text');
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x1B');
                 await waitFor(() => {
@@ -2060,7 +2240,7 @@ describe('InputPrompt', () => {
         });
         it('should handle ESC in shell mode by disabling shell mode', async () => {
             props.shellModeActive = true;
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x1B');
                 vi.advanceTimersByTime(100);
@@ -2068,13 +2248,43 @@ describe('InputPrompt', () => {
             });
             unmount();
         });
+        it('should not propagate ESC to global cancellation handler when shell mode is active (responding)', async () => {
+            props.shellModeActive = true;
+            props.streamingState = StreamingState.Responding;
+            const onGlobalEscape = vi.fn();
+            const { stdin, unmount } = await renderWithProviders(_jsxs(_Fragment, { children: [_jsx(GlobalEscapeHandler, { onEscape: onGlobalEscape }), _jsx(TestInputPrompt, { ...props })] }));
+            await act(async () => {
+                stdin.write('\x1B');
+                vi.advanceTimersByTime(100);
+            });
+            await waitFor(() => {
+                expect(props.setShellModeActive).toHaveBeenCalledWith(false);
+            });
+            expect(onGlobalEscape).not.toHaveBeenCalled();
+            unmount();
+        });
+        it('should allow ESC to reach global cancellation handler when responding and no overlay is active', async () => {
+            props.shellModeActive = false;
+            props.streamingState = StreamingState.Responding;
+            const onGlobalEscape = vi.fn();
+            const { stdin, unmount } = await renderWithProviders(_jsxs(_Fragment, { children: [_jsx(GlobalEscapeHandler, { onEscape: onGlobalEscape }), _jsx(TestInputPrompt, { ...props })] }));
+            await act(async () => {
+                stdin.write('\x1B');
+                vi.advanceTimersByTime(100);
+            });
+            await waitFor(() => {
+                expect(onGlobalEscape).toHaveBeenCalledTimes(1);
+            });
+            expect(props.setShellModeActive).not.toHaveBeenCalled();
+            unmount();
+        });
         it('should handle ESC when completion suggestions are showing', async () => {
             mockedUseCommandCompletion.mockReturnValue({
                 ...mockCommandCompletion,
                 showSuggestions: true,
                 suggestions: [{ label: 'suggestion', value: 'suggestion' }],
             });
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x1B');
                 vi.advanceTimersByTime(100);
@@ -2085,7 +2295,7 @@ describe('InputPrompt', () => {
         it('should not call onEscapePromptChange when not provided', async () => {
             props.onEscapePromptChange = undefined;
             props.buffer.setText('some text');
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 await vi.runAllTimersAsync();
             });
@@ -2098,7 +2308,7 @@ describe('InputPrompt', () => {
             unmount();
         });
         it('should not interfere with existing keyboard shortcuts', async () => {
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x0C');
             });
@@ -2133,7 +2343,7 @@ describe('InputPrompt', () => {
                 showSuggestions: true,
                 activeSuggestionIndex: 0,
             });
-            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             // Trigger reverse search with Ctrl+R
             await act(async () => {
                 stdin.write('\x12');
@@ -2151,7 +2361,7 @@ describe('InputPrompt', () => {
             { name: 'standard', escapeSequence: '\x1B' },
             { name: 'kitty', escapeSequence: '\u001b[27u' },
         ])('resets reverse search state on Escape ($name)', async ({ escapeSequence }) => {
-            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x12');
             });
@@ -2186,7 +2396,7 @@ describe('InputPrompt', () => {
                 activeSuggestionIndex: reverseSearchActive ? 0 : -1,
                 handleAutocomplete: mockHandleAutocomplete,
             }));
-            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             // Enter reverse search mode with Ctrl+R
             await act(async () => {
                 stdin.write('\x12');
@@ -2214,7 +2424,7 @@ describe('InputPrompt', () => {
                 activeSuggestionIndex: 0,
                 handleAutocomplete: mockHandleAutocomplete,
             });
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiActions,
             });
             await act(async () => {
@@ -2239,7 +2449,7 @@ describe('InputPrompt', () => {
                 showSuggestions: true,
                 activeSuggestionIndex: 0,
             });
-            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x12');
             });
@@ -2268,7 +2478,7 @@ describe('InputPrompt', () => {
                     : [],
                 showSuggestions: reverseSearchActiveFromInputPrompt,
             }));
-            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             // reverse search with Ctrl+R
             await act(async () => {
                 stdin.write('\x12');
@@ -2293,7 +2503,7 @@ describe('InputPrompt', () => {
             props.buffer.text = 'line 1\nline 2\nline 3';
             props.buffer.cursor = [1, 2];
             props.buffer.lines = ['line 1', 'line 2', 'line 3'];
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x05'); // Ctrl+E
             });
@@ -2307,7 +2517,7 @@ describe('InputPrompt', () => {
             props.buffer.text = 'single line text';
             props.buffer.cursor = [0, 5];
             props.buffer.lines = ['single line text'];
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x05'); // Ctrl+E
             });
@@ -2332,7 +2542,7 @@ describe('InputPrompt', () => {
                 showSuggestions: !!isActive,
                 activeSuggestionIndex: isActive ? 0 : -1,
             }));
-            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x12'); // Ctrl+R
             });
@@ -2355,7 +2565,7 @@ describe('InputPrompt', () => {
                 visibleStartIndex: 0,
                 isLoadingSuggestions: false,
             });
-            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x12');
             });
@@ -2391,7 +2601,7 @@ describe('InputPrompt', () => {
                 visibleStartIndex: 0,
                 isLoadingSuggestions: false,
             });
-            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x12');
             });
@@ -2421,7 +2631,7 @@ describe('InputPrompt', () => {
                 visibleStartIndex: 0,
                 isLoadingSuggestions: false,
             });
-            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\x12');
             });
@@ -2437,7 +2647,7 @@ describe('InputPrompt', () => {
         it('ensures Ctrl+R search results are prioritized newest-to-oldest by reversing userMessages', async () => {
             props.shellModeActive = false;
             props.userMessages = ['oldest', 'middle', 'newest'];
-            await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             const calls = vi.mocked(useReverseSearchCompletion).mock.calls;
             const commandSearchCall = calls.find((call) => call[1] === props.userMessages ||
                 (Array.isArray(call[1]) && call[1][0] === 'newest'));
@@ -2484,7 +2694,7 @@ describe('InputPrompt', () => {
                     markSelected: vi.fn(),
                 },
             });
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiActions,
                 uiState: {},
             });
@@ -2522,7 +2732,7 @@ describe('InputPrompt', () => {
                     markSelected: vi.fn(),
                 },
             });
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiActions,
             });
             await act(async () => {
@@ -2546,7 +2756,7 @@ describe('InputPrompt', () => {
                     markSelected: vi.fn(),
                 },
             });
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiActions,
                 uiState: { activePtyId: 1, cleanUiDetailsVisible: false },
             });
@@ -2572,7 +2782,7 @@ describe('InputPrompt', () => {
                     markSelected: vi.fn(),
                 },
             });
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiActions,
                 uiState: {},
             });
@@ -2619,14 +2829,16 @@ describe('InputPrompt', () => {
         ])('should move cursor on mouse click - $name', async ({ relX, relY, mouseCol, mouseRow }) => {
             props.buffer.text = 'hello world\nsecond line';
             props.buffer.lines = ['hello world', 'second line'];
+            props.buffer.allVisualLines = ['hello world', 'second line'];
             props.buffer.viewportVisualLines = ['hello world', 'second line'];
+            props.buffer.viewportHeight = 10;
             props.buffer.visualToLogicalMap = [
                 [0, 0],
                 [1, 0],
             ];
             props.buffer.visualCursor = [0, 11];
             props.buffer.visualScrollRow = 0;
-            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), { mouseEventsEnabled: true, uiActions });
+            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), { mouseEventsEnabled: true, uiActions });
             // Wait for initial render
             await waitFor(() => {
                 expect(stdout.lastFrame()).toContain('hello world');
@@ -2642,12 +2854,9 @@ describe('InputPrompt', () => {
             unmount();
         });
         it('should unfocus embedded shell on click', async () => {
-            props.buffer.text = 'hello';
-            props.buffer.lines = ['hello'];
-            props.buffer.viewportVisualLines = ['hello'];
-            props.buffer.visualToLogicalMap = [[0, 0]];
+            props.buffer.setText('hello');
             props.isEmbeddedShellFocused = true;
-            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), { mouseEventsEnabled: true, uiActions });
+            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), { mouseEventsEnabled: true, uiActions });
             await waitFor(() => {
                 expect(stdout.lastFrame()).toContain('hello');
             });
@@ -2674,6 +2883,7 @@ describe('InputPrompt', () => {
                     lines: currentLines,
                     viewportVisualLines: currentLines,
                     allVisualLines: currentLines,
+                    viewportHeight: 10,
                     pastedContent: { [id]: largeText },
                     transformationsByLine: isExpanded
                         ? currentLines.map(() => [])
@@ -2703,7 +2913,7 @@ describe('InputPrompt', () => {
                         .fn()
                         .mockReturnValue(isExpanded ? id : null),
                 };
-                return _jsx(InputPrompt, { ...baseProps, buffer: buffer });
+                return _jsx(TestInputPrompt, { ...baseProps, buffer: buffer });
             };
             const { stdout, unmount, simulateClick } = await renderWithProviders(_jsx(TestWrapper, {}), {
                 mouseEventsEnabled: true,
@@ -2745,6 +2955,7 @@ describe('InputPrompt', () => {
                     lines: currentLines,
                     viewportVisualLines: currentLines,
                     allVisualLines: currentLines,
+                    viewportHeight: 10,
                     pastedContent: { [id]: largeText },
                     transformationsByLine: isExpanded
                         ? currentLines.map(() => [])
@@ -2774,7 +2985,7 @@ describe('InputPrompt', () => {
                         .fn()
                         .mockImplementation((row) => isExpanded && row >= 0 && row < 10 ? id : null),
                 };
-                return _jsx(InputPrompt, { ...baseProps, buffer: buffer });
+                return _jsx(TestInputPrompt, { ...baseProps, buffer: buffer });
             };
             const { stdout, unmount, simulateClick } = await renderWithProviders(_jsx(TestWrapper, {}), {
                 mouseEventsEnabled: true,
@@ -2800,11 +3011,13 @@ describe('InputPrompt', () => {
             props.config.getUseBackgroundColor = () => false;
             props.buffer.text = 'hello world';
             props.buffer.lines = ['hello world'];
+            props.buffer.allVisualLines = ['hello world'];
             props.buffer.viewportVisualLines = ['hello world'];
+            props.buffer.viewportHeight = 10;
             props.buffer.visualToLogicalMap = [[0, 0]];
             props.buffer.visualCursor = [0, 11];
             props.buffer.visualScrollRow = 0;
-            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), { mouseEventsEnabled: true, uiActions });
+            const { stdin, stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), { mouseEventsEnabled: true, uiActions });
             // Wait for initial render
             await waitFor(() => {
                 expect(stdout.lastFrame()).toContain('hello world');
@@ -2825,7 +3038,7 @@ describe('InputPrompt', () => {
             mockPopAllMessages.mockReturnValue('Message 1\n\nMessage 2\n\nMessage 3');
             props.popAllMessages = mockPopAllMessages;
             props.buffer.text = '';
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\u001B[A');
             });
@@ -2837,7 +3050,7 @@ describe('InputPrompt', () => {
             const mockPopAllMessages = vi.fn();
             props.popAllMessages = mockPopAllMessages;
             props.buffer.text = 'some text';
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\u001B[A');
             });
@@ -2850,7 +3063,7 @@ describe('InputPrompt', () => {
             mockPopAllMessages.mockReturnValue(undefined);
             props.popAllMessages = mockPopAllMessages;
             props.buffer.text = '';
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\u001B[A');
             });
@@ -2866,7 +3079,7 @@ describe('InputPrompt', () => {
             props.buffer.allVisualLines = [''];
             props.buffer.visualCursor = [0, 0];
             props.buffer.visualScrollRow = 0;
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\u001B[A');
             });
@@ -2878,7 +3091,7 @@ describe('InputPrompt', () => {
             mockPopAllMessages.mockReturnValue('Single message');
             props.popAllMessages = mockPopAllMessages;
             props.buffer.text = '';
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\u001B[A');
             });
@@ -2890,7 +3103,7 @@ describe('InputPrompt', () => {
             const mockPopAllMessages = vi.fn();
             props.popAllMessages = mockPopAllMessages;
             props.buffer.text = '   '; // Whitespace only
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\u001B[A');
             });
@@ -2900,7 +3113,7 @@ describe('InputPrompt', () => {
         it('should not call popAllMessages if it is not provided', async () => {
             props.popAllMessages = undefined;
             props.buffer.text = '';
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\u001B[A');
             });
@@ -2912,7 +3125,7 @@ describe('InputPrompt', () => {
             mockPopAllMessages.mockReturnValue(undefined);
             props.popAllMessages = mockPopAllMessages;
             props.buffer.text = '';
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\u001B[A');
             });
@@ -2925,21 +3138,21 @@ describe('InputPrompt', () => {
     describe('snapshots', () => {
         it('should render correctly in shell mode', async () => {
             props.shellModeActive = true;
-            const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await waitFor(() => expect(stdout.lastFrame()).toContain('!'));
             expect(stdout.lastFrame()).toMatchSnapshot();
             unmount();
         });
         it('should render correctly when accepting edits', async () => {
             props.approvalMode = ApprovalMode.AUTO_EDIT;
-            const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await waitFor(() => expect(stdout.lastFrame()).toContain('>'));
             expect(stdout.lastFrame()).toMatchSnapshot();
             unmount();
         });
         it('should render correctly in yolo mode', async () => {
             props.approvalMode = ApprovalMode.YOLO;
-            const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await waitFor(() => expect(stdout.lastFrame()).toContain('*'));
             expect(stdout.lastFrame()).toMatchSnapshot();
             unmount();
@@ -2947,14 +3160,14 @@ describe('InputPrompt', () => {
         it('should not show inverted cursor when shell is focused', async () => {
             props.isEmbeddedShellFocused = true;
             props.focus = false;
-            const renderResult = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const renderResult = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await renderResult.waitUntilReady();
             await expect(renderResult).toMatchSvgSnapshot();
             renderResult.unmount();
         });
     });
     it('should still allow input when shell is not focused', async () => {
-        const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+        const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
             shellFocus: false,
         });
         await act(async () => {
@@ -3001,7 +3214,7 @@ describe('InputPrompt', () => {
         ])('$name', async ({ bufferText, shellMode, shouldSubmit, errorMessage }) => {
             props.buffer.text = bufferText;
             props.shellModeActive = shellMode;
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await act(async () => {
                 stdin.write('\r');
             });
@@ -3021,13 +3234,10 @@ describe('InputPrompt', () => {
     describe('IME Cursor Support', () => {
         it('should report correct cursor position for simple ASCII text', async () => {
             const text = 'hello';
-            mockBuffer.text = text;
-            mockBuffer.lines = [text];
-            mockBuffer.viewportVisualLines = [text];
-            mockBuffer.visualToLogicalMap = [[0, 0]];
+            mockBuffer.setText(text);
             mockBuffer.visualCursor = [0, 3]; // Cursor after 'hel'
             mockBuffer.visualScrollRow = 0;
-            const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), { uiActions });
+            const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), { uiActions });
             await waitFor(() => {
                 expect(stdout.lastFrame()).toContain('hello');
             });
@@ -3043,13 +3253,10 @@ describe('InputPrompt', () => {
         });
         it('should report correct cursor position for text with double-width characters', async () => {
             const text = '👍hello';
-            mockBuffer.text = text;
-            mockBuffer.lines = [text];
-            mockBuffer.viewportVisualLines = [text];
-            mockBuffer.visualToLogicalMap = [[0, 0]];
+            mockBuffer.setText(text);
             mockBuffer.visualCursor = [0, 2]; // Cursor after '👍h' (Note: '👍' is one code point but width 2)
             mockBuffer.visualScrollRow = 0;
-            const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), { uiActions });
+            const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), { uiActions });
             await waitFor(() => {
                 expect(stdout.lastFrame()).toContain('👍hello');
             });
@@ -3064,13 +3271,10 @@ describe('InputPrompt', () => {
         });
         it('should report correct cursor position for a line full of "😀" emojis', async () => {
             const text = '😀😀😀';
-            mockBuffer.text = text;
-            mockBuffer.lines = [text];
-            mockBuffer.viewportVisualLines = [text];
-            mockBuffer.visualToLogicalMap = [[0, 0]];
+            mockBuffer.setText(text);
             mockBuffer.visualCursor = [0, 2]; // Cursor after 2 emojis (each 1 code point, width 2)
             mockBuffer.visualScrollRow = 0;
-            const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), { uiActions });
+            const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), { uiActions });
             await waitFor(() => {
                 expect(stdout.lastFrame()).toContain('😀😀😀');
             });
@@ -3087,7 +3291,9 @@ describe('InputPrompt', () => {
             const lines = ['😀😀', 'hello 😀', 'world'];
             mockBuffer.text = lines.join('\n');
             mockBuffer.lines = lines;
+            mockBuffer.allVisualLines = lines;
             mockBuffer.viewportVisualLines = lines;
+            mockBuffer.viewportHeight = 10;
             mockBuffer.visualToLogicalMap = [
                 [0, 0],
                 [1, 0],
@@ -3095,7 +3301,7 @@ describe('InputPrompt', () => {
             ];
             mockBuffer.visualCursor = [1, 7]; // Second line, after 'hello 😀' (6 chars + 1 emoji = 7 code points)
             mockBuffer.visualScrollRow = 0;
-            const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), { uiActions });
+            const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), { uiActions });
             await waitFor(() => {
                 expect(stdout.lastFrame()).toContain('hello 😀');
             });
@@ -3112,7 +3318,9 @@ describe('InputPrompt', () => {
             const lines = ['first line', 'second line', 'third line'];
             mockBuffer.text = lines.join('\n');
             mockBuffer.lines = lines;
+            mockBuffer.allVisualLines = lines;
             mockBuffer.viewportVisualLines = lines;
+            mockBuffer.viewportHeight = 10;
             mockBuffer.visualToLogicalMap = [
                 [0, 0],
                 [1, 0],
@@ -3120,7 +3328,7 @@ describe('InputPrompt', () => {
             ];
             mockBuffer.visualCursor = [1, 7]; // Cursor on second line, after 'second '
             mockBuffer.visualScrollRow = 0;
-            const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), { uiActions });
+            const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), { uiActions });
             await waitFor(() => {
                 expect(stdout.lastFrame()).toContain('second line');
             });
@@ -3139,11 +3347,12 @@ describe('InputPrompt', () => {
         it('should report cursor position 0 when input is empty and placeholder is shown', async () => {
             mockBuffer.text = '';
             mockBuffer.lines = [''];
+            mockBuffer.allVisualLines = [''];
             mockBuffer.viewportVisualLines = [''];
             mockBuffer.visualToLogicalMap = [[0, 0]];
             mockBuffer.visualCursor = [0, 0];
             mockBuffer.visualScrollRow = 0;
-            const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props, placeholder: "Type here" }), { uiActions });
+            const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props, placeholder: "Type here" }), { uiActions });
             await waitFor(() => {
                 expect(stdout.lastFrame()).toContain('Type here');
             });
@@ -3162,18 +3371,19 @@ describe('InputPrompt', () => {
         const applyVisualState = (visualLine, cursorCol) => {
             mockBuffer.text = logicalLine;
             mockBuffer.lines = [logicalLine];
-            mockBuffer.viewportVisualLines = [visualLine];
             mockBuffer.allVisualLines = [visualLine];
+            mockBuffer.viewportVisualLines = [visualLine];
             mockBuffer.visualToLogicalMap = [[0, 0]];
             mockBuffer.visualToTransformedMap = [0];
             mockBuffer.transformationsByLine = [transformations];
             mockBuffer.cursor = [0, cursorCol];
-            mockBuffer.visualCursor = [0, 0];
+            mockBuffer.visualCursor = [0, cursorCol];
+            mockBuffer.visualScrollRow = 0;
         };
         it('should snapshot collapsed image path', async () => {
             const { transformedLine } = calculateTransformedLine(logicalLine, 0, [0, transformations[0].logEnd + 5], transformations);
             applyVisualState(transformedLine, transformations[0].logEnd + 5);
-            const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await waitFor(() => {
                 expect(stdout.lastFrame()).toContain('[Image');
             });
@@ -3183,7 +3393,7 @@ describe('InputPrompt', () => {
         it('should snapshot expanded image path when cursor is on it', async () => {
             const { transformedLine } = calculateTransformedLine(logicalLine, 0, [0, transformations[0].logStart + 1], transformations);
             applyVisualState(transformedLine, transformations[0].logStart + 1);
-            const { stdout, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { stdout, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }));
             await waitFor(() => {
                 expect(stdout.lastFrame()).toContain('@/path/to/screenshots');
             });
@@ -3219,7 +3429,7 @@ describe('InputPrompt', () => {
                 getExpandedPasteAtLine: vi.fn().mockReturnValue(null),
                 togglePasteExpansion: toggleFn,
             };
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props, buffer: buffer }), { uiActions });
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props, buffer: buffer }), { uiActions });
             await act(async () => {
                 stdin.write(CTRL_O);
             });
@@ -3262,7 +3472,7 @@ describe('InputPrompt', () => {
                     checking: false,
                 });
             }
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props, buffer: method === 'terminal-paste' ? buffer : props.buffer }));
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props, buffer: method === 'terminal-paste' ? buffer : props.buffer }));
             await act(async () => {
                 if (method === 'ctrl-v') {
                     stdin.write('\x16'); // Ctrl+V
@@ -3424,7 +3634,7 @@ describe('InputPrompt', () => {
             { name: 'Up arrow', key: '\u001B[A', position: 'start' },
             { name: 'Ctrl+P', key: '\u0010', position: 'start' },
         ])('should move cursor to $position on $name (older history)', async ({ key, position }) => {
-            const { stdin } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdin } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiActions,
             });
             await act(async () => {
@@ -3438,7 +3648,7 @@ describe('InputPrompt', () => {
             { name: 'Down arrow', key: '\u001B[B', position: 'end' },
             { name: 'Ctrl+N', key: '\u000E', position: 'end' },
         ])('should move cursor to $position on $name (newer history)', async ({ key, position }) => {
-            const { stdin } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdin } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiActions,
             });
             // First go up
@@ -3458,7 +3668,7 @@ describe('InputPrompt', () => {
             });
         });
         it('should suppress completion after history navigation', async () => {
-            const { stdin } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdin } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiActions,
             });
             await act(async () => {
@@ -3486,7 +3696,7 @@ describe('InputPrompt', () => {
                     ? [{ value: 'suggestion', label: 'suggestion' }]
                     : [],
             }));
-            const { stdout, stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), { uiActions });
+            const { stdout, stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), { uiActions });
             // 2. Verify suggestions ARE showing initially because active is true by default
             await waitFor(() => {
                 expect(stdout.lastFrame()).toContain('suggestion');
@@ -3503,7 +3713,7 @@ describe('InputPrompt', () => {
             unmount();
         });
         it('should continue to suppress completion after manual cursor movement', async () => {
-            const { stdin } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdin } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiActions,
             });
             // Navigate history (suppresses)
@@ -3541,7 +3751,7 @@ describe('InputPrompt', () => {
             });
         });
         it('should re-enable completion after typing', async () => {
-            const { stdin } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdin } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiActions,
             });
             // Navigate history (suppresses)
@@ -3567,7 +3777,7 @@ describe('InputPrompt', () => {
             const settings = createMockSettings({
                 ui: { showShortcutsHint: false },
             });
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 settings,
                 uiActions: { setShortcutsHelpVisible },
             });
@@ -3606,8 +3816,8 @@ describe('InputPrompt', () => {
                 input: '\x12',
             },
             {
-                name: 'Ctrl+X hotkey is pressed',
-                input: '\x18',
+                name: 'Ctrl+G hotkey is pressed',
+                input: '\x07',
             },
             {
                 name: 'F12 hotkey is pressed',
@@ -3616,7 +3826,7 @@ describe('InputPrompt', () => {
         ])('should close shortcuts help when a $name', async ({ input, setupMocks, mouseEventsEnabled }) => {
             setupMocks?.();
             const setShortcutsHelpVisible = vi.fn();
-            const { stdin, unmount } = await renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            const { stdin, unmount } = await renderWithProviders(_jsx(TestInputPrompt, { ...props }), {
                 uiState: { shortcutsHelpVisible: true },
                 uiActions: { setShortcutsHelpVisible },
                 mouseEventsEnabled,