From b2d8815a34b584a8704d2285e33ea45b279fc337 Mon Sep 17 00:00:00 2001
From: imccyu <276526105+imccyu@users.noreply.github.com>
Date: Sun, 26 Jul 2026 02:29:13 +0800
Subject: [PATCH] test(llm-mock-server): poll for the client-closed outcome

The fixed 5ms sleep raced slow CI runners (the server observes the socket
close asynchronously); vi.waitFor polls until the outcome lands.
---
 packages/support/llm-mock-server/tests/server.spec.ts | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/packages/support/llm-mock-server/tests/server.spec.ts b/packages/support/llm-mock-server/tests/server.spec.ts
index 1f15ba1a0b..34db57aa82 100644
--- a/packages/support/llm-mock-server/tests/server.spec.ts
+++ b/packages/support/llm-mock-server/tests/server.spec.ts
@@ -1,5 +1,5 @@
 import { request } from 'node:http'
-import { afterEach, describe, expect, it } from 'vitest'
+import { afterEach, describe, expect, it, vi } from 'vitest'
 import type { MockLlmBehavior, MockLlmServer, MockLlmServerEvent } from '../src/index.ts'
 import { startMockLlmServer } from '../src/index.ts'
 
@@ -179,9 +179,11 @@ describe('mock LLM server wire behaviors', () => {
     const response = await chat(server, { signal: controller.signal })
     controller.abort()
     await expect(response.text()).rejects.toThrow()
-    await new Promise((resolve) => { setTimeout(resolve, 5) })
-
-    expect(server.requests[0]).toMatchObject({ behavior, outcome: 'client_closed' })
+    // The server observes the socket close asynchronously; a fixed sleep
+    // raced slow runners, so poll until the outcome lands.
+    await vi.waitFor(() => {
+      expect(server.requests[0]).toMatchObject({ behavior, outcome: 'client_closed' })
+    })
     expect(events.filter(event => event.type === 'result')).toEqual([
       expect.objectContaining({ behavior, outcome: 'client_closed' }),
     ])