Use wait helper

Author: steida

packages/common/test/WebSocket.test.ts

@@ -1,6 +1,6 @@
 import { afterEach, beforeEach, expect, test, vi } from "vitest";
 import WebSocket, { WebSocketServer } from "ws";
-import { RetryError } from "../src/Promise.js";
+import { RetryError, wait } from "../src/Promise.js";
 import { err, ok } from "../src/Result.js";
 import {
   createWebSocket,
@@ -299,7 +299,7 @@ test("cleans up all resources when dispose is called", async () => {
   expect(sendResult).toEqual(err({ type: "WebSocketSendError" }));
 
   // Wait a bit to ensure no reconnection attempts
-  await new Promise((resolve) => setTimeout(resolve, 500));
+  await wait(500);
 
   // No additional onOpen calls should happen after disposal
   expect(onOpen).toHaveBeenCalledTimes(1);
@@ -436,7 +436,7 @@ test("aborts connection attempts when disposed", async () => {
   const callCountAtDispose = onRetry.mock.calls.length;
 
   // Wait some time to ensure no more retries happen
-  await new Promise((resolve) => setTimeout(resolve, 500));
+  await wait(500);
 
   // Verify no additional retry calls happened after dispose
   expect(onRetry.mock.calls.length).toBe(callCountAtDispose);

packages/web/test/SharedWebWorker.test.ts

@@ -4,7 +4,7 @@
 /* eslint-disable @typescript-eslint/no-unsafe-assignment */
 import { expect, test, vi, beforeEach, afterEach } from "vitest";
 import { createSharedWebWorker } from "../src/SharedWebWorker.js";
-import { getOrThrow, SimpleName } from "@evolu/common";
+import { getOrThrow, SimpleName, wait } from "@evolu/common";
 
 // Mock BroadcastChannel
 class MockBroadcastChannel {
@@ -231,7 +231,7 @@ test("createSharedWebWorker handles multiple tabs - first tab becomes owner", as
   });
 
   // Wait for lock acquisition
-  await new Promise((resolve) => setTimeout(resolve, 10));
+  await wait(10);
 
   // Only first tab should create worker (it became owner)
   expect(mockCreateWorker).toHaveBeenCalledTimes(1);