diff --git a/rivetkit-typescript/packages/rivetkit/fixtures/driver-test-suite/sleep-db.ts b/rivetkit-typescript/packages/rivetkit/fixtures/driver-test-suite/sleep-db.ts index 92cee78db8..ec2232c493 100644 --- a/rivetkit-typescript/packages/rivetkit/fixtures/driver-test-suite/sleep-db.ts +++ b/rivetkit-typescript/packages/rivetkit/fixtures/driver-test-suite/sleep-db.ts @@ -1273,6 +1273,10 @@ export const sleepWsConcurrentDbExceedsGrace = actor({ // After the delay, the VFS may be destroyed. const dbRef = c.db; + await dbRef.execute( + `INSERT INTO sleep_log (event, created_at) VALUES ('handler-${index}-start', ${Date.now()})`, + ); + ws.send(JSON.stringify({ type: "started", index })); // Stagger the delay slightly per index so handlers resume at diff --git a/rivetkit-typescript/packages/rivetkit/tests/driver/actor-sleep-db.test.ts b/rivetkit-typescript/packages/rivetkit/tests/driver/actor-sleep-db.test.ts index 09dc1bbe70..be2def8739 100644 --- a/rivetkit-typescript/packages/rivetkit/tests/driver/actor-sleep-db.test.ts +++ b/rivetkit-typescript/packages/rivetkit/tests/driver/actor-sleep-db.test.ts @@ -1014,8 +1014,7 @@ describeDriverMatrix("Actor Sleep Db", (driverTestConfig) => { { timeout: 15_000 }, ); - // TODO(#4705): Root-cause task-model shutdown ordering for concurrent WebSocket DB handlers and re-enable this coverage. - test.skip( + test( "concurrent ws handlers with cached db ref get errors when grace period exceeded", async (c) => { const { client } = await setupDriverTest(c, driverTestConfig);