rivetkit 2.0.20 → 2.0.22-rc.1

package/src/driver-test-suite/tests/actor-sleep.ts

@@ -64,7 +64,8 @@ export function runActorSleepTests(driverTestConfig: DriverTestConfig) {
 			// Reconnect to get sleep count after restore
 			const sleepActor2 = client.sleep.getOrCreate();
 			{
-				const { startCount, sleepCount } = await sleepActor2.getCounts();
+				const { startCount, sleepCount } =
+					await sleepActor2.getCounts();
 				expect(sleepCount).toBe(1);
 				expect(startCount).toBe(2);
 			}
@@ -116,7 +117,8 @@ export function runActorSleepTests(driverTestConfig: DriverTestConfig) {
 			// Reconnect to get sleep count after restore
 			const sleepActor2 = client.sleep.getOrCreate();
 			{
-				const { startCount, sleepCount } = await sleepActor2.getCounts();
+				const { startCount, sleepCount } =
+					await sleepActor2.getCounts();
 				expect(sleepCount).toBe(1);
 				expect(startCount).toBe(2);
 			}
@@ -257,7 +259,8 @@ export function runActorSleepTests(driverTestConfig: DriverTestConfig) {
 			// Actor should have slept after the timeout
 			const sleepActor2 = client.sleepWithLongRpc.getOrCreate();
 			{
-				const { startCount, sleepCount } = await sleepActor2.getCounts();
+				const { startCount, sleepCount } =
+					await sleepActor2.getCounts();
 				expect(sleepCount).toBe(1); // Slept once
 				expect(startCount).toBe(2); // New instance after sleep
 			}

package/src/driver-test-suite/tests/actor-vars.ts

@@ -25,8 +25,12 @@ export function runActorVarsTests(driverTestConfig: DriverTestConfig) {
 				const { client } = await setupDriverTest(c, driverTestConfig);
 
 				// Create two separate instances
-				const instance1 = client.nestedVarActor.getOrCreate(["instance1"]);
-				const instance2 = client.nestedVarActor.getOrCreate(["instance2"]);
+				const instance1 = client.nestedVarActor.getOrCreate([
+					"instance1",
+				]);
+				const instance2 = client.nestedVarActor.getOrCreate([
+					"instance2",
+				]);
 
 				// Modify vars in the first instance
 				const modifiedVars = await instance1.modifyNested();

package/src/driver-test-suite/tests/manager-driver.ts

@@ -19,7 +19,10 @@ export function runManagerDriverTests(driverTestConfig: DriverTestConfig) {
 				expect(count).toBe(5);
 
 				// Connect with key creates a new actor with specific parameters
-				const counterB = client.counter.getOrCreate(["counter-b", "testing"]);
+				const counterB = client.counter.getOrCreate([
+					"counter-b",
+					"testing",
+				]);
 
 				await counterB.increment(10);
 				const countB = await counterB.increment(0);
@@ -66,7 +69,8 @@ export function runManagerDriverTests(driverTestConfig: DriverTestConfig) {
 				}
 
 				// Create the actor
-				const createdCounter = client.counter.getOrCreate(nonexistentId);
+				const createdCounter =
+					client.counter.getOrCreate(nonexistentId);
 				await createdCounter.increment(3);
 
 				// Now no create should work since the actor exists
@@ -249,7 +253,9 @@ export function runManagerDriverTests(driverTestConfig: DriverTestConfig) {
 				expect(subsetMatchCount).toBe(0); // Should be a new counter with 0
 
 				// Should NOT match with just one key - should create new actor
-				const singleKeyCounter = client.counter.getOrCreate(["counter-match"]);
+				const singleKeyCounter = client.counter.getOrCreate([
+					"counter-match",
+				]);
 				const singleKeyCount = await singleKeyCounter.increment(0);
 				expect(singleKeyCount).toBe(0); // Should be a new counter with 0
 			});
@@ -258,11 +264,14 @@ export function runManagerDriverTests(driverTestConfig: DriverTestConfig) {
 				const { client } = await setupDriverTest(c, driverTestConfig);
 
 				// Create actor with string key
-				const stringKeyCounter = client.counter.getOrCreate("string-key-test");
+				const stringKeyCounter =
+					client.counter.getOrCreate("string-key-test");
 				await stringKeyCounter.increment(7);
 
 				// Should match with equivalent array key
-				const arrayKeyCounter = client.counter.getOrCreate(["string-key-test"]);
+				const arrayKeyCounter = client.counter.getOrCreate([
+					"string-key-test",
+				]);
 				const count = await arrayKeyCounter.increment(0);
 				expect(count).toBe(7);
 			});
@@ -271,7 +280,8 @@ export function runManagerDriverTests(driverTestConfig: DriverTestConfig) {
 				const { client } = await setupDriverTest(c, driverTestConfig);
 
 				// Create actor with undefined key
-				const undefinedKeyCounter = client.counter.getOrCreate(undefined);
+				const undefinedKeyCounter =
+					client.counter.getOrCreate(undefined);
 				await undefinedKeyCounter.increment(12);
 
 				// Should match with empty array key

package/src/driver-test-suite/tests/raw-http-request-properties.ts

@@ -10,7 +10,9 @@ export function runRawHttpRequestPropertiesTests(
 	describe("raw http request properties", () => {
 		test("should pass all Request properties correctly to onFetch", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpRequestPropertiesActor.getOrCreate(["test"]);
+			const actor = client.rawHttpRequestPropertiesActor.getOrCreate([
+				"test",
+			]);
 
 			// Test basic request properties
 			const response = await actor.fetch("test/path?foo=bar&baz=qux", {
@@ -49,7 +51,9 @@ export function runRawHttpRequestPropertiesTests(
 
 		test("should handle GET requests with no body", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpRequestPropertiesActor.getOrCreate(["test"]);
+			const actor = client.rawHttpRequestPropertiesActor.getOrCreate([
+				"test",
+			]);
 
 			const response = await actor.fetch("test/get", {
 				method: "GET",
@@ -64,7 +68,9 @@ export function runRawHttpRequestPropertiesTests(
 
 		test("should handle different content types", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpRequestPropertiesActor.getOrCreate(["test"]);
+			const actor = client.rawHttpRequestPropertiesActor.getOrCreate([
+				"test",
+			]);
 
 			// Test form data
 			const formData = new URLSearchParams();
@@ -103,7 +109,9 @@ export function runRawHttpRequestPropertiesTests(
 
 		test("should preserve all header casing and values", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpRequestPropertiesActor.getOrCreate(["test"]);
+			const actor = client.rawHttpRequestPropertiesActor.getOrCreate([
+				"test",
+			]);
 
 			const response = await actor.fetch("test/headers", {
 				headers: {
@@ -129,7 +137,9 @@ export function runRawHttpRequestPropertiesTests(
 
 		test("should handle empty and special URL paths", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpRequestPropertiesActor.getOrCreate(["test"]);
+			const actor = client.rawHttpRequestPropertiesActor.getOrCreate([
+				"test",
+			]);
 
 			// Test root path
 			const rootResponse = await actor.fetch("");
@@ -145,7 +155,9 @@ export function runRawHttpRequestPropertiesTests(
 			const specialData = (await specialResponse.json()) as any;
 			// Note: The URL path may or may not be decoded depending on the HTTP client/server
 			// Just verify it contains the expected segments
-			expect(specialData.pathname).toMatch(/path.*with.*spaces.*and.*slashes/);
+			expect(specialData.pathname).toMatch(
+				/path.*with.*spaces.*and.*slashes/,
+			);
 
 			// Test path with fragment (should be ignored in server-side)
 			const fragmentResponse = await actor.fetch("test/path#fragment");
@@ -157,7 +169,9 @@ export function runRawHttpRequestPropertiesTests(
 
 		test("should handle request properties for all HTTP methods", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpRequestPropertiesActor.getOrCreate(["test"]);
+			const actor = client.rawHttpRequestPropertiesActor.getOrCreate([
+				"test",
+			]);
 
 			const methods = [
 				"GET",
@@ -170,13 +184,16 @@ export function runRawHttpRequestPropertiesTests(
 			];
 
 			for (const method of methods) {
-				const response = await actor.fetch(`test/${method.toLowerCase()}`, {
-					method,
-					// Only include body for methods that support it
-					body: ["POST", "PUT", "PATCH"].includes(method)
-						? JSON.stringify({ method })
-						: undefined,
-				});
+				const response = await actor.fetch(
+					`test/${method.toLowerCase()}`,
+					{
+						method,
+						// Only include body for methods that support it
+						body: ["POST", "PUT", "PATCH"].includes(method)
+							? JSON.stringify({ method })
+							: undefined,
+					},
+				);
 
 				// HEAD responses have no body
 				if (method === "HEAD") {
@@ -197,7 +214,9 @@ export function runRawHttpRequestPropertiesTests(
 
 		test("should handle complex query parameters", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpRequestPropertiesActor.getOrCreate(["test"]);
+			const actor = client.rawHttpRequestPropertiesActor.getOrCreate([
+				"test",
+			]);
 
 			// Test multiple values for same key
 			const response = await actor.fetch(
@@ -214,7 +233,9 @@ export function runRawHttpRequestPropertiesTests(
 
 		test("should handle multipart form data", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpRequestPropertiesActor.getOrCreate(["test"]);
+			const actor = client.rawHttpRequestPropertiesActor.getOrCreate([
+				"test",
+			]);
 
 			// Create multipart boundary
 			const boundary = "----RivetKitBoundary";
@@ -240,14 +261,18 @@ export function runRawHttpRequestPropertiesTests(
 
 			expect(response.ok).toBe(true);
 			const data = (await response.json()) as any;
-			expect(data.headers["content-type"]).toContain("multipart/form-data");
+			expect(data.headers["content-type"]).toContain(
+				"multipart/form-data",
+			);
 			expect(data.bodyText).toContain("field1");
 			expect(data.bodyText).toContain("value1");
 		});
 
 		test("should handle very long URLs", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpRequestPropertiesActor.getOrCreate(["test"]);
+			const actor = client.rawHttpRequestPropertiesActor.getOrCreate([
+				"test",
+			]);
 
 			// Create a very long query string
 			const longValue = "x".repeat(1000);
@@ -261,7 +286,9 @@ export function runRawHttpRequestPropertiesTests(
 
 		test("should handle large request bodies", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpRequestPropertiesActor.getOrCreate(["test"]);
+			const actor = client.rawHttpRequestPropertiesActor.getOrCreate([
+				"test",
+			]);
 
 			// Create a large JSON body (1MB+)
 			const largeArray = new Array(10000).fill({
@@ -285,7 +312,9 @@ export function runRawHttpRequestPropertiesTests(
 
 		test("should handle missing content-type header", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpRequestPropertiesActor.getOrCreate(["test"]);
+			const actor = client.rawHttpRequestPropertiesActor.getOrCreate([
+				"test",
+			]);
 
 			const response = await actor.fetch("test/no-content-type", {
 				method: "POST",
@@ -299,7 +328,9 @@ export function runRawHttpRequestPropertiesTests(
 
 		test("should handle empty request body", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpRequestPropertiesActor.getOrCreate(["test"]);
+			const actor = client.rawHttpRequestPropertiesActor.getOrCreate([
+				"test",
+			]);
 
 			const response = await actor.fetch("test/empty", {
 				method: "POST",
@@ -317,7 +348,9 @@ export function runRawHttpRequestPropertiesTests(
 
 		test("should handle custom HTTP methods", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpRequestPropertiesActor.getOrCreate(["test"]);
+			const actor = client.rawHttpRequestPropertiesActor.getOrCreate([
+				"test",
+			]);
 
 			// Test a custom method (though most HTTP clients may not support this)
 			try {
@@ -338,7 +371,9 @@ export function runRawHttpRequestPropertiesTests(
 
 		test("should handle cookies in headers", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpRequestPropertiesActor.getOrCreate(["test"]);
+			const actor = client.rawHttpRequestPropertiesActor.getOrCreate([
+				"test",
+			]);
 
 			const response = await actor.fetch("test/cookies", {
 				headers: {
@@ -355,7 +390,9 @@ export function runRawHttpRequestPropertiesTests(
 
 		test("should handle URL encoding properly", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpRequestPropertiesActor.getOrCreate(["test"]);
+			const actor = client.rawHttpRequestPropertiesActor.getOrCreate([
+				"test",
+			]);
 
 			// Test various encoded characters
 			const response = await actor.fetch(
@@ -373,7 +410,9 @@ export function runRawHttpRequestPropertiesTests(
 
 		test("should handle concurrent requests maintaining separate contexts", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpRequestPropertiesActor.getOrCreate(["test"]);
+			const actor = client.rawHttpRequestPropertiesActor.getOrCreate([
+				"test",
+			]);
 
 			// Send multiple concurrent requests with different data
 			const requests = [

package/src/driver-test-suite/tests/raw-http.ts

@@ -81,7 +81,9 @@ export function runRawHttpTests(driverTestConfig: DriverTestConfig) {
 
 		test("should return 404 when no onFetch handler defined", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpNoHandlerActor.getOrCreate(["no-handler"]);
+			const actor = client.rawHttpNoHandlerActor.getOrCreate([
+				"no-handler",
+			]);
 
 			const response = await actor.fetch("api/anything");
 			expect(response.ok).toBe(false);
@@ -92,7 +94,9 @@ export function runRawHttpTests(driverTestConfig: DriverTestConfig) {
 
 		test("should return 500 error when onFetch returns void", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawHttpVoidReturnActor.getOrCreate(["void-return"]);
+			const actor = client.rawHttpVoidReturnActor.getOrCreate([
+				"void-return",
+			]);
 
 			const response = await actor.fetch("api/anything");
 			expect(response.ok).toBe(false);
@@ -100,7 +104,9 @@ export function runRawHttpTests(driverTestConfig: DriverTestConfig) {
 
 			// Check error message - response might be CBOR encoded
 			try {
-				const errorData = (await response.json()) as { message: string };
+				const errorData = (await response.json()) as {
+					message: string;
+				};
 				expect(errorData.message).toContain(
 					"onFetch handler must return a Response",
 				);
@@ -276,7 +282,10 @@ export function runRawHttpTests(driverTestConfig: DriverTestConfig) {
 
 			// This simulates calling actor.fetch(truncatedPath, newRequest)
 			// which was causing double slashes in the example
-			const response = await actor.fetch(truncatedPath, newRequest as any);
+			const response = await actor.fetch(
+				truncatedPath,
+				newRequest as any,
+			);
 			expect(response.ok).toBe(true);
 			const users = await response.json();
 			expect(users).toEqual([
@@ -338,7 +347,10 @@ export function runRawHttpTests(driverTestConfig: DriverTestConfig) {
 				headers: { "X-Override": "init-header" },
 			});
 			expect(overrideResponse.ok).toBe(true);
-			const headers = (await overrideResponse.json()) as Record<string, string>;
+			const headers = (await overrideResponse.json()) as Record<
+				string,
+				string
+			>;
 			expect(headers["x-override"]).toBe("init-header");
 			// Original headers should be present too
 			expect(headers["x-original"]).toBe("request-header");

package/src/driver-test-suite/tests/raw-websocket.ts

@@ -55,7 +55,9 @@ export function runRawWebSocketTests(driverTestConfig: DriverTestConfig) {
 			// Check if WebSocket is already open
 			if (ws.readyState !== WebSocket.OPEN) {
 				await new Promise<void>((resolve, reject) => {
-					ws.addEventListener("open", () => resolve(), { once: true });
+					ws.addEventListener("open", () => resolve(), {
+						once: true,
+					});
 					ws.addEventListener("close", reject);
 				});
 			}
@@ -95,7 +97,9 @@ export function runRawWebSocketTests(driverTestConfig: DriverTestConfig) {
 			// Check if WebSocket is already open
 			if (ws.readyState !== WebSocket.OPEN) {
 				await new Promise<void>((resolve, reject) => {
-					ws.addEventListener("open", () => resolve(), { once: true });
+					ws.addEventListener("open", () => resolve(), {
+						once: true,
+					});
 					ws.addEventListener("close", reject);
 				});
 			}
@@ -132,7 +136,9 @@ export function runRawWebSocketTests(driverTestConfig: DriverTestConfig) {
 			// Create first connection to ensure actor exists
 			const ws1 = await actor1.websocket();
 			const ws1MessagePromise = new Promise<void>((resolve, reject) => {
-				ws1.addEventListener("message", () => resolve(), { once: true });
+				ws1.addEventListener("message", () => resolve(), {
+					once: true,
+				});
 				ws1.addEventListener("close", reject);
 			});
 
@@ -143,7 +149,9 @@ export function runRawWebSocketTests(driverTestConfig: DriverTestConfig) {
 			const actor2 = client.rawWebSocketActor.get(["stats"]);
 			const ws2 = await actor2.websocket();
 			const ws2MessagePromise = new Promise<void>((resolve, reject) => {
-				ws2.addEventListener("message", () => resolve(), { once: true });
+				ws2.addEventListener("message", () => resolve(), {
+					once: true,
+				});
 				ws2.addEventListener("close", reject);
 			});
 
@@ -191,14 +199,18 @@ export function runRawWebSocketTests(driverTestConfig: DriverTestConfig) {
 
 		test("should handle binary data", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawWebSocketBinaryActor.getOrCreate(["binary"]);
+			const actor = client.rawWebSocketBinaryActor.getOrCreate([
+				"binary",
+			]);
 
 			const ws = await actor.websocket();
 
 			// Check if WebSocket is already open
 			if (ws.readyState !== WebSocket.OPEN) {
 				await new Promise<void>((resolve, reject) => {
-					ws.addEventListener("open", () => resolve(), { once: true });
+					ws.addEventListener("open", () => resolve(), {
+						once: true,
+					});
 					ws.addEventListener("close", reject);
 				});
 			}
@@ -220,7 +232,9 @@ export function runRawWebSocketTests(driverTestConfig: DriverTestConfig) {
 
 				// Convert Blob to ArrayBuffer if needed
 				const buffer =
-					response instanceof Blob ? await response.arrayBuffer() : response;
+					response instanceof Blob
+						? await response.arrayBuffer()
+						: response;
 
 				return new Uint8Array(buffer);
 			};
@@ -241,7 +255,9 @@ export function runRawWebSocketTests(driverTestConfig: DriverTestConfig) {
 
 			// Verify it's reversed correctly
 			for (let i = 0; i < largeData.length; i++) {
-				expect(largeReversed[i]).toBe(largeData[largeData.length - 1 - i]);
+				expect(largeReversed[i]).toBe(
+					largeData[largeData.length - 1 - i],
+				);
 			}
 
 			ws.close();
@@ -286,7 +302,9 @@ export function runRawWebSocketTests(driverTestConfig: DriverTestConfig) {
 			// Check if WebSocket is already open
 			if (ws.readyState !== WebSocket.OPEN) {
 				await new Promise<void>((resolve, reject) => {
-					ws.addEventListener("open", () => resolve(), { once: true });
+					ws.addEventListener("open", () => resolve(), {
+						once: true,
+					});
 					ws.addEventListener("close", reject);
 				});
 			}
@@ -320,7 +338,9 @@ export function runRawWebSocketTests(driverTestConfig: DriverTestConfig) {
 
 		test("should properly handle onWebSocket open and close events", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawWebSocketActor.getOrCreate(["open-close-test"]);
+			const actor = client.rawWebSocketActor.getOrCreate([
+				"open-close-test",
+			]);
 
 			// Create first connection
 			const ws1 = await actor.websocket();
@@ -413,10 +433,14 @@ export function runRawWebSocketTests(driverTestConfig: DriverTestConfig) {
 
 		test("should handle query parameters in websocket paths", async (c) => {
 			const { client } = await setupDriverTest(c, driverTestConfig);
-			const actor = client.rawWebSocketActor.getOrCreate(["query-params"]);
+			const actor = client.rawWebSocketActor.getOrCreate([
+				"query-params",
+			]);
 
 			// Test WebSocket with query parameters
-			const ws = await actor.websocket("api/v1/stream?token=abc123&user=test");
+			const ws = await actor.websocket(
+				"api/v1/stream?token=abc123&user=test",
+			);
 
 			await new Promise<void>((resolve, reject) => {
 				ws.addEventListener("open", () => resolve(), { once: true });

package/src/driver-test-suite/tests/request-access.ts

@@ -8,9 +8,12 @@ export function runRequestAccessTests(driverTestConfig: DriverTestConfig) {
 			const { client } = await setupDriverTest(c, driverTestConfig);
 
 			// Create actor with request tracking enabled
-			const handle = client.requestAccessActor.getOrCreate(["test-request"], {
-				params: { trackRequest: true },
-			});
+			const handle = client.requestAccessActor.getOrCreate(
+				["test-request"],
+				{
+					params: { trackRequest: true },
+				},
+			);
 			const connection = handle.connect();
 
 			// Get request info that was captured in onBeforeConnect
@@ -22,13 +25,17 @@ export function runRequestAccessTests(driverTestConfig: DriverTestConfig) {
 				expect(requestInfo.onBeforeConnect.hasRequest).toBe(true);
 				expect(requestInfo.onBeforeConnect.requestUrl).toBeDefined();
 				expect(requestInfo.onBeforeConnect.requestMethod).toBeDefined();
-				expect(requestInfo.onBeforeConnect.requestHeaders).toBeDefined();
+				expect(
+					requestInfo.onBeforeConnect.requestHeaders,
+				).toBeDefined();
 
 				// Check createConnState
 				expect(requestInfo.createConnState.hasRequest).toBe(true);
 				expect(requestInfo.createConnState.requestUrl).toBeDefined();
 				expect(requestInfo.createConnState.requestMethod).toBeDefined();
-				expect(requestInfo.createConnState.requestHeaders).toBeDefined();
+				expect(
+					requestInfo.createConnState.requestHeaders,
+				).toBeDefined();
 			} else {
 				// Inline client may or may not have request object depending on the driver
 				//
@@ -79,9 +86,12 @@ export function runRequestAccessTests(driverTestConfig: DriverTestConfig) {
 			const { client } = await setupDriverTest(c, driverTestConfig);
 
 			// Create actor and connect with request tracking
-			const handle = client.requestAccessActor.getOrCreate(["test-headers"], {
-				params: { trackRequest: true },
-			});
+			const handle = client.requestAccessActor.getOrCreate(
+				["test-headers"],
+				{
+					params: { trackRequest: true },
+				},
+			);
 			const connection = handle.connect();
 
 			// Get request info