@machina.ai/cell-cli 1.20.2-rc1 → 1.25.0-rc3

package/dist/src/ui/AppContainer.test.js

@@ -42,8 +42,17 @@ vi.mock('@google/gemini-cli-core', async (importOriginal) => {
         })),
         enableMouseEvents: vi.fn(),
         disableMouseEvents: vi.fn(),
+        FileDiscoveryService: vi.fn().mockImplementation(() => ({
+            initialize: vi.fn(),
+        })),
+        startupProfiler: {
+            flush: vi.fn(),
+            start: vi.fn(),
+            end: vi.fn(),
+        },
     };
 });
+import ansiEscapes from 'ansi-escapes';
 import { useQuotaAndFallback } from './hooks/useQuotaAndFallback.js';
 import { UIStateContext } from './contexts/UIStateContext.js';
 import { UIActionsContext, } from './contexts/UIActionsContext.js';
@@ -96,6 +105,7 @@ vi.mock('./contexts/SessionContext.js');
 vi.mock('./components/shared/text-buffer.js');
 vi.mock('./hooks/useLogger.js');
 vi.mock('./hooks/useInputHistoryStore.js');
+vi.mock('./hooks/useHookDisplayState.js');
 // Mock external utilities
 vi.mock('../utils/events.js');
 vi.mock('../utils/handleAutoUpdate.js');
@@ -122,6 +132,7 @@ import { useTextBuffer } from './components/shared/text-buffer.js';
 import { useLogger } from './hooks/useLogger.js';
 import { useLoadingIndicator } from './hooks/useLoadingIndicator.js';
 import { useInputHistoryStore } from './hooks/useInputHistoryStore.js';
+import { useHookDisplayState } from './hooks/useHookDisplayState.js';
 import { useKeypress } from './hooks/useKeypress.js';
 import { measureElement } from 'ink';
 import { useTerminalSize } from './hooks/useTerminalSize.js';
