File tree Expand file tree Collapse file tree 1 file changed +14
-11
lines changed
packages/rivetkit/src/driver-test-suite/tests Expand file tree Collapse file tree 1 file changed +14
-11
lines changed Original file line number Diff line number Diff line change @@ -237,17 +237,20 @@ export function runActorConnStateTests(driverTestConfig: DriverTestConfig) {
237
237
receivedMessages . push ( data ) ;
238
238
} ) ;
239
239
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
+ } ) ;
251
254
252
255
// Clean up
253
256
await conn1 . dispose ( ) ;
You can’t perform that action at this time.
0 commit comments