test(driver): wait for wasm db sleep recovery

Author: NathanFlurry

rivetkit-typescript/packages/rivetkit/tests/driver/actor-db.test.ts

@@ -13,6 +13,7 @@ const CHUNK_SIZE = 4096;
 const LARGE_PAYLOAD_SIZE = 32768;
 const HIGH_VOLUME_COUNT = 1000;
 const SLEEP_WAIT_MS = 150;
+const WASM_REMOTE_SLEEP_RECOVERY_TIMEOUT_MS = 30_000;
 const LIFECYCLE_POLL_INTERVAL_MS = 25;
 const LIFECYCLE_POLL_ATTEMPTS = 40;
 const REAL_TIMER_DB_TIMEOUT_MS = 180_000;
@@ -79,7 +80,7 @@ async function runWithActorStoppingRetry(
 }
 
 async function expectIntegrityCheckOk(
-	_driverTestConfig: DriverTestConfig,
+	driverTestConfig: DriverTestConfig,
 	integrityCheck: () => Promise<string>,
 ): Promise<void> {
 	// Same lifecycle window as `runWithActorStoppingRetry`: the integrity
@@ -96,7 +97,13 @@ async function expectIntegrityCheckOk(
 				throw error;
 			}
 		},
-		{ timeout: SLEEP_WAIT_MS * 8, interval: 100 },
+		{
+			timeout:
+				driverTestConfig.runtime === "wasm"
+					? WASM_REMOTE_SLEEP_RECOVERY_TIMEOUT_MS
+					: SLEEP_WAIT_MS * 8,
+			interval: 100,
+		},
 	);
 }
 
@@ -276,6 +283,7 @@ describeDriverMatrix("Actor Db", (driverTestConfig) => {
 						expect(
 							sleepEvent.timestamp - sleepRequestedAt,
 						).toBeLessThanOrEqual(SLEEP_OBSERVER_TIMEOUT_MS);
+						await waitFor(driverTestConfig, SLEEP_WAIT_MS);
 
 						const countAfterWake = await actor.getCount();
 						expect(countAfterWake).toBe(baselineCount);
@@ -527,6 +535,7 @@ describeDriverMatrix("Actor Db", (driverTestConfig) => {
 					);
 
 					await actor.triggerSleep();
+					await waitFor(driverTestConfig, SLEEP_WAIT_MS);
 					await expectIntegrityCheckOk(
 						driverTestConfig,
 						async () => await actor.integrityCheck(),