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

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

@@ -41,8 +41,9 @@ describe('AskUserDialog', () => {
             multiSelect: false,
         },
     ];
-    it('renders question and options', () => {
-        const { lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: authQuestion, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+    it('renders question and options', async () => {
+        const { lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: authQuestion, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+        await waitUntilReady();
         expect(lastFrame()).toMatchSnapshot();
     });
     describe.each([
@@ -100,30 +101,32 @@ describe('AskUserDialog', () => {
             const onSubmit = vi.fn();
             const { stdin } = renderWithProviders(_jsx(AskUserDialog, { questions: questions, onSubmit: onSubmit, onCancel: vi.fn(), width: 120 }), { width: 120 });
             actions(stdin);
-            await waitFor(() => {
+            await waitFor(async () => {
                 expect(onSubmit).toHaveBeenCalledWith(expectedSubmit);
             });
         });
     });
     it('handles custom option in single select with inline typing', async () => {
         const onSubmit = vi.fn();
-        const { stdin, lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: authQuestion, onSubmit: onSubmit, onCancel: vi.fn(), width: 120 }), { width: 120 });
+        const { stdin, lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: authQuestion, onSubmit: onSubmit, onCancel: vi.fn(), width: 120 }), { width: 120 });
         // Move down to custom option
         writeKey(stdin, '\x1b[B');
         writeKey(stdin, '\x1b[B');
-        await waitFor(() => {
+        await waitFor(async () => {
+            await waitUntilReady();
             expect(lastFrame()).toContain('Enter a custom value');
         });
         // Type directly (inline)
         for (const char of 'API Key') {
             writeKey(stdin, char);
         }
-        await waitFor(() => {
+        await waitFor(async () => {
+            await waitUntilReady();
             expect(lastFrame()).toContain('API Key');
         });
         // Press Enter to submit the custom value
         writeKey(stdin, '\r');
-        await waitFor(() => {
+        await waitFor(async () => {
             expect(onSubmit).toHaveBeenCalledWith({ '0': 'API Key' });
         });
     });
@@ -138,7 +141,7 @@ describe('AskUserDialog', () => {
             },
         ];
         const onSubmit = vi.fn();
-        const { stdin, lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: authQuestionWithOther, onSubmit: onSubmit, onCancel: vi.fn(), width: 120 }), { width: 120 });
+        const { stdin, lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: authQuestionWithOther, onSubmit: onSubmit, onCancel: vi.fn(), width: 120 }), { width: 120 });
         // Navigate to "Other" option
         writeKey(stdin, '\x1b[B'); // Down to "Other"
         // Type first line
@@ -152,13 +155,15 @@ describe('AskUserDialog', () => {
         for (const char of 'Line 2') {
             writeKey(stdin, char);
         }
-        await waitFor(() => {
+        await waitFor(async () => {
+            await waitUntilReady();
             expect(lastFrame()).toContain('Line 1');
+            await waitUntilReady();
             expect(lastFrame()).toContain('Line 2');
         });
         // Press Enter to submit
         writeKey(stdin, '\r');
-        await waitFor(() => {
+        await waitFor(async () => {
             expect(onSubmit).toHaveBeenCalledWith({ '0': 'Line 1\nLine 2' });
         });
     });
@@ -179,38 +184,46 @@ describe('AskUserDialog', () => {
                     multiSelect: false,
                 },
             ];
-            const { lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 80, availableHeight: 10 }), { useAlternateBuffer });
-            await waitFor(() => {
+            const { lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 80, availableHeight: 10 }), { useAlternateBuffer });
+            await waitFor(async () => {
                 if (expectedArrows) {
+                    await waitUntilReady();
                     expect(lastFrame()).toContain('▲');
+                    await waitUntilReady();
                     expect(lastFrame()).toContain('▼');
                 }
                 else {
+                    await waitUntilReady();
                     expect(lastFrame()).not.toContain('▲');
+                    await waitUntilReady();
                     expect(lastFrame()).not.toContain('▼');
                 }
+                await waitUntilReady();
                 expect(lastFrame()).toMatchSnapshot();
             });
         });
     });
     it('navigates to custom option when typing unbound characters (Type-to-Jump)', async () => {
-        const { stdin, lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: authQuestion, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+        const { stdin, lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: authQuestion, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
         // Type a character without navigating down
         writeKey(stdin, 'A');
-        await waitFor(() => {
+        await waitFor(async () => {
             // Should show the custom input with 'A'
             // Placeholder is hidden when text is present
+            await waitUntilReady();
             expect(lastFrame()).toContain('A');
+            await waitUntilReady();
             expect(lastFrame()).toContain('3.  A');
         });
         // Continue typing
         writeKey(stdin, 'P');
         writeKey(stdin, 'I');
-        await waitFor(() => {
+        await waitFor(async () => {
+            await waitUntilReady();
             expect(lastFrame()).toContain('API');
         });
     });
-    it('shows progress header for multiple questions', () => {
+    it('shows progress header for multiple questions', async () => {
         const multiQuestions = [
             {
                 question: 'Which database should we use?',
@@ -233,15 +246,18 @@ describe('AskUserDialog', () => {
                 multiSelect: false,
             },
         ];
-        const { lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: multiQuestions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+        const { lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: multiQuestions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+        await waitUntilReady();
         expect(lastFrame()).toMatchSnapshot();
     });
-    it('hides progress header for single question', () => {
-        const { lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: authQuestion, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+    it('hides progress header for single question', async () => {
+        const { lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: authQuestion, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+        await waitUntilReady();
         expect(lastFrame()).toMatchSnapshot();
     });
-    it('shows keyboard hints', () => {
-        const { lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: authQuestion, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+    it('shows keyboard hints', async () => {
+        const { lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: authQuestion, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+        await waitUntilReady();
         expect(lastFrame()).toMatchSnapshot();
     });
     it('navigates between questions with arrow keys', async () => {
@@ -263,14 +279,17 @@ describe('AskUserDialog', () => {
                 multiSelect: false,
             },
         ];
-        const { stdin, lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: multiQuestions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+        const { stdin, lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: multiQuestions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+        await waitUntilReady();
         expect(lastFrame()).toContain('Which testing framework?');
         writeKey(stdin, '\x1b[C'); // Right arrow
-        await waitFor(() => {
+        await waitFor(async () => {
+            await waitUntilReady();
             expect(lastFrame()).toContain('Which CI provider?');
         });
         writeKey(stdin, '\x1b[D'); // Left arrow
-        await waitFor(() => {
+        await waitFor(async () => {
+            await waitUntilReady();
             expect(lastFrame()).toContain('Which testing framework?');
         });
     });
@@ -292,34 +311,38 @@ describe('AskUserDialog', () => {
             },
         ];
         const onSubmit = vi.fn();
-        const { stdin, lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: multiQuestions, onSubmit: onSubmit, onCancel: vi.fn(), width: 120 }), { width: 120 });
+        const { stdin, lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: multiQuestions, onSubmit: onSubmit, onCancel: vi.fn(), width: 120 }), { width: 120 });
         // Answer first question (should auto-advance)
         writeKey(stdin, '\r');
-        await waitFor(() => {
+        await waitFor(async () => {
+            await waitUntilReady();
             expect(lastFrame()).toContain('Which bundler?');
         });
         // Navigate back
         writeKey(stdin, '\x1b[D');
-        await waitFor(() => {
+        await waitFor(async () => {
+            await waitUntilReady();
             expect(lastFrame()).toContain('Which package manager?');
         });
         // Navigate forward
         writeKey(stdin, '\x1b[C');
-        await waitFor(() => {
+        await waitFor(async () => {
+            await waitUntilReady();
             expect(lastFrame()).toContain('Which bundler?');
         });
         // Answer second question
         writeKey(stdin, '\r');
-        await waitFor(() => {
+        await waitFor(async () => {
+            await waitUntilReady();
             expect(lastFrame()).toContain('Review your answers:');
         });
         // Submit from Review
         writeKey(stdin, '\r');
-        await waitFor(() => {
+        await waitFor(async () => {
             expect(onSubmit).toHaveBeenCalledWith({ '0': 'pnpm', '1': 'Vite' });
         });
     });
-    it('shows Review tab in progress header for multiple questions', () => {
+    it('shows Review tab in progress header for multiple questions', async () => {
         const multiQuestions = [
             {
                 question: 'Which framework?',
@@ -342,7 +365,8 @@ describe('AskUserDialog', () => {
                 multiSelect: false,
             },
         ];
-        const { lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: multiQuestions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+        const { lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: multiQuestions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+        await waitUntilReady();
         expect(lastFrame()).toMatchSnapshot();
     });
     it('allows navigating to Review tab and back', async () => {
@@ -362,17 +386,20 @@ describe('AskUserDialog', () => {
                 multiSelect: false,
             },
         ];
-        const { stdin, lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: multiQuestions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+        const { stdin, lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: multiQuestions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
         writeKey(stdin, '\x1b[C'); // Right arrow
-        await waitFor(() => {
+        await waitFor(async () => {
+            await waitUntilReady();
             expect(lastFrame()).toContain('Add documentation?');
         });
         writeKey(stdin, '\x1b[C'); // Right arrow to Review
-        await waitFor(() => {
+        await waitFor(async () => {
+            await waitUntilReady();
             expect(lastFrame()).toMatchSnapshot();
         });
         writeKey(stdin, '\x1b[D'); // Left arrow back
-        await waitFor(() => {
+        await waitFor(async () => {
+            await waitUntilReady();
             expect(lastFrame()).toContain('Add documentation?');
         });
     });
@@ -393,11 +420,12 @@ describe('AskUserDialog', () => {
                 multiSelect: false,
             },
         ];
-        const { stdin, lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: multiQuestions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+        const { stdin, lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: multiQuestions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
         // Navigate directly to Review tab without answering
         writeKey(stdin, '\x1b[C');
         writeKey(stdin, '\x1b[C');
-        await waitFor(() => {
+        await waitFor(async () => {
+            await waitUntilReady();
             expect(lastFrame()).toMatchSnapshot();
         });
     });
@@ -426,12 +454,12 @@ describe('AskUserDialog', () => {
         writeKey(stdin, '\x1b[C');
         // Submit
         writeKey(stdin, '\r');
-        await waitFor(() => {
+        await waitFor(async () => {
             expect(onSubmit).toHaveBeenCalledWith({ '0': 'Node 20' });
         });
     });
     describe('Text type questions', () => {
-        it('renders text input for type: "text"', () => {
+        it('renders text input for type: "text"', async () => {
             const textQuestion = [
                 {
                     question: 'What should we name this component?',
@@ -440,10 +468,11 @@ describe('AskUserDialog', () => {
                     placeholder: 'e.g., UserProfileCard',
                 },
             ];
-            const { lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: textQuestion, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+            const { lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: textQuestion, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+            await waitUntilReady();
             expect(lastFrame()).toMatchSnapshot();
         });
-        it('shows default placeholder when none provided', () => {
+        it('shows default placeholder when none provided', async () => {
             const textQuestion = [
                 {
                     question: 'Enter the database connection string:',
@@ -451,7 +480,8 @@ describe('AskUserDialog', () => {
                     type: QuestionType.TEXT,
                 },
             ];
-            const { lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: textQuestion, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+            const { lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: textQuestion, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+            await waitUntilReady();
             expect(lastFrame()).toMatchSnapshot();
         });
         it('supports backspace in text mode', async () => {
@@ -462,20 +492,23 @@ describe('AskUserDialog', () => {
                     type: QuestionType.TEXT,
                 },
             ];
-            const { stdin, lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: textQuestion, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+            const { stdin, lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: textQuestion, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
             for (const char of 'abc') {
                 writeKey(stdin, char);
             }
-            await waitFor(() => {
+            await waitFor(async () => {
+                await waitUntilReady();
                 expect(lastFrame()).toContain('abc');
             });
             writeKey(stdin, '\x7f'); // Backspace
-            await waitFor(() => {
+            await waitFor(async () => {
+                await waitUntilReady();
                 expect(lastFrame()).toContain('ab');
+                await waitUntilReady();
                 expect(lastFrame()).not.toContain('abc');
             });
         });
-        it('shows correct keyboard hints for text type', () => {
+        it('shows correct keyboard hints for text type', async () => {
             const textQuestion = [
                 {
                     question: 'Enter the variable name:',
@@ -483,7 +516,8 @@ describe('AskUserDialog', () => {
                     type: QuestionType.TEXT,
                 },
             ];
-            const { lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: textQuestion, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+            const { lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: textQuestion, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+            await waitUntilReady();
             expect(lastFrame()).toMatchSnapshot();
         });
         it('preserves text answer when navigating between questions', async () => {
@@ -504,16 +538,18 @@ describe('AskUserDialog', () => {
                     multiSelect: false,
                 },
             ];
-            const { stdin, lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: mixedQuestions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+            const { stdin, lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: mixedQuestions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
             for (const char of 'useAuth') {
                 writeKey(stdin, char);
             }
             writeKey(stdin, '\t'); // Use Tab instead of Right arrow when text input is active
-            await waitFor(() => {
+            await waitFor(async () => {
+                await waitUntilReady();
                 expect(lastFrame()).toContain('Should it be async?');
             });
             writeKey(stdin, '\x1b[D'); // Left arrow should work when NOT focusing a text input
-            await waitFor(() => {
+            await waitFor(async () => {
+                await waitUntilReady();
                 expect(lastFrame()).toContain('useAuth');
             });
         });
@@ -537,24 +573,30 @@ describe('AskUserDialog', () => {
                 },
             ];
             const onSubmit = vi.fn();
-            const { stdin, lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: mixedQuestions, onSubmit: onSubmit, onCancel: vi.fn(), width: 120 }), { width: 120 });
+            const { stdin, lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: mixedQuestions, onSubmit: onSubmit, onCancel: vi.fn(), width: 120 }), { width: 120 });
             for (const char of 'DataTable') {
                 writeKey(stdin, char);
             }
             writeKey(stdin, '\r');
-            await waitFor(() => {
+            await waitFor(async () => {
+                await waitUntilReady();
                 expect(lastFrame()).toContain('Which styling approach?');
             });
             writeKey(stdin, '\r');
-            await waitFor(() => {
+            await waitFor(async () => {
+                await waitUntilReady();
                 expect(lastFrame()).toContain('Review your answers:');
+                await waitUntilReady();
                 expect(lastFrame()).toContain('Name');
+                await waitUntilReady();
                 expect(lastFrame()).toContain('DataTable');
+                await waitUntilReady();
                 expect(lastFrame()).toContain('Style');
+                await waitUntilReady();
                 expect(lastFrame()).toContain('CSS Modules');
             });
             writeKey(stdin, '\r');
-            await waitFor(() => {
+            await waitFor(async () => {
                 expect(onSubmit).toHaveBeenCalledWith({
                     '0': 'DataTable',
                     '1': 'CSS Modules',
@@ -572,7 +614,7 @@ describe('AskUserDialog', () => {
             const onSubmit = vi.fn();
             const { stdin } = renderWithProviders(_jsx(AskUserDialog, { questions: textQuestion, onSubmit: onSubmit, onCancel: vi.fn(), width: 120 }), { width: 120 });
             writeKey(stdin, '\r');
-            await waitFor(() => {
+            await waitFor(async () => {
                 expect(onSubmit).toHaveBeenCalledWith({});
             });
         });
@@ -585,18 +627,21 @@ describe('AskUserDialog', () => {
                 },
             ];
             const onCancel = vi.fn();
-            const { stdin, lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: textQuestion, onSubmit: vi.fn(), onCancel: onCancel, width: 120 }), { width: 120 });
+            const { stdin, lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: textQuestion, onSubmit: vi.fn(), onCancel: onCancel, width: 120 }), { width: 120 });
             for (const char of 'SomeText') {
                 writeKey(stdin, char);
             }
-            await waitFor(() => {
+            await waitFor(async () => {
+                await waitUntilReady();
                 expect(lastFrame()).toContain('SomeText');
             });
             // Send Ctrl+C
             writeKey(stdin, '\x03'); // Ctrl+C
-            await waitFor(() => {
+            await waitFor(async () => {
                 // Text should be cleared
+                await waitUntilReady();
                 expect(lastFrame()).not.toContain('SomeText');
+                await waitUntilReady();
                 expect(lastFrame()).toContain('>');
             });
             // Should NOT call onCancel (dialog should stay open)
@@ -617,31 +662,36 @@ describe('AskUserDialog', () => {
                     type: QuestionType.TEXT,
                 },
             ];
-            const { stdin, lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: multiQuestions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
+            const { stdin, lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: multiQuestions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120 }), { width: 120 });
             // 1. Move to Text Q (Right arrow works for Choice Q)
             writeKey(stdin, '\x1b[C');
-            await waitFor(() => {
+            await waitFor(async () => {
+                await waitUntilReady();
                 expect(lastFrame()).toContain('Text Q?');
             });
             // 2. Type something in Text Q to make isEditingCustomOption true
             writeKey(stdin, 'a');
-            await waitFor(() => {
+            await waitFor(async () => {
+                await waitUntilReady();
                 expect(lastFrame()).toContain('a');
             });
             // 3. Move back to Choice Q (Left arrow works because cursor is at left edge)
             // When typing 'a', cursor is at index 1.
             // We need to move cursor to index 0 first for Left arrow to work for navigation.
             writeKey(stdin, '\x1b[D'); // Left arrow moves cursor to index 0
-            await waitFor(() => {
+            await waitFor(async () => {
+                await waitUntilReady();
                 expect(lastFrame()).toContain('Text Q?');
             });
             writeKey(stdin, '\x1b[D'); // Second Left arrow should now trigger navigation
-            await waitFor(() => {
+            await waitFor(async () => {
+                await waitUntilReady();
                 expect(lastFrame()).toContain('Choice Q?');
             });
             // 4. Immediately try Right arrow to go back to Text Q
             writeKey(stdin, '\x1b[C');
-            await waitFor(() => {
+            await waitFor(async () => {
+                await waitUntilReady();
                 expect(lastFrame()).toContain('Text Q?');
             });
         });
@@ -663,24 +713,26 @@ describe('AskUserDialog', () => {
                 },
             ];
             const onSubmit = vi.fn();
-            const { stdin, lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: multiQuestions, onSubmit: onSubmit, onCancel: vi.fn(), width: 120 }), { width: 120 });
+            const { stdin, lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: multiQuestions, onSubmit: onSubmit, onCancel: vi.fn(), width: 120 }), { width: 120 });
             // Answer Q1 and Q2 sequentialy
             act(() => {
                 stdin.write('\r'); // Select A1 for Q1 -> triggers autoAdvance
             });
-            await waitFor(() => {
+            await waitFor(async () => {
+                await waitUntilReady();
                 expect(lastFrame()).toContain('Question 2?');
             });
             act(() => {
                 stdin.write('\r'); // Select A2 for Q2 -> triggers autoAdvance to Review
             });
-            await waitFor(() => {
+            await waitFor(async () => {
+                await waitUntilReady();
                 expect(lastFrame()).toContain('Review your answers:');
             });
             act(() => {
                 stdin.write('\r'); // Submit from Review
             });
-            await waitFor(() => {
+            await waitFor(async () => {
                 expect(onSubmit).toHaveBeenCalledWith({
                     '0': 'A1',
                     '1': 'A2',
@@ -699,8 +751,9 @@ describe('AskUserDialog', () => {
                     multiSelect: false,
                 },
             ];
-            const { lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120, availableHeight: 40 }), { width: 120 });
-            await waitFor(() => {
+            const { lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120, availableHeight: 40 }), { width: 120 });
+            await waitFor(async () => {
+                await waitUntilReady();
                 const frame = lastFrame();
                 // Plain text should be rendered as bold
                 expect(frame).toContain(chalk.bold('Which option do you prefer?'));
@@ -716,8 +769,9 @@ describe('AskUserDialog', () => {
                     multiSelect: false,
                 },
             ];
-            const { lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120, availableHeight: 40 }), { width: 120 });
-            await waitFor(() => {
+            const { lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120, availableHeight: 40 }), { width: 120 });
+            await waitFor(async () => {
+                await waitUntilReady();
                 const frame = lastFrame();
                 // Should NOT have double-bold (the whole question bolded AND "this" bolded)
                 // "Is " should not be bold, only "this" should be bold
@@ -736,8 +790,9 @@ describe('AskUserDialog', () => {
                     multiSelect: false,
                 },
             ];
-            const { lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120, availableHeight: 40 }), { width: 120 });
-            await waitFor(() => {
+            const { lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120, availableHeight: 40 }), { width: 120 });
+            await waitFor(async () => {
+                await waitUntilReady();
                 const frame = lastFrame();
                 // Check for chalk.bold('this') - asterisks should be gone, text should be bold
                 expect(frame).toContain(chalk.bold('this'));
@@ -754,8 +809,9 @@ describe('AskUserDialog', () => {
                     multiSelect: false,
                 },
             ];
-            const { lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120, availableHeight: 40 }), { width: 120 });
-            await waitFor(() => {
+            const { lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 120, availableHeight: 40 }), { width: 120 });
+            await waitFor(async () => {
+                await waitUntilReady();
                 const frame = lastFrame();
                 // Backticks should be removed
                 expect(frame).toContain('npm start');
@@ -763,7 +819,7 @@ describe('AskUserDialog', () => {
             });
         });
     });
-    it('uses availableTerminalHeight from UIStateContext if availableHeight prop is missing', () => {
+    it('uses availableTerminalHeight from UIStateContext if availableHeight prop is missing', async () => {
         const questions = [
             {
                 question: 'Choose an option',
@@ -779,12 +835,14 @@ describe('AskUserDialog', () => {
         const mockUIState = {
             availableTerminalHeight: 5, // Small height to force scroll arrows
         };
-        const { lastFrame } = renderWithProviders(_jsx(UIStateContext.Provider, { value: mockUIState, children: _jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 80 }) }), { useAlternateBuffer: false });
+        const { lastFrame, waitUntilReady } = renderWithProviders(_jsx(UIStateContext.Provider, { value: mockUIState, children: _jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 80 }) }), { useAlternateBuffer: false });
         // With height 5 and alternate buffer disabled, it should show scroll arrows (▲)
+        await waitUntilReady();
         expect(lastFrame()).toContain('▲');
+        await waitUntilReady();
         expect(lastFrame()).toContain('▼');
     });
-    it('does NOT truncate the question when in alternate buffer mode even with small height', () => {
+    it('does NOT truncate the question when in alternate buffer mode even with small height', async () => {
         const longQuestion = 'This is a very long question ' + 'with many words '.repeat(10);
         const questions = [
             {
@@ -798,10 +856,12 @@ describe('AskUserDialog', () => {
         const mockUIState = {
             availableTerminalHeight: 5,
         };
-        const { lastFrame } = renderWithProviders(_jsx(UIStateContext.Provider, { value: mockUIState, children: _jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 40 }) }), { useAlternateBuffer: true });
+        const { lastFrame, waitUntilReady } = renderWithProviders(_jsx(UIStateContext.Provider, { value: mockUIState, children: _jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 40 }) }), { useAlternateBuffer: true });
         // Should NOT contain the truncation message
+        await waitUntilReady();
         expect(lastFrame()).not.toContain('hidden ...');
         // Should contain the full long question (or at least its parts)
+        await waitUntilReady();
         expect(lastFrame()).toContain('This is a very long question');
     });
     describe('Choice question placeholder', () => {
@@ -819,11 +879,12 @@ describe('AskUserDialog', () => {
                     multiSelect: false,
                 },
             ];
-            const { stdin, lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 80 }), { width: 80 });
+            const { stdin, lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 80 }), { width: 80 });
             // Navigate to the "Other" option
             writeKey(stdin, '\x1b[B'); // Down
             writeKey(stdin, '\x1b[B'); // Down to Other
-            await waitFor(() => {
+            await waitFor(async () => {
+                await waitUntilReady();
                 expect(lastFrame()).toMatchSnapshot();
             });
         });
@@ -840,11 +901,12 @@ describe('AskUserDialog', () => {
                     multiSelect: false,
                 },
             ];
-            const { stdin, lastFrame } = renderWithProviders(_jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 80 }), { width: 80 });
+            const { stdin, lastFrame, waitUntilReady } = renderWithProviders(_jsx(AskUserDialog, { questions: questions, onSubmit: vi.fn(), onCancel: vi.fn(), width: 80 }), { width: 80 });
             // Navigate to the "Other" option
             writeKey(stdin, '\x1b[B'); // Down
             writeKey(stdin, '\x1b[B'); // Down to Other
-            await waitFor(() => {
+            await waitFor(async () => {
+                await waitUntilReady();
                 expect(lastFrame()).toMatchSnapshot();
             });
         });