diff --git a/extensions/mistral/realtime-transcription-provider.test.ts b/extensions/mistral/realtime-transcription-provider.test.ts index 89a4cbeb8807..56c0abf87e7d 100644 --- a/extensions/mistral/realtime-transcription-provider.test.ts +++ b/extensions/mistral/realtime-transcription-provider.test.ts @@ -378,7 +378,11 @@ describe("buildMistralRealtimeTranscriptionProvider", () => { { type: "transcription.segment", text: "late segment", start: 0, end: 1 }, { type: "transcription.done", text: "late done" }, ]); - const onError = vi.fn(); + let resolveError: (() => void) | undefined; + const errorReceived = new Promise((resolve) => { + resolveError = resolve; + }); + const onError = vi.fn(() => resolveError?.()); const onTranscript = vi.fn(); let lastPartialLength = 0; let partialCalls = 0; @@ -393,6 +397,7 @@ describe("buildMistralRealtimeTranscriptionProvider", () => { }); await session.connect(); + await errorReceived; await vi.waitFor(() => { expect(onError).toHaveBeenCalledExactlyOnceWith( expect.objectContaining({ diff --git a/extensions/voice-call/src/webhook/realtime-handler.test.ts b/extensions/voice-call/src/webhook/realtime-handler.test.ts index e3f779ae3210..2def232f1b47 100644 --- a/extensions/voice-call/src/webhook/realtime-handler.test.ts +++ b/extensions/voice-call/src/webhook/realtime-handler.test.ts @@ -1944,7 +1944,9 @@ describe("RealtimeCallHandler path routing", () => { const oldClosed = waitForClose(oldWs); callbacks[0]?.onClose?.("error"); await oldClosed; - expect(oldCloseBridge).toHaveBeenCalledOnce(); + await waitForRealtimeTest(() => { + expect(oldCloseBridge).toHaveBeenCalledOnce(); + }); expect(replacementCloseBridge).not.toHaveBeenCalled(); expect(hangupCall).not.toHaveBeenCalled(); expect(