fix tui test fails by waiting escape key to work

Author: jesseturner21

package-lock.json

@@ -53,11 +53,12 @@ describe('FullScreenLogView', () => {
     expect(lastFrame()).toContain('No logs yet');
   });
 
-  it('calls onExit on Escape key', () => {
+  it('calls onExit on Escape key', async () => {
     const onExit = vi.fn();
     const { stdin } = render(<FullScreenLogView logs={makeLogs(3)} onExit={onExit} />);
 
     stdin.write(ESCAPE);
+    await new Promise(resolve => setImmediate(resolve));
 
     expect(onExit).toHaveBeenCalledTimes(1);
   });

src/cli/tui/components/__tests__/FullScreenLogView.test.tsx

@@ -49,7 +49,7 @@ describe('PromptScreen', () => {
     expect(onConfirm).toHaveBeenCalledTimes(1);
   });
 
-  it('calls onExit on Escape key', () => {
+  it('calls onExit on Escape key', async () => {
     const onExit = vi.fn();
     const { stdin } = render(
       <PromptScreen helpText="help" onExit={onExit}>
@@ -58,6 +58,7 @@ describe('PromptScreen', () => {
     );
 
     stdin.write(ESCAPE);
+    await new Promise(resolve => setImmediate(resolve));
 
     expect(onExit).toHaveBeenCalledTimes(1);
   });
@@ -200,11 +201,12 @@ describe('ErrorPrompt', () => {
     expect(onBack).toHaveBeenCalledTimes(1);
   });
 
-  it('calls onExit on Escape key', () => {
+  it('calls onExit on Escape key', async () => {
     const onExit = vi.fn();
     const { stdin } = render(<ErrorPrompt message="Failed" onExit={onExit} />);
 
     stdin.write(ESCAPE);
+    await new Promise(resolve => setImmediate(resolve));
 
     expect(onExit).toHaveBeenCalledTimes(1);
   });
@@ -257,11 +259,12 @@ describe('ConfirmPrompt', () => {
     expect(onConfirm).toHaveBeenCalledTimes(1);
   });
 
-  it('calls onCancel on Escape key', () => {
+  it('calls onCancel on Escape key', async () => {
     const onCancel = vi.fn();
     const { stdin } = render(<ConfirmPrompt message="Delete?" onConfirm={vi.fn()} onCancel={onCancel} />);
 
     stdin.write(ESCAPE);
+    await new Promise(resolve => setImmediate(resolve));
 
     expect(onCancel).toHaveBeenCalledTimes(1);
   });

src/cli/tui/components/__tests__/PromptScreen.test.tsx

@@ -49,7 +49,7 @@ describe('Screen', () => {
     expect(lastFrame()).toContain('Press Enter to continue');
   });
 
-  it('calls onExit on Escape key', () => {
+  it('calls onExit on Escape key', async () => {
     const onExit = vi.fn();
     const { stdin } = render(
       <Screen title="Test" onExit={onExit}>
@@ -58,6 +58,7 @@ describe('Screen', () => {
     );
 
     stdin.write(ESCAPE);
+    await new Promise(resolve => setImmediate(resolve));
 
     expect(onExit).toHaveBeenCalledTimes(1);
   });

src/cli/tui/components/__tests__/Screen.test.tsx

@@ -14,11 +14,12 @@ function ExitHandlerHarness({ onExit, enabled }: { onExit: () => void; enabled?:
 }
 
 describe('useExitHandler', () => {
-  it('calls onExit when Escape is pressed', () => {
+  it('calls onExit when Escape is pressed', async () => {
     const onExit = vi.fn();
     const { stdin } = render(<ExitHandlerHarness onExit={onExit} />);
 
     stdin.write(ESCAPE);
+    await new Promise(resolve => setImmediate(resolve));
 
     expect(onExit).toHaveBeenCalledTimes(1);
   });
@@ -53,11 +54,12 @@ describe('useExitHandler', () => {
     expect(onExit).not.toHaveBeenCalled();
   });
 
-  it('enabled defaults to true', () => {
+  it('enabled defaults to true', async () => {
     const onExit = vi.fn();
     const { stdin } = render(<ExitHandlerHarness onExit={onExit} />);
 
     stdin.write(ESCAPE);
+    await new Promise(resolve => setImmediate(resolve));
 
     expect(onExit).toHaveBeenCalledTimes(1);
   });