From 8394898ef57fdcd6862963007680bae5335344c2 Mon Sep 17 00:00:00 2001 From: Yichen Jiang Date: Tue, 21 Jul 2026 16:50:45 +0800 Subject: [PATCH] test(tui): await async session suggestions --- packages/ui/tui/tests/tui.spec.ts | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/packages/ui/tui/tests/tui.spec.ts b/packages/ui/tui/tests/tui.spec.ts index a4d6179194..7dd6252c77 100644 --- a/packages/ui/tui/tests/tui.spec.ts +++ b/packages/ui/tui/tests/tui.spec.ts @@ -517,17 +517,16 @@ describe('pi-tui chat lifecycle and transcript', () => { }) result.terminal.send('@no-cwd') - await tick() - expect(result.terminal.output).toContain('Session · no-cwd') + await vi.waitFor(() => { expect(result.terminal.output).toContain('Session · no-cwd') }) expect(result.terminal.output).toContain('(no cwd)') result.terminal.send('\x03') result.terminal.send('@source-session') - await tick() + await vi.waitFor(() => { expect(result.terminal.output).toContain('Session · source-session') }) result.terminal.send('\t') await tick() result.terminal.send('\r') - await tick() + await vi.waitFor(() => { expect(result.agent.sent).toHaveLength(1) }) expect(result.agent.sent).toEqual([[{ type: 'text', text: '@source-session' }]]) expect(result.agent.sentOptions[0]?.contexts).toHaveLength(1) @@ -540,7 +539,7 @@ describe('pi-tui chat lifecycle and transcript', () => { result.agent.status = 'running' result.terminal.send(`steer ${mention}`) result.terminal.send('\r') - await tick() + await vi.waitFor(() => { expect(result.agent.steered).toHaveLength(1) }) expect(result.agent.steered).toEqual([[{ type: 'text', text: 'steer @Source chat' }]]) expect(result.agent.steeredOptions[0]?.contexts).toHaveLength(1) await dispose(result)