test: wait for openai realtime socket

steipete · steipete · commit 166f2bcee476 · 2026-05-11T17:01:27.000+01:00
diff --git a/extensions/openai/realtime-transcription-provider.test.ts b/extensions/openai/realtime-transcription-provider.test.ts
@@ -83,14 +83,15 @@ function parseSent(socket: FakeWebSocketInstance): SentRealtimeEvent[] {
 }
 
 async function waitForFakeSocket(): Promise<FakeWebSocketInstance> {
-  for (let attempt = 0; attempt < 20; attempt += 1) {
-    const socket = FakeWebSocket.instances[0];
-    if (socket) {
-      return socket;
-    }
-    await new Promise((resolve) => setTimeout(resolve, 0));
+  let socket: FakeWebSocketInstance | undefined;
+  await vi.waitFor(() => {
+    socket = FakeWebSocket.instances[0];
+    expect(socket).toBeDefined();
+  });
+  if (!socket) {
+    throw new Error("expected session to create a websocket");
   }
-  throw new Error("expected session to create a websocket");
+  return socket;
 }
 
 function mockCallArg(mock: { mock: { calls: unknown[][] } }, index = 0): Record<string, unknown> {
