chore(rivetkit): remove setpreventsleep from tests

Author: NathanFlurry

rivetkit-typescript/packages/rivetkit/fixtures/driver-test-suite/sleep-db.ts

@@ -486,6 +486,8 @@ export const sleepEnqueue = actor({
 	},
 });
 
+const scheduleAfterKeepAwakeResolvers = new Map<string, () => void>();
+
 export const sleepScheduleAfter = actor({
 	state: {
 		startCount: 0,
@@ -508,7 +510,11 @@ export const sleepScheduleAfter = actor({
 		c.state.startCount += 1;
 		if (c.state.holdAfterWake) {
 			// Keep the alarm wake observable before idle sleep can run again.
-			c.setPreventSleep(true);
+			c.keepAwake(
+				new Promise<void>((resolve) => {
+					scheduleAfterKeepAwakeResolvers.set(c.actorId, resolve);
+				}),
+			);
 		}
 		await c.db.execute(
 			`INSERT INTO sleep_log (event, created_at) VALUES ('wake', ${Date.now()})`,
@@ -539,7 +545,8 @@ export const sleepScheduleAfter = actor({
 			};
 			if (c.state.scheduledActionCount > 0) {
 				c.state.holdAfterWake = false;
-				c.setPreventSleep(false);
+				scheduleAfterKeepAwakeResolvers.get(c.actorId)?.();
+				scheduleAfterKeepAwakeResolvers.delete(c.actorId);
 			}
 			return counts;
 		},