From 64a10e64e40ba07bba07b008e82df11fed8d1fb4 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Wed, 18 Feb 2026 17:22:18 +0000
Subject: [PATCH] perf(test): replace reconnect polling sleeps with waitFor

---
 ...onnects-after-connection-close.e2e.test.ts | 23 +++++++++++--------
 1 file changed, 13 insertions(+), 10 deletions(-)

diff --git a/src/web/auto-reply.web-auto-reply.reconnects-after-connection-close.e2e.test.ts b/src/web/auto-reply.web-auto-reply.reconnects-after-connection-close.e2e.test.ts
index 3b69c123fb..2c677cd890 100644
--- a/src/web/auto-reply.web-auto-reply.reconnects-after-connection-close.e2e.test.ts
+++ b/src/web/auto-reply.web-auto-reply.reconnects-after-connection-close.e2e.test.ts
@@ -105,19 +105,18 @@ describe("web auto-reply", () => {
     expect(listenerFactory).toHaveBeenCalledTimes(1);
 
     closeResolvers[0]?.();
-    const waitForSecondCall = async () => {
-      const started = Date.now();
-      while (listenerFactory.mock.calls.length < 2 && Date.now() - started < 200) {
-        await new Promise((resolve) => setTimeout(resolve, 10));
-      }
-    };
-    await waitForSecondCall();
+    await vi.waitFor(
+      () => {
+        expect(listenerFactory).toHaveBeenCalledTimes(2);
+      },
+      { timeout: 500, interval: 5 },
+    );
     expect(listenerFactory).toHaveBeenCalledTimes(2);
     expect(runtime.error).toHaveBeenCalledWith(expect.stringContaining("Retry 1"));
 
     controller.abort();
     closeResolvers[1]?.();
-    await new Promise((resolve) => setTimeout(resolve, 5));
+    await Promise.resolve();
     await run;
   });
   it("forces reconnect when watchdog closes without onClose", async () => {
@@ -219,11 +218,15 @@ describe("web auto-reply", () => {
     expect(listenerFactory).toHaveBeenCalledTimes(1);
 
     closeResolvers.shift()?.();
-    await new Promise((resolve) => setTimeout(resolve, 15));
+    await vi.waitFor(
+      () => {
+        expect(listenerFactory).toHaveBeenCalledTimes(2);
+      },
+      { timeout: 500, interval: 5 },
+    );
     expect(listenerFactory).toHaveBeenCalledTimes(2);
 
     closeResolvers.shift()?.();
-    await new Promise((resolve) => setTimeout(resolve, 15));
     await run;
 
     expect(runtime.error).toHaveBeenCalledWith(expect.stringContaining("max attempts reached"));