From 656ae95263cfd9f832bdeefbf9da6a91b8ce72bb Mon Sep 17 00:00:00 2001
From: Tianyi Cui <53024+tianyicui@users.noreply.github.com>
Date: Tue, 14 Jul 2026 10:28:04 +0800
Subject: [PATCH] test: await exact stdio startup condition

---
 packages/ui/stdio-agent/tests/stdio-agent.spec.ts | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/packages/ui/stdio-agent/tests/stdio-agent.spec.ts b/packages/ui/stdio-agent/tests/stdio-agent.spec.ts
index 9a7bf10bad..a14848a549 100644
--- a/packages/ui/stdio-agent/tests/stdio-agent.spec.ts
+++ b/packages/ui/stdio-agent/tests/stdio-agent.spec.ts
@@ -114,9 +114,8 @@ describe('dsh-stdio-agent app', () => {
     const ctx = new Context()
     // No persona: covers the omitted-persona forwarding branch too.
     stdioAgent.apply(ctx, { model: 'mock', skills: await isolatedSkillsConfig() })
-    await new Promise(resolve => setTimeout(resolve, 80))
-    expect(ctx.get('sessionPersistence')).toBeDefined()
-    expect(ctx.get('agents')?.list()).toHaveLength(1)
+    await expect.poll(() => ctx.get('sessionPersistence')).toBeDefined()
+    await expect.poll(() => ctx.get('agents')?.list()).toHaveLength(1)
     await ctx.fiber.dispose()
   })