fix(rivetkit): fix engine driver tests, engine sqlite teardown logic

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

@@ -275,6 +275,7 @@ export const dbActorDrizzle = actor({
 		},
 	},
 	options: {
+		actionTimeout: 120_000,
 		sleepTimeout: 100,
 	},
 });

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

@@ -281,6 +281,7 @@ export const dbActorRaw = actor({
 		},
 	},
 	options: {
+		actionTimeout: 120_000,
 		sleepTimeout: 100,
 	},
 });

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

@@ -25,6 +25,14 @@ function getCounts(actorId: string): LifecycleCounts {
 	};
 }
 
+function getTotalCleanupCount(): number {
+	let total = 0;
+	for (const count of cleanupCounts.values()) {
+		total += count;
+	}
+	return total;
+}
+
 const baseProvider = db({
 	onMigrate: async (dbHandle) => {
 		await dbHandle.execute(`
@@ -125,5 +133,8 @@ export const dbLifecycleObserver = actor({
 		getCounts: (_c, actorId: string) => {
 			return getCounts(actorId);
 		},
+		getTotalCleanupCount: () => {
+			return getTotalCleanupCount();
+		},
 	},
 });

rivetkit-typescript/packages/rivetkit/src/actor/instance/mod.ts

@@ -486,7 +486,10 @@ export class ActorInstance<
 				this.#sleepTimeout = undefined;
 			}
 
-			// Abort listeners
+			// Abort listeners in the canonical stop path.
+			// This must run for all stop modes, including sleep and remote stop.
+			// Destroy may have already triggered an early abort, but repeating abort
+			// is intentional and safe.
 			try {
 				this.#abortController.abort();
 			} catch { }
@@ -574,6 +577,14 @@ export class ActorInstance<
 		}
 		this.#destroyCalled = true;
 
+		// Abort immediately so in flight waits can exit before the driver stop
+		// handshake completes.
+		// The onStop path will call abort again as a safety net for all stop
+		// modes.
+		try {
+			this.#abortController.abort();
+		} catch {}
+
 		const destroy = this.driver.startDestroy.bind(
 			this.driver,
 			this.#actorId,
@@ -958,7 +969,7 @@ export class ActorInstance<
 	 * Errors are propagated to the caller.
 	 */
 	async keepAwake<T>(promise: Promise<T>): Promise<T> {
-		this.assertReady();
+		this.assertReady(true);
 
 		this.#activeKeepAwakeCount++;
 		this.resetSleepTimer();