@@ -160,18 +171,13 @@ describe('AppContainer State Management', () => {
     const mockedUseLoadingIndicator = useLoadingIndicator;
     const mockedUseKeypress = useKeypress;
     const mockedUseInputHistoryStore = useInputHistoryStore;
+    const mockedUseHookDisplayState = useHookDisplayState;
     beforeEach(() => {
         vi.clearAllMocks();
+        mockIdeClient.getInstance.mockReturnValue(new Promise(() => { }));
         // Initialize mock stdout for terminal title tests
         mocks.mockStdout.write.mockClear();
-        // Mock computeWindowTitle function to centralize title logic testing
-        vi.mock('../utils/windowTitle.js', async () => ({
-            computeWindowTitle: vi.fn((folderName) => 
-            // Default behavior: return "Gemini - {folderName}" unless CLI_TITLE is set
-            process.env['CLI_TITLE'] || `Gemini - ${folderName}`),
-        }));
         capturedUIState = null;
-        capturedUIActions = null;
         // **Provide a default return value for EVERY mocked hook.**
         mockedUseQuotaAndFallback.mockReturnValue({
             proQuotaRequest: null,
@@ -270,7 +276,7 @@ describe('AppContainer State Management', () => {
             // Add other properties if AppContainer uses them
         });
         mockedUseLogger.mockReturnValue({
-            getPreviousUserMessages: vi.fn().mockReturnValue(new Promise(() => { })),
+            getPreviousUserMessages: vi.fn().mockResolvedValue([]),
         });
         mockedUseInputHistoryStore.mockReturnValue({
             inputHistory: [],
@@ -281,10 +287,13 @@ describe('AppContainer State Management', () => {
             elapsedTime: '0.0s',
             currentLoadingPhrase: '',
         });
+        mockedUseHookDisplayState.mockReturnValue([]);
         // Mock Config
         mockConfig = makeFakeConfig();
         // Mock config's getTargetDir to return consistent workspace directory
         vi.spyOn(mockConfig, 'getTargetDir').mockReturnValue('/test/workspace');
+        vi.spyOn(mockConfig, 'initialize').mockResolvedValue(undefined);
+        vi.spyOn(mockConfig, 'getDebugMode').mockReturnValue(false);
         mockExtensionManager = vi.mockObject({
             getExtensions: vi.fn().mockReturnValue([]),
             setRequestConsent: vi.fn(),
@@ -317,16 +326,25 @@ describe('AppContainer State Management', () => {
     });
     afterEach(() => {
         cleanup();
+        vi.restoreAllMocks();
     });
     describe('Basic Rendering', () => {
         it('renders without crashing with minimal props', async () => {
-            const { unmount } = renderAppContainer();
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(capturedUIState).toBeTruthy());
             unmount();
         });
         it('renders with startup warnings', async () => {
             const startupWarnings = ['Warning 1', 'Warning 2'];
-            const { unmount } = renderAppContainer({ startupWarnings });
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer({ startupWarnings });
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(capturedUIState).toBeTruthy());
             unmount();
         });
@@ -337,8 +355,12 @@ describe('AppContainer State Management', () => {
                 ...mockInitResult,
                 themeError: 'Failed to load theme',
             };
-            const { unmount } = renderAppContainer({
-                initResult: initResultWithError,
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer({
+                    initResult: initResultWithError,
+                });
+                unmount = result.unmount;
             });
             await waitFor(() => expect(capturedUIState).toBeTruthy());
             unmount();
@@ -353,23 +375,39 @@ describe('AppContainer State Management', () => {
     });
     describe('Context Providers', () => {
         it('provides AppContext with correct values', async () => {
-            const { unmount } = renderAppContainer({ version: '2.0.0' });
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer({ version: '2.0.0' });
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(capturedUIState).toBeTruthy());
             // Should render and unmount cleanly
             expect(() => unmount()).not.toThrow();
         });
         it('provides UIStateContext with state management', async () => {
-            const { unmount } = renderAppContainer();
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(capturedUIState).toBeTruthy());
             unmount();
         });
         it('provides UIActionsContext with action handlers', async () => {
-            const { unmount } = renderAppContainer();
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(capturedUIState).toBeTruthy());
             unmount();
         });
         it('provides ConfigContext with config object', async () => {
-            const { unmount } = renderAppContainer();
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(capturedUIState).toBeTruthy());
             unmount();
         });
@@ -384,7 +422,11 @@ describe('AppContainer State Management', () => {
                     showMemoryUsage: false,
                 },
             };
-            const { unmount } = renderAppContainer({ settings: settingsAllHidden });
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer({ settings: settingsAllHidden });
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(capturedUIState).toBeTruthy());
             unmount();
         });
@@ -397,14 +439,22 @@ describe('AppContainer State Management', () => {
                     showMemoryUsage: true,
                 },
             };
-            const { unmount } = renderAppContainer({ settings: settingsWithMemory });
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer({ settings: settingsWithMemory });
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(capturedUIState).toBeTruthy());
             unmount();
         });
     });
     describe('Version Handling', () => {
         it.each(['1.0.0', '2.1.3-beta', '3.0.0-nightly'])('handles version format: %s', async (version) => {
-            const { unmount } = renderAppContainer({ version });
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer({ version });
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(capturedUIState).toBeTruthy());
             unmount();
         });
@@ -423,7 +473,11 @@ describe('AppContainer State Management', () => {
             const undefinedSettings = {
                 merged: {},
             };
-            const { unmount } = renderAppContainer({ settings: undefinedSettings });
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer({ settings: undefinedSettings });
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(capturedUIState).toBeTruthy());
             unmount();
         });
@@ -719,7 +773,11 @@ describe('AppContainer State Management', () => {
     describe('Quota and Fallback Integration', () => {
         it('passes a null proQuotaRequest to UIStateContext by default', async () => {
             // The default mock from beforeEach already sets proQuotaRequest to null
-            const { unmount } = renderAppContainer();
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => {
                 // Assert that the context value is as expected
                 expect(capturedUIState.proQuotaRequest).toBeNull();
@@ -738,7 +796,11 @@ describe('AppContainer State Management', () => {
                 handleProQuotaChoice: vi.fn(),
             });
             // Act: Render the container
-            const { unmount } = renderAppContainer();
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => {
                 // Assert: The mock request is correctly passed through the context
                 expect(capturedUIState.proQuotaRequest).toEqual(mockRequest);
@@ -753,7 +815,11 @@ describe('AppContainer State Management', () => {
                 handleProQuotaChoice: mockHandler,
             });
             // Act: Render the container
-            const { unmount } = renderAppContainer();
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => {
                 // Assert: The action in the context is the mock handler we provided
                 expect(capturedUIActions.handleProQuotaChoice).toBe(mockHandler);
@@ -776,7 +842,7 @@ describe('AppContainer State Management', () => {
             const { stdout } = useStdout();
             expect(stdout).toBe(mocks.mockStdout);
         });
-        it('should not update terminal title when showStatusInTitle is false', () => {
+        it('should update terminal title with Working… when showStatusInTitle is false', () => {
             // Arrange: Set up mock settings with showStatusInTitle disabled
             const mockSettingsWithShowStatusFalse = {
                 ...mockSettings,
@@ -789,13 +855,55 @@ describe('AppContainer State Management', () => {
                     },
                 },
             };
+            // Mock the streaming state as Active
+            mockedUseGeminiStream.mockReturnValue({
+                streamingState: 'responding',
+                submitQuery: vi.fn(),
+                initError: null,
+                pendingHistoryItems: [],
+                thought: { subject: 'Some thought' },
+                cancelOngoingRequest: vi.fn(),
+            });
             // Act: Render the container
             const { unmount } = renderAppContainer({
                 settings: mockSettingsWithShowStatusFalse,
             });
-            // Assert: Check that no title-related writes occurred
-            const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]2;'));
-            expect(titleWrites).toHaveLength(0);
+            // Assert: Check that title was updated with "Working…"
+            const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]0;'));
+            expect(titleWrites).toHaveLength(1);
+            expect(titleWrites[0][0]).toBe(`\x1b]0;${'✦  Working… (workspace)'.padEnd(80, ' ')}\x07`);
+            unmount();
+        });
+        it('should use legacy terminal title when dynamicWindowTitle is false', () => {
+            // Arrange: Set up mock settings with dynamicWindowTitle disabled
+            const mockSettingsWithDynamicTitleFalse = {
+                ...mockSettings,
+                merged: {
+                    ...mockSettings.merged,
+                    ui: {
+                        ...mockSettings.merged.ui,
+                        dynamicWindowTitle: false,
+                        hideWindowTitle: false,
+                    },
+                },
+            };
+            // Mock the streaming state
+            mockedUseGeminiStream.mockReturnValue({
+                streamingState: 'responding',
+                submitQuery: vi.fn(),
+                initError: null,
+                pendingHistoryItems: [],
+                thought: { subject: 'Some thought' },
+                cancelOngoingRequest: vi.fn(),
+            });
+            // Act: Render the container
+            const { unmount } = renderAppContainer({
+                settings: mockSettingsWithDynamicTitleFalse,
+            });
+            // Assert: Check that legacy title was used
+            const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]0;'));
+            expect(titleWrites).toHaveLength(1);
+            expect(titleWrites[0][0]).toBe(`\x1b]0;${'Cell CLI (workspace)'.padEnd(80, ' ')}\x07`);
             unmount();
         });
         it('should not update terminal title when hideWindowTitle is true', () => {
@@ -816,7 +924,7 @@ describe('AppContainer State Management', () => {
                 settings: mockSettingsWithHideTitleTrue,
             });
             // Assert: Check that no title-related writes occurred
-            const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]2;'));
+            const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]0;'));
             expect(titleWrites).toHaveLength(0);
             unmount();
         });
@@ -847,10 +955,10 @@ describe('AppContainer State Management', () => {
             const { unmount } = renderAppContainer({
                 settings: mockSettingsWithTitleEnabled,
             });
-            // Assert: Check that title was updated with thought subject
-            const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]2;'));
+            // Assert: Check that title was updated with thought subject and suffix
+            const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]0;'));
             expect(titleWrites).toHaveLength(1);
-            expect(titleWrites[0][0]).toBe(`\x1b]2;${thoughtSubject.padEnd(80, ' ')}\x07`);
+            expect(titleWrites[0][0]).toBe(`\x1b]0;${`✦  ${thoughtSubject} (workspace)`.padEnd(80, ' ')}\x07`);
             unmount();
         });
         it('should update terminal title with default text when in Idle state and no thought subject', () => {
@@ -880,12 +988,12 @@ describe('AppContainer State Management', () => {
                 settings: mockSettingsWithTitleEnabled,
             });
             // Assert: Check that title was updated with default Idle text
-            const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]2;'));
+            const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]0;'));
             expect(titleWrites).toHaveLength(1);
-            expect(titleWrites[0][0]).toBe(`\x1b]2;${'Gemini - workspace'.padEnd(80, ' ')}\x07`);
+            expect(titleWrites[0][0]).toBe(`\x1b]0;${'◇  Ready (workspace)'.padEnd(80, ' ')}\x07`);
             unmount();
         });
-        it('should update terminal title when in WaitingForConfirmation state with thought subject', () => {
+        it('should update terminal title when in WaitingForConfirmation state with thought subject', async () => {
             // Arrange: Set up mock settings with showStatusInTitle enabled
             const mockSettingsWithTitleEnabled = {
                 ...mockSettings,
@@ -901,7 +1009,7 @@ describe('AppContainer State Management', () => {
             // Mock the streaming state and thought
             const thoughtSubject = 'Confirm tool execution';
             mockedUseGeminiStream.mockReturnValue({
-                streamingState: 'waitingForConfirmation',
+                streamingState: 'waiting_for_confirmation',
                 submitQuery: vi.fn(),
                 initError: null,
                 pendingHistoryItems: [],
@@ -909,15 +1017,146 @@ describe('AppContainer State Management', () => {
                 cancelOngoingRequest: vi.fn(),
             });
             // Act: Render the container
-            const { unmount } = renderAppContainer({
-                settings: mockSettingsWithTitleEnabled,
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer({
+                    settings: mockSettingsWithTitleEnabled,
+                });
+                unmount = result.unmount;
             });
             // Assert: Check that title was updated with confirmation text
-            const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]2;'));
+            const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]0;'));
             expect(titleWrites).toHaveLength(1);
-            expect(titleWrites[0][0]).toBe(`\x1b]2;${thoughtSubject.padEnd(80, ' ')}\x07`);
+            expect(titleWrites[0][0]).toBe(`\x1b]0;${'✋  Action Required (workspace)'.padEnd(80, ' ')}\x07`);
             unmount();
         });
+        describe('Shell Focus Action Required', () => {
+            beforeEach(() => {
+                vi.useFakeTimers();
+            });
+            afterEach(() => {
+                vi.useRealTimers();
+            });
+            it('should show Action Required in title after a delay when shell is awaiting focus', async () => {
+                const startTime = 1000000;
+                vi.setSystemTime(startTime);
+                // Arrange: Set up mock settings with showStatusInTitle enabled
+                const mockSettingsWithTitleEnabled = {
+                    ...mockSettings,
+                    merged: {
+                        ...mockSettings.merged,
+                        ui: {
+                            ...mockSettings.merged.ui,
+                            showStatusInTitle: true,
+                            hideWindowTitle: false,
+                        },
+                    },
+                };
+                // Mock an active shell pty but not focused
+                mockedUseGeminiStream.mockReturnValue({
+                    streamingState: 'responding',
+                    submitQuery: vi.fn(),
+                    initError: null,
+                    pendingHistoryItems: [],
+                    thought: { subject: 'Executing shell command' },
+                    cancelOngoingRequest: vi.fn(),
+                    activePtyId: 'pty-1',
+                    lastOutputTime: 0,
+                });
+                vi.spyOn(mockConfig, 'isInteractive').mockReturnValue(true);
+                vi.spyOn(mockConfig, 'isInteractiveShellEnabled').mockReturnValue(true);
+                // Act: Render the container (embeddedShellFocused is false by default in state)
+                const { unmount } = renderAppContainer({
+                    settings: mockSettingsWithTitleEnabled,
+                });
+                // Initially it should show the working status
+                const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]0;'));
+                expect(titleWrites[titleWrites.length - 1][0]).toContain('✦  Executing shell command');
+                // Fast-forward time by 40 seconds
+                await act(async () => {
+                    await vi.advanceTimersByTimeAsync(40000);
+                });
+                // Now it should show Action Required
+                const titleWritesDelayed = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]0;'));
+                const lastTitle = titleWritesDelayed[titleWritesDelayed.length - 1][0];
+                expect(lastTitle).toContain('✋  Action Required');
+                unmount();
+            });
+            it('should NOT show Action Required in title if shell is streaming output', async () => {
+                const startTime = 1000000;
+                vi.setSystemTime(startTime);
+                // Arrange: Set up mock settings with showStatusInTitle enabled
+                const mockSettingsWithTitleEnabled = {
+                    ...mockSettings,
+                    merged: {
+                        ...mockSettings.merged,
+                        ui: {
+                            ...mockSettings.merged.ui,
+                            showStatusInTitle: true,
+                            hideWindowTitle: false,
+                        },
+                    },
+                };
+                // Mock an active shell pty but not focused
+                let lastOutputTime = 1000;
+                mockedUseGeminiStream.mockImplementation(() => ({
+                    streamingState: 'responding',
+                    submitQuery: vi.fn(),
+                    initError: null,
+                    pendingHistoryItems: [],
+                    thought: { subject: 'Executing shell command' },
+                    cancelOngoingRequest: vi.fn(),
+                    activePtyId: 'pty-1',
+                    lastOutputTime,
+                }));
+                vi.spyOn(mockConfig, 'isInteractive').mockReturnValue(true);
+                vi.spyOn(mockConfig, 'isInteractiveShellEnabled').mockReturnValue(true);
+                // Act: Render the container
+                const { unmount, rerender } = renderAppContainer({
+                    settings: mockSettingsWithTitleEnabled,
+                });
+                // Fast-forward time by 20 seconds
+                await act(async () => {
+                    await vi.advanceTimersByTimeAsync(20000);
+                });
+                // Update lastOutputTime to simulate new output
+                lastOutputTime = 21000;
+                mockedUseGeminiStream.mockImplementation(() => ({
+                    streamingState: 'responding',
+                    submitQuery: vi.fn(),
+                    initError: null,
+                    pendingHistoryItems: [],
+                    thought: { subject: 'Executing shell command' },
+                    cancelOngoingRequest: vi.fn(),
+                    activePtyId: 'pty-1',
+                    lastOutputTime,
+                }));
+                // Rerender to propagate the new lastOutputTime
+                await act(async () => {
+                    rerender(getAppContainer({ settings: mockSettingsWithTitleEnabled }));
+                });
+                // Fast-forward time by another 20 seconds
+                // Total time elapsed: 40s.
+                // Time since last output: 20s.
+                // It should NOT show Action Required yet.
+                await act(async () => {
+                    await vi.advanceTimersByTimeAsync(20000);
+                });
+                const titleWritesAfterOutput = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]0;'));
+                const lastTitle = titleWritesAfterOutput[titleWritesAfterOutput.length - 1][0];
+                expect(lastTitle).not.toContain('✋  Action Required');
+                expect(lastTitle).toContain('✦  Executing shell command');
+                // Fast-forward another 40 seconds (Total 60s since last output)
+                await act(async () => {
+                    await vi.advanceTimersByTimeAsync(40000);
+                });
+                // Now it SHOULD show Action Required
+                const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]0;'));
+                const lastTitleFinal = titleWrites[titleWrites.length - 1][0];
+                expect(lastTitleFinal).toContain('✋  Action Required');
+                unmount();
+            });
+        });
         it('should pad title to exactly 80 characters', () => {
             // Arrange: Set up mock settings with showStatusInTitle enabled
             const mockSettingsWithTitleEnabled = {
@@ -946,14 +1185,12 @@ describe('AppContainer State Management', () => {
                 settings: mockSettingsWithTitleEnabled,
             });
             // Assert: Check that title is padded to exactly 80 characters
-            const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]2;'));
+            const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]0;'));
             expect(titleWrites).toHaveLength(1);
             const calledWith = titleWrites[0][0];
-            const expectedTitle = shortTitle.padEnd(80, ' ');
-            expect(calledWith).toContain(shortTitle);
-            expect(calledWith).toContain('\x1b]2;');
-            expect(calledWith).toContain('\x07');
-            expect(calledWith).toBe('\x1b]2;' + expectedTitle + '\x07');
+            const expectedTitle = `✦  ${shortTitle} (workspace)`.padEnd(80, ' ');
+            const expectedEscapeSequence = `\x1b]0;${expectedTitle}\x07`;
+            expect(calledWith).toBe(expectedEscapeSequence);
             unmount();
         });
         it('should use correct ANSI escape code format', () => {
@@ -984,30 +1221,30 @@ describe('AppContainer State Management', () => {
                 settings: mockSettingsWithTitleEnabled,
             });
             // Assert: Check that the correct ANSI escape sequence is used
-            const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]2;'));
+            const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]0;'));
             expect(titleWrites).toHaveLength(1);
-            const expectedEscapeSequence = `\x1b]2;${title.padEnd(80, ' ')}\x07`;
+            const expectedEscapeSequence = `\x1b]0;${`✦  ${title} (workspace)`.padEnd(80, ' ')}\x07`;
             expect(titleWrites[0][0]).toBe(expectedEscapeSequence);
             unmount();
         });
         it('should use CLI_TITLE environment variable when set', () => {
-            // Arrange: Set up mock settings with showStatusInTitle enabled
-            const mockSettingsWithTitleEnabled = {
+            // Arrange: Set up mock settings with showStatusInTitle disabled (so it shows suffix)
+            const mockSettingsWithTitleDisabled = {
                 ...mockSettings,
                 merged: {
                     ...mockSettings.merged,
                     ui: {
                         ...mockSettings.merged.ui,
-                        showStatusInTitle: true,
+                        showStatusInTitle: false,
                         hideWindowTitle: false,
                     },
                 },
             };
             // Mock CLI_TITLE environment variable
             vi.stubEnv('CLI_TITLE', 'Custom Gemini Title');
-            // Mock the streaming state as Idle with no thought
+            // Mock the streaming state
             mockedUseGeminiStream.mockReturnValue({
-                streamingState: 'idle',
+                streamingState: 'responding',
                 submitQuery: vi.fn(),
                 initError: null,
                 pendingHistoryItems: [],
@@ -1016,12 +1253,12 @@ describe('AppContainer State Management', () => {
             });
             // Act: Render the container
             const { unmount } = renderAppContainer({
-                settings: mockSettingsWithTitleEnabled,
+                settings: mockSettingsWithTitleDisabled,
             });
             // Assert: Check that title was updated with CLI_TITLE value
-            const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]2;'));
+            const titleWrites = mocks.mockStdout.write.mock.calls.filter((call) => call[0].includes('\x1b]0;'));
             expect(titleWrites).toHaveLength(1);
-            expect(titleWrites[0][0]).toBe(`\x1b]2;${'Custom Gemini Title'.padEnd(80, ' ')}\x07`);
+            expect(titleWrites[0][0]).toBe(`\x1b]0;${'✦  Working… (Custom Gemini Title)'.padEnd(80, ' ')}\x07`);
             unmount();
         });
     });
@@ -1031,6 +1268,7 @@ describe('AppContainer State Management', () => {
         });
         afterEach(() => {
             vi.useRealTimers();
+            vi.restoreAllMocks();
         });
         it('should set and clear the queue error message after a timeout', async () => {
             const { rerender, unmount } = renderAppContainer();
@@ -1099,7 +1337,11 @@ describe('AppContainer State Management', () => {
                 cancelOngoingRequest: vi.fn(),
                 activePtyId: 'some-id',
             });
-            const { unmount } = renderAppContainer();
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(resizePtySpy).toHaveBeenCalled());
             const lastCall = resizePtySpy.mock.calls[resizePtySpy.mock.calls.length - 1];
             // Check the height argument specifically
@@ -1170,6 +1412,7 @@ describe('AppContainer State Management', () => {
         });
         afterEach(() => {
             vi.useRealTimers();
+            vi.restoreAllMocks();
         });
         describe('CTRL+C', () => {
             it('should cancel ongoing request on first press', async () => {
@@ -1270,6 +1513,7 @@ describe('AppContainer State Management', () => {
         });
         afterEach(() => {
             vi.useRealTimers();
+            vi.restoreAllMocks();
         });
         describe.each([
             {
@@ -1383,7 +1627,11 @@ describe('AppContainer State Management', () => {
                 openModelDialog: vi.fn(),
                 closeModelDialog: vi.fn(),
             });
-            const { unmount } = renderAppContainer();
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(capturedUIState).toBeTruthy());
             expect(capturedUIState.isModelDialogOpen).toBe(true);
             unmount();
@@ -1395,7 +1643,11 @@ describe('AppContainer State Management', () => {
                 openModelDialog: vi.fn(),
                 closeModelDialog: mockCloseModelDialog,
             });
-            const { unmount } = renderAppContainer();
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(capturedUIState).toBeTruthy());
             // Verify that the actions are correctly passed through context
             act(() => {
@@ -1407,20 +1659,32 @@ describe('AppContainer State Management', () => {
     });
     describe('CoreEvents Integration', () => {
         it('subscribes to UserFeedback and drains backlog on mount', async () => {
-            const { unmount } = renderAppContainer();
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(capturedUIState).toBeTruthy());
             expect(mockCoreEvents.on).toHaveBeenCalledWith(CoreEvent.UserFeedback, expect.any(Function));
             expect(mockCoreEvents.drainBacklogs).toHaveBeenCalledTimes(1);
             unmount();
         });
         it('unsubscribes from UserFeedback on unmount', async () => {
-            const { unmount } = renderAppContainer();
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(capturedUIState).toBeTruthy());
             unmount();
             expect(mockCoreEvents.off).toHaveBeenCalledWith(CoreEvent.UserFeedback, expect.any(Function));
         });
         it('adds history item when UserFeedback event is received', async () => {
-            const { unmount } = renderAppContainer();
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(capturedUIState).toBeTruthy());
             // Get the registered handler
             const handler = mockCoreEvents.on.mock.calls.find((call) => call[0] === CoreEvent.UserFeedback)?.[1];
@@ -1442,7 +1706,11 @@ describe('AppContainer State Management', () => {
         it('updates currentModel when ModelChanged event is received', async () => {
             // Arrange: Mock initial model
             vi.spyOn(mockConfig, 'getModel').mockReturnValue('initial-model');
-            const { unmount } = renderAppContainer();
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => {
                 expect(capturedUIState?.currentModel).toBe('initial-model');
             });
@@ -1450,11 +1718,27 @@ describe('AppContainer State Management', () => {
             const handler = mockCoreEvents.on.mock.calls.find((call) => call[0] === CoreEvent.ModelChanged)?.[1];
             expect(handler).toBeDefined();
             // Act: Simulate ModelChanged event
+            // Update config mock to return new model since the handler reads from config
+            vi.spyOn(mockConfig, 'getModel').mockReturnValue('new-model');
             act(() => {
                 handler({ model: 'new-model' });
             });
             // Assert: Verify model is updated
-            expect(capturedUIState.currentModel).toBe('new-model');
+            await waitFor(() => {
+                expect(capturedUIState.currentModel).toBe('new-model');
+            });
+            unmount();
+        });
+        it('provides activeHooks from useHookDisplayState', async () => {
+            const mockHooks = [{ name: 'hook1', eventName: 'event1' }];
+            mockedUseHookDisplayState.mockReturnValue(mockHooks);
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
+            await waitFor(() => expect(capturedUIState).toBeTruthy());
+            expect(capturedUIState.activeHooks).toEqual(mockHooks);
             unmount();
         });
     });
@@ -1475,7 +1759,11 @@ describe('AppContainer State Management', () => {
                 activePtyId: 'some-pty-id', // Make sure activePtyId is set
             });
             // The main assertion is that the render does not throw.
-            const { unmount } = render(_jsx(SettingsContext.Provider, { value: mockSettings, children: _jsx(AppContainer, { config: mockConfig, version: "1.0.0", initializationResult: mockInitResult }) }));
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(resizePtySpy).toHaveBeenCalled());
             unmount();
         });
@@ -1487,7 +1775,11 @@ describe('AppContainer State Management', () => {
                 authType: AuthType.USE_GEMINI,
                 apiKey: 'fake-key',
             });
-            const { unmount } = renderAppContainer();
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => {
                 expect(capturedUIState.bannerData.defaultText).toBeDefined();
                 unmount();
@@ -1509,7 +1801,11 @@ describe('AppContainer State Management', () => {
             });
         });
         it('clears the prompt when onCancelSubmit is called with shouldRestorePrompt=false', async () => {
-            const { unmount } = renderAppContainer();
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(capturedUIState).toBeTruthy());
             const { onCancelSubmit } = extractUseGeminiStreamArgs(mockedUseGeminiStream.mock.lastCall);
             act(() => {
@@ -1525,13 +1821,19 @@ describe('AppContainer State Management', () => {
                 addInput: vi.fn(),
                 initializeFromLogger: vi.fn(),
             });
-            const { unmount } = renderAppContainer();
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                unmount = result.unmount;
+            });
             await waitFor(() => expect(capturedUIState.userMessages).toContain('previous message'));
             const { onCancelSubmit } = extractUseGeminiStreamArgs(mockedUseGeminiStream.mock.lastCall);
             await act(async () => {
                 onCancelSubmit(true);
             });
-            expect(mockSetText).toHaveBeenCalledWith('previous message');
+            await waitFor(() => {
+                expect(mockSetText).toHaveBeenCalledWith('previous message');
+            });
             unmount();
         });
         it('input history is independent from conversation history (survives /clear)', async () => {
@@ -1543,7 +1845,13 @@ describe('AppContainer State Management', () => {
                 addInput: mockAddInput,
                 initializeFromLogger: vi.fn(),
             });
-            const { unmount } = renderAppContainer();
+            let rerender;
+            let unmount;
+            await act(async () => {
+                const result = renderAppContainer();
+                rerender = result.rerender;
+                unmount = result.unmount;
+            });
             // Verify userMessages is populated from inputHistory
             await waitFor(() => expect(capturedUIState.userMessages).toContain('first prompt'));
             expect(capturedUIState.userMessages).toContain('second prompt');
@@ -1556,6 +1864,10 @@ describe('AppContainer State Management', () => {
                 clearItems: mockClearItems,
                 loadHistory: vi.fn(),
             });
+            await act(async () => {
+                // Rerender to apply the new mock.
+                rerender(getAppContainer());
+            });
             // Verify that userMessages still contains the input history
             // (it should not be affected by clearing conversation history)
             expect(capturedUIState.userMessages).toContain('first prompt');
@@ -1563,5 +1875,29 @@ describe('AppContainer State Management', () => {
             unmount();
         });
     });
+    describe('Regression Tests', () => {
+        it('does not refresh static on startup if banner text is empty', async () => {
+            // Mock banner text to be empty strings
+            vi.spyOn(mockConfig, 'getBannerTextNoCapacityIssues').mockResolvedValue('');
+            vi.spyOn(mockConfig, 'getBannerTextCapacityIssues').mockResolvedValue('');
+            // Clear previous calls
+            mocks.mockStdout.write.mockClear();
+            let compUnmount = () => { };
+            await act(async () => {
+                const { unmount } = renderAppContainer();
+                compUnmount = unmount;
+            });
+            // Allow async effects to run
+            await waitFor(() => expect(capturedUIState).toBeTruthy());
+            // Wait for fetchBannerTexts to complete
+            await act(async () => {
+                await new Promise((resolve) => setTimeout(resolve, 100));
+            });
+            // Check that clearTerminal was NOT written to stdout
+            const clearTerminalCalls = mocks.mockStdout.write.mock.calls.filter((call) => call[0] === ansiEscapes.clearTerminal);
+            expect(clearTerminalCalls).toHaveLength(0);
+            compUnmount();
+        });
+    });
 });
 //# sourceMappingURL=AppContainer.test.js.map