Fix tests to wrap all calls changing the UI with act. (#12268)

packages/cli/src/ui/components/ProQuotaDialog.test.tsx

@@ -4,7 +4,8 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import { render } from 'ink-testing-library';
+import { render } from '../../test-utils/render.js';
+import { act } from 'react';
 import { describe, it, expect, vi, beforeEach, type Mock } from 'vitest';
 import { ProQuotaDialog } from './ProQuotaDialog.js';
 import { RadioButtonSelect } from './shared/RadioButtonSelect.js';
@@ -20,7 +21,7 @@ describe('ProQuotaDialog', () => {
   });
 
   it('should render with correct title and options', () => {
-    const { lastFrame } = render(
+    const { lastFrame, unmount } = render(
       <ProQuotaDialog
         failedModel="gemini-2.5-pro"
         fallbackModel="gemini-2.5-flash"
@@ -49,11 +50,12 @@ describe('ProQuotaDialog', () => {
       }),
       undefined,
     );
+    unmount();
   });
 
   it('should call onChoice with "auth" when "Change auth" is selected', () => {
     const mockOnChoice = vi.fn();
-    render(
+    const { unmount } = render(
       <ProQuotaDialog
         failedModel="gemini-2.5-pro"
         fallbackModel="gemini-2.5-flash"
@@ -65,14 +67,17 @@ describe('ProQuotaDialog', () => {
     const onSelect = (RadioButtonSelect as Mock).mock.calls[0][0].onSelect;
 
     // Simulate the selection
-    onSelect('auth');
+    act(() => {
+      onSelect('auth');
+    });
 
     expect(mockOnChoice).toHaveBeenCalledWith('auth');
+    unmount();
   });
 
   it('should call onChoice with "continue" when "Continue with flash" is selected', () => {
     const mockOnChoice = vi.fn();
-    render(
+    const { unmount } = render(
       <ProQuotaDialog
         failedModel="gemini-2.5-pro"
         fallbackModel="gemini-2.5-flash"
@@ -84,8 +89,11 @@ describe('ProQuotaDialog', () => {
     const onSelect = (RadioButtonSelect as Mock).mock.calls[0][0].onSelect;
 
     // Simulate the selection
-    onSelect('continue');
+    act(() => {
+      onSelect('continue');
+    });
 
     expect(mockOnChoice).toHaveBeenCalledWith('continue');
+    unmount();
   });
 });