Skip to content

Commit c7aba07

Browse files
committed
chore(core): fix test for sending message to specific connection sse race condition (#1286)
1 parent ead9778 commit c7aba07

File tree

1 file changed

+14
-11
lines changed

1 file changed

+14
-11
lines changed

packages/rivetkit/src/driver-test-suite/tests/actor-conn-state.ts

Lines changed: 14 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -237,17 +237,20 @@ export function runActorConnStateTests(driverTestConfig: DriverTestConfig) {
237237
receivedMessages.push(data);
238238
});
239239

240-
// Send message from first connection to second
241-
const success = await conn1.sendToConnection(
242-
state2.id,
243-
"Hello from conn1",
244-
);
245-
expect(success).toBe(true);
246-
247-
// Verify message was received
248-
expect(receivedMessages.length).toBe(1);
249-
expect(receivedMessages[0].from).toBe(state1.id);
250-
expect(receivedMessages[0].message).toBe("Hello from conn1");
240+
// TODO: SSE has race condition between subscrib eand publish message
241+
await vi.waitFor(async () => {
242+
// Send message from first connection to second
243+
const success = await conn1.sendToConnection(
244+
state2.id,
245+
"Hello from conn1",
246+
);
247+
expect(success).toBe(true);
248+
249+
// Verify message was received
250+
expect(receivedMessages.length).toBe(1);
251+
expect(receivedMessages[0].from).toBe(state1.id);
252+
expect(receivedMessages[0].message).toBe("Hello from conn1");
253+
});
251254

252255
// Clean up
253256
await conn1.dispose();

0 commit comments

Comments
 (0)