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

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

@@ -5,6 +5,7 @@
  */
 
 import { renderWithProviders } from '../../test-utils/render.js';
+import { waitFor } from '../../test-utils/async.js';
 import { act } from 'react';
 import { vi } from 'vitest';
 import { FolderTrustDialog } from './FolderTrustDialog.js';
@@ -54,12 +55,12 @@ describe('FolderTrustDialog', () => {
       stdin.write('\u001b[27u'); // Press kitty escape key
     });
 
-    await vi.waitFor(() => {
+    await waitFor(() => {
       expect(lastFrame()).toContain(
         'A folder trust level must be selected to continue. Exiting since escape was pressed.',
       );
     });
-    await vi.waitFor(() => {
+    await waitFor(() => {
       expect(mockedExit).toHaveBeenCalledWith(1);
     });
     expect(onSelect).not.toHaveBeenCalled();
@@ -93,7 +94,7 @@ describe('FolderTrustDialog', () => {
       stdin.write('r');
     });
 
-    await vi.waitFor(() => {
+    await waitFor(() => {
       expect(mockedExit).not.toHaveBeenCalled();
     });
   });