@google/gemini-cli 0.30.1 → 0.31.0-preview.1

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

@@ -190,7 +190,11 @@ const createMockConfig = (overrides = {}) => ({
     }),
     ...overrides,
 });
-const renderComposer = (uiState, settings = createMockSettings(), config = createMockConfig(), uiActions = createMockUIActions()) => render(_jsx(ConfigContext.Provider, { value: config, children: _jsx(SettingsContext.Provider, { value: settings, children: _jsx(UIStateContext.Provider, { value: uiState, children: _jsx(UIActionsContext.Provider, { value: uiActions, children: _jsx(Composer, {}) }) }) }) }));
+const renderComposer = async (uiState, settings = createMockSettings(), config = createMockConfig(), uiActions = createMockUIActions()) => {
+    const result = render(_jsx(ConfigContext.Provider, { value: config, children: _jsx(SettingsContext.Provider, { value: settings, children: _jsx(UIStateContext.Provider, { value: uiState, children: _jsx(UIActionsContext.Provider, { value: uiActions, children: _jsx(Composer, {}) }) }) }) }));
+    await result.waitUntilReady();
+    return result;
+};
 describe('Composer', () => {
     beforeEach(() => {
         composerTestControls.suggestionsVisible = false;
@@ -200,16 +204,16 @@ describe('Composer', () => {
         vi.restoreAllMocks();
     });
     describe('Footer Display Settings', () => {
-        it('renders Footer by default when hideFooter is false', () => {
+        it('renders Footer by default when hideFooter is false', async () => {
             const uiState = createMockUIState();
             const settings = createMockSettings({ ui: { hideFooter: false } });
-            const { lastFrame } = renderComposer(uiState, settings);
+            const { lastFrame } = await renderComposer(uiState, settings);
             expect(lastFrame()).toContain('Footer');
         });
-        it('does NOT render Footer when hideFooter is true', () => {
+        it('does NOT render Footer when hideFooter is true', async () => {
             const uiState = createMockUIState();
             const settings = createMockSettings({ ui: { hideFooter: true } });
-            const { lastFrame } = renderComposer(uiState, settings);
+            const { lastFrame } = await renderComposer(uiState, settings);
             // Check for content that only appears IN the Footer component itself
             expect(lastFrame()).not.toContain('[NORMAL]'); // Vim mode indicator
             expect(lastFrame()).not.toContain('(main'); // Branch name with parentheses
@@ -250,13 +254,13 @@ describe('Composer', () => {
                 toggleVimEnabled: vi.fn(),
                 setVimMode: vi.fn(),
             });
-            const { lastFrame } = renderComposer(uiState, settings, config);
+            const { lastFrame } = await renderComposer(uiState, settings, config);
             expect(lastFrame()).toContain('Footer');
             // Footer should be rendered with all the state passed through
         });
     });
     describe('Loading Indicator', () => {
-        it('renders LoadingIndicator with thought when streaming', () => {
+        it('renders LoadingIndicator with thought when streaming', async () => {
             const uiState = createMockUIState({
                 streamingState: StreamingState.Responding,
                 thought: {
@@ -266,11 +270,11 @@ describe('Composer', () => {
                 currentLoadingPhrase: 'Analyzing',
                 elapsedTime: 1500,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             const output = lastFrame();
             expect(output).toContain('LoadingIndicator: Processing');
         });
-        it('renders generic thinking text in loading indicator when full inline thinking is enabled', () => {
+        it('renders generic thinking text in loading indicator when full inline thinking is enabled', async () => {
             const uiState = createMockUIState({
                 streamingState: StreamingState.Responding,
                 thought: {
@@ -281,35 +285,35 @@ describe('Composer', () => {
             const settings = createMockSettings({
                 ui: { inlineThinkingMode: 'full' },
             });
-            const { lastFrame } = renderComposer(uiState, settings);
+            const { lastFrame } = await renderComposer(uiState, settings);
             const output = lastFrame();
             expect(output).toContain('LoadingIndicator: Thinking ...');
         });
-        it('hides shortcuts hint while loading', () => {
+        it('hides shortcuts hint while loading', async () => {
             const uiState = createMockUIState({
                 streamingState: StreamingState.Responding,
                 elapsedTime: 1,
                 cleanUiDetailsVisible: false,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             const output = lastFrame();
             expect(output).toContain('LoadingIndicator');
             expect(output).not.toContain('ShortcutsHint');
         });
-        it('renders LoadingIndicator without thought when accessibility disables loading phrases', () => {
+        it('renders LoadingIndicator without thought when loadingPhrases is off', async () => {
             const uiState = createMockUIState({
                 streamingState: StreamingState.Responding,
                 thought: { subject: 'Hidden', description: 'Should not show' },
             });
-            const config = createMockConfig({
-                getAccessibility: vi.fn(() => ({ enableLoadingPhrases: false })),
+            const settings = createMockSettings({
+                merged: { ui: { loadingPhrases: 'off' } },
             });
-            const { lastFrame } = renderComposer(uiState, undefined, config);
+            const { lastFrame } = await renderComposer(uiState, settings);
             const output = lastFrame();
             expect(output).toContain('LoadingIndicator');
             expect(output).not.toContain('Should not show');
         });
-        it('does not render LoadingIndicator when waiting for confirmation', () => {
+        it('does not render LoadingIndicator when waiting for confirmation', async () => {
             const uiState = createMockUIState({
                 streamingState: StreamingState.WaitingForConfirmation,
                 thought: {
@@ -317,11 +321,11 @@ describe('Composer', () => {
                     description: 'Should not show during confirmation',
                 },
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             const output = lastFrame();
             expect(output).not.toContain('LoadingIndicator');
         });
-        it('does not render LoadingIndicator when a tool confirmation is pending', () => {
+        it('does not render LoadingIndicator when a tool confirmation is pending', async () => {
             const uiState = createMockUIState({
                 streamingState: StreamingState.Responding,
                 pendingHistoryItems: [
@@ -340,22 +344,22 @@ describe('Composer', () => {
                     },
                 ],
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             const output = lastFrame();
             expect(output).not.toContain('LoadingIndicator');
             expect(output).not.toContain('esc to cancel');
         });
-        it('renders LoadingIndicator when embedded shell is focused but background shell is visible', () => {
+        it('renders LoadingIndicator when embedded shell is focused but background shell is visible', async () => {
             const uiState = createMockUIState({
                 streamingState: StreamingState.Responding,
                 embeddedShellFocused: true,
                 isBackgroundShellVisible: true,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             const output = lastFrame();
             expect(output).toContain('LoadingIndicator');
         });
-        it('renders both LoadingIndicator and ApprovalModeIndicator when streaming in full UI mode', () => {
+        it('renders both LoadingIndicator and ApprovalModeIndicator when streaming in full UI mode', async () => {
             const uiState = createMockUIState({
                 streamingState: StreamingState.Responding,
                 thought: {
@@ -364,24 +368,24 @@ describe('Composer', () => {
                 },
                 showApprovalModeIndicator: ApprovalMode.PLAN,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             const output = lastFrame();
             expect(output).toContain('LoadingIndicator: Thinking');
             expect(output).toContain('ApprovalModeIndicator');
         });
-        it('does NOT render LoadingIndicator when embedded shell is focused and background shell is NOT visible', () => {
+        it('does NOT render LoadingIndicator when embedded shell is focused and background shell is NOT visible', async () => {
             const uiState = createMockUIState({
                 streamingState: StreamingState.Responding,
                 embeddedShellFocused: true,
                 isBackgroundShellVisible: false,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             const output = lastFrame();
             expect(output).not.toContain('LoadingIndicator');
         });
     });
     describe('Message Queue Display', () => {
-        it('displays queued messages when present', () => {
+        it('displays queued messages when present', async () => {
             const uiState = createMockUIState({
                 messageQueue: [
                     'First queued message',
@@ -389,17 +393,17 @@ describe('Composer', () => {
                     'Third queued message',
                 ],
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             const output = lastFrame();
             expect(output).toContain('First queued message');
             expect(output).toContain('Second queued message');
             expect(output).toContain('Third queued message');
         });
-        it('renders QueuedMessageDisplay with empty message queue', () => {
+        it('renders QueuedMessageDisplay with empty message queue', async () => {
             const uiState = createMockUIState({
                 messageQueue: [],
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             // The component should render but return null for empty queue
             // This test verifies that the component receives the correct prop
             const output = lastFrame();
@@ -407,47 +411,47 @@ describe('Composer', () => {
         });
     });
     describe('Context and Status Display', () => {
-        it('shows StatusDisplay and ApprovalModeIndicator in normal state', () => {
+        it('shows StatusDisplay and ApprovalModeIndicator in normal state', async () => {
             const uiState = createMockUIState({
                 ctrlCPressedOnce: false,
                 ctrlDPressedOnce: false,
                 showEscapePrompt: false,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             const output = lastFrame();
             expect(output).toContain('StatusDisplay');
             expect(output).toContain('ApprovalModeIndicator');
             expect(output).not.toContain('ToastDisplay');
         });
-        it('shows ToastDisplay and hides ApprovalModeIndicator when a toast is present', () => {
+        it('shows ToastDisplay and hides ApprovalModeIndicator when a toast is present', async () => {
             const uiState = createMockUIState({
                 ctrlCPressedOnce: true,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             const output = lastFrame();
             expect(output).toContain('ToastDisplay');
             expect(output).not.toContain('ApprovalModeIndicator');
             expect(output).toContain('StatusDisplay');
         });
-        it('shows ToastDisplay for other toast types', () => {
+        it('shows ToastDisplay for other toast types', async () => {
             const uiState = createMockUIState({
                 transientMessage: {
                     text: 'Warning',
                     type: TransientMessageType.Warning,
                 },
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             const output = lastFrame();
             expect(output).toContain('ToastDisplay');
             expect(output).not.toContain('ApprovalModeIndicator');
         });
     });
     describe('Input and Indicators', () => {
-        it('hides non-essential UI details in clean mode', () => {
+        it('hides non-essential UI details in clean mode', async () => {
             const uiState = createMockUIState({
                 cleanUiDetailsVisible: false,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             const output = lastFrame();
             expect(output).toContain('ShortcutsHint');
             expect(output).toContain('InputPrompt');
@@ -455,18 +459,18 @@ describe('Composer', () => {
             expect(output).not.toContain('ApprovalModeIndicator');
             expect(output).not.toContain('ContextSummaryDisplay');
         });
-        it('renders InputPrompt when input is active', () => {
+        it('renders InputPrompt when input is active', async () => {
             const uiState = createMockUIState({
                 isInputActive: true,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toContain('InputPrompt');
         });
-        it('does not render InputPrompt when input is inactive', () => {
+        it('does not render InputPrompt when input is inactive', async () => {
             const uiState = createMockUIState({
                 isInputActive: false,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).not.toContain('InputPrompt');
         });
         it.each([
@@ -474,83 +478,83 @@ describe('Composer', () => {
             [ApprovalMode.AUTO_EDIT],
             [ApprovalMode.PLAN],
             [ApprovalMode.YOLO],
-        ])('shows ApprovalModeIndicator when approval mode is %s and shell mode is inactive', (mode) => {
+        ])('shows ApprovalModeIndicator when approval mode is %s and shell mode is inactive', async (mode) => {
             const uiState = createMockUIState({
                 showApprovalModeIndicator: mode,
                 shellModeActive: false,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toMatch(/ApprovalModeIndic[\s\S]*ator/);
         });
-        it('shows ShellModeIndicator when shell mode is active', () => {
+        it('shows ShellModeIndicator when shell mode is active', async () => {
             const uiState = createMockUIState({
                 shellModeActive: true,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toMatch(/ShellModeIndic[\s\S]*tor/);
         });
-        it('shows RawMarkdownIndicator when renderMarkdown is false', () => {
+        it('shows RawMarkdownIndicator when renderMarkdown is false', async () => {
             const uiState = createMockUIState({
                 renderMarkdown: false,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toContain('raw markdown mode');
         });
-        it('does not show RawMarkdownIndicator when renderMarkdown is true', () => {
+        it('does not show RawMarkdownIndicator when renderMarkdown is true', async () => {
             const uiState = createMockUIState({
                 renderMarkdown: true,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).not.toContain('raw markdown mode');
         });
         it.each([
             [ApprovalMode.YOLO, 'YOLO'],
             [ApprovalMode.PLAN, 'plan'],
             [ApprovalMode.AUTO_EDIT, 'auto edit'],
-        ])('shows minimal mode badge "%s" when clean UI details are hidden', (mode, label) => {
+        ])('shows minimal mode badge "%s" when clean UI details are hidden', async (mode, label) => {
             const uiState = createMockUIState({
                 cleanUiDetailsVisible: false,
                 showApprovalModeIndicator: mode,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toContain(label);
         });
-        it('hides minimal mode badge while loading in clean mode', () => {
+        it('hides minimal mode badge while loading in clean mode', async () => {
             const uiState = createMockUIState({
                 cleanUiDetailsVisible: false,
                 streamingState: StreamingState.Responding,
                 elapsedTime: 1,
                 showApprovalModeIndicator: ApprovalMode.PLAN,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             const output = lastFrame();
             expect(output).toContain('LoadingIndicator');
             expect(output).not.toContain('plan');
             expect(output).not.toContain('ShortcutsHint');
         });
-        it('hides minimal mode badge while action-required state is active', () => {
+        it('hides minimal mode badge while action-required state is active', async () => {
             const uiState = createMockUIState({
                 cleanUiDetailsVisible: false,
                 showApprovalModeIndicator: ApprovalMode.PLAN,
                 customDialog: (_jsx(Box, { children: _jsx(Text, { children: "Prompt" }) })),
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             const output = lastFrame();
             expect(output).not.toContain('plan');
             expect(output).not.toContain('ShortcutsHint');
         });
-        it('shows Esc rewind prompt in minimal mode without showing full UI', () => {
+        it('shows Esc rewind prompt in minimal mode without showing full UI', async () => {
             const uiState = createMockUIState({
                 cleanUiDetailsVisible: false,
                 showEscapePrompt: true,
                 history: [{ id: 1, type: 'user', text: 'msg' }],
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             const output = lastFrame();
             expect(output).toContain('ToastDisplay');
             expect(output).not.toContain('ContextSummaryDisplay');
         });
-        it('shows context usage bleed-through when over 60%', () => {
+        it('shows context usage bleed-through when over 60%', async () => {
             const model = 'gemini-2.5-pro';
             const uiState = createMockUIState({
                 cleanUiDetailsVisible: false,
@@ -569,12 +573,12 @@ describe('Composer', () => {
                     footer: { hideContextPercentage: false },
                 },
             });
-            const { lastFrame } = renderComposer(uiState, settings);
+            const { lastFrame } = await renderComposer(uiState, settings);
             expect(lastFrame()).toContain('%');
         });
     });
     describe('Error Details Display', () => {
-        it('shows DetailedMessagesDisplay when showErrorDetails is true', () => {
+        it('shows DetailedMessagesDisplay when showErrorDetails is true', async () => {
             const uiState = createMockUIState({
                 showErrorDetails: true,
                 filteredConsoleMessages: [
@@ -585,15 +589,15 @@ describe('Composer', () => {
                     },
                 ],
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toContain('DetailedMessagesDisplay');
             expect(lastFrame()).toContain('ShowMoreLines');
         });
-        it('does not show error details when showErrorDetails is false', () => {
+        it('does not show error details when showErrorDetails is false', async () => {
             const uiState = createMockUIState({
                 showErrorDetails: false,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).not.toContain('DetailedMessagesDisplay');
         });
     });
@@ -607,7 +611,7 @@ describe('Composer', () => {
                 toggleVimEnabled: vi.fn(),
                 setVimMode: vi.fn(),
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toContain("InputPrompt:   Press 'Esc' for NORMAL mode.");
         });
         it('shows correct placeholder in NORMAL mode', async () => {
@@ -619,123 +623,123 @@ describe('Composer', () => {
                 toggleVimEnabled: vi.fn(),
                 setVimMode: vi.fn(),
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toContain("InputPrompt:   Press 'i' for INSERT mode.");
         });
     });
     describe('Shortcuts Hint', () => {
-        it('hides shortcuts hint when showShortcutsHint setting is false', () => {
+        it('hides shortcuts hint when showShortcutsHint setting is false', async () => {
             const uiState = createMockUIState();
             const settings = createMockSettings({
                 ui: {
                     showShortcutsHint: false,
                 },
             });
-            const { lastFrame } = renderComposer(uiState, settings);
+            const { lastFrame } = await renderComposer(uiState, settings);
             expect(lastFrame()).not.toContain('ShortcutsHint');
         });
-        it('hides shortcuts hint when a action is required (e.g. dialog is open)', () => {
+        it('hides shortcuts hint when a action is required (e.g. dialog is open)', async () => {
             const uiState = createMockUIState({
                 customDialog: (_jsxs(Box, { children: [_jsx(Text, { children: "Test Dialog" }), _jsx(Text, { children: "Test Content" })] })),
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).not.toContain('ShortcutsHint');
         });
-        it('keeps shortcuts hint visible when no action is required', () => {
+        it('keeps shortcuts hint visible when no action is required', async () => {
             const uiState = createMockUIState({
                 cleanUiDetailsVisible: false,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toContain('ShortcutsHint');
         });
-        it('shows shortcuts hint when full UI details are visible', () => {
+        it('shows shortcuts hint when full UI details are visible', async () => {
             const uiState = createMockUIState({
                 cleanUiDetailsVisible: true,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toContain('ShortcutsHint');
         });
-        it('hides shortcuts hint while loading in minimal mode', () => {
+        it('hides shortcuts hint while loading in minimal mode', async () => {
             const uiState = createMockUIState({
                 cleanUiDetailsVisible: false,
                 streamingState: StreamingState.Responding,
                 elapsedTime: 1,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).not.toContain('ShortcutsHint');
         });
-        it('shows shortcuts help in minimal mode when toggled on', () => {
+        it('shows shortcuts help in minimal mode when toggled on', async () => {
             const uiState = createMockUIState({
                 cleanUiDetailsVisible: false,
                 shortcutsHelpVisible: true,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toContain('ShortcutsHelp');
         });
-        it('hides shortcuts hint when suggestions are visible above input in alternate buffer', () => {
+        it('hides shortcuts hint when suggestions are visible above input in alternate buffer', async () => {
             composerTestControls.isAlternateBuffer = true;
             composerTestControls.suggestionsVisible = true;
             const uiState = createMockUIState({
                 cleanUiDetailsVisible: false,
                 showApprovalModeIndicator: ApprovalMode.PLAN,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).not.toContain('ShortcutsHint');
             expect(lastFrame()).not.toContain('plan');
         });
-        it('hides approval mode indicator when suggestions are visible above input in alternate buffer', () => {
+        it('hides approval mode indicator when suggestions are visible above input in alternate buffer', async () => {
             composerTestControls.isAlternateBuffer = true;
             composerTestControls.suggestionsVisible = true;
             const uiState = createMockUIState({
                 cleanUiDetailsVisible: true,
                 showApprovalModeIndicator: ApprovalMode.YOLO,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).not.toContain('ApprovalModeIndicator');
         });
-        it('keeps shortcuts hint when suggestions are visible below input in regular buffer', () => {
+        it('keeps shortcuts hint when suggestions are visible below input in regular buffer', async () => {
             composerTestControls.isAlternateBuffer = false;
             composerTestControls.suggestionsVisible = true;
             const uiState = createMockUIState({
                 cleanUiDetailsVisible: false,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toContain('ShortcutsHint');
         });
     });
     describe('Shortcuts Help', () => {
-        it('shows shortcuts help in passive state', () => {
+        it('shows shortcuts help in passive state', async () => {
             const uiState = createMockUIState({
                 shortcutsHelpVisible: true,
                 streamingState: StreamingState.Idle,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toContain('ShortcutsHelp');
         });
-        it('hides shortcuts help while streaming', () => {
+        it('hides shortcuts help while streaming', async () => {
             const uiState = createMockUIState({
                 shortcutsHelpVisible: true,
                 streamingState: StreamingState.Responding,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).not.toContain('ShortcutsHelp');
         });
-        it('hides shortcuts help when action is required', () => {
+        it('hides shortcuts help when action is required', async () => {
             const uiState = createMockUIState({
                 shortcutsHelpVisible: true,
                 customDialog: (_jsx(Box, { children: _jsx(Text, { children: "Dialog content" }) })),
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).not.toContain('ShortcutsHelp');
         });
     });
     describe('Snapshots', () => {
-        it('matches snapshot in idle state', () => {
+        it('matches snapshot in idle state', async () => {
             const uiState = createMockUIState();
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toMatchSnapshot();
         });
-        it('matches snapshot while streaming', () => {
+        it('matches snapshot while streaming', async () => {
             const uiState = createMockUIState({
                 streamingState: StreamingState.Responding,
                 thought: {
@@ -743,30 +747,30 @@ describe('Composer', () => {
                     description: 'Thinking about the meaning of life...',
                 },
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toMatchSnapshot();
         });
-        it('matches snapshot in narrow view', () => {
+        it('matches snapshot in narrow view', async () => {
             const uiState = createMockUIState({
                 terminalWidth: 40,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toMatchSnapshot();
         });
-        it('matches snapshot in minimal UI mode', () => {
+        it('matches snapshot in minimal UI mode', async () => {
             const uiState = createMockUIState({
                 cleanUiDetailsVisible: false,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toMatchSnapshot();
         });
-        it('matches snapshot in minimal UI mode while loading', () => {
+        it('matches snapshot in minimal UI mode while loading', async () => {
             const uiState = createMockUIState({
                 cleanUiDetailsVisible: false,
                 streamingState: StreamingState.Responding,
                 elapsedTime: 1000,
             });
-            const { lastFrame } = renderComposer(uiState);
+            const { lastFrame } = await renderComposer(uiState);
             expect(lastFrame()).toMatchSnapshot();
         });
     });