rivetkit 2.0.37 → 2.0.39

package/dist/tsup/driver-test-suite/mod.cjs

@@ -1,19 +1,19 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } }
 
-var _chunk22NKW7F5cjs = require('../chunk-22NKW7F5.cjs');
+var _chunkMZPYVTVGcjs = require('../chunk-MZPYVTVG.cjs');
 
 
-var _chunkG4N7FZMMcjs = require('../chunk-G4N7FZMM.cjs');
-require('../chunk-LYYTV7DN.cjs');
+var _chunk7E3RWMR6cjs = require('../chunk-7E3RWMR6.cjs');
+require('../chunk-DY4H3ASE.cjs');
 
 
 
 
 
-var _chunk5VVIFC6Mcjs = require('../chunk-5VVIFC6M.cjs');
+var _chunkUAX5E3EUcjs = require('../chunk-UAX5E3EU.cjs');
 
 
-var _chunk4V7MS7SOcjs = require('../chunk-4V7MS7SO.cjs');
+var _chunkC64FV764cjs = require('../chunk-C64FV764.cjs');
 
 
 
@@ -24,13 +24,13 @@ var _chunk4V7MS7SOcjs = require('../chunk-4V7MS7SO.cjs');
 
 
 
-var _chunkB6BP74X3cjs = require('../chunk-B6BP74X3.cjs');
+var _chunkKMYFL3OLcjs = require('../chunk-KMYFL3OL.cjs');
 require('../chunk-KDFWJKMJ.cjs');
 
 
 
 
-var _chunkX5IX3YPOcjs = require('../chunk-X5IX3YPO.cjs');
+var _chunkX72X7I7Tcjs = require('../chunk-X72X7I7T.cjs');
 
 
 
@@ -45,7 +45,7 @@ var _vitest = require('vitest');
 
 // src/driver-test-suite/log.ts
 function logger() {
-  return _chunkX5IX3YPOcjs.getLogger.call(void 0, "test-suite");
+  return _chunkX72X7I7Tcjs.getLogger.call(void 0, "test-suite");
 }
 
 // src/driver-test-suite/tests/action-features.ts
@@ -97,7 +97,7 @@ function createTestInlineClientDriver(endpoint, encoding) {
         to: url
       });
       const headers = new Headers(actorRequest.headers);
-      headers.set(_chunkB6BP74X3cjs.HEADER_ACTOR_ID, actorId);
+      headers.set(_chunkKMYFL3OLcjs.HEADER_ACTOR_ID, actorId);
       const response = await fetch(
         new Request(url, {
           method: actorRequest.method,
@@ -113,7 +113,7 @@ function createTestInlineClientDriver(endpoint, encoding) {
           const errorData = await clonedResponse.json();
           if (errorData.error) {
             if (typeof errorData.error === "object") {
-              throw new (0, _chunk5VVIFC6Mcjs.ActorError)(
+              throw new (0, _chunkUAX5E3EUcjs.ActorError)(
                 errorData.error.code,
                 errorData.error.message,
                 errorData.error.metadata
@@ -121,7 +121,7 @@ function createTestInlineClientDriver(endpoint, encoding) {
             }
           }
         } catch (e) {
-          if (!(e instanceof _chunk5VVIFC6Mcjs.ActorError)) {
+          if (!(e instanceof _chunkUAX5E3EUcjs.ActorError)) {
             return response;
           }
           throw e;
@@ -130,7 +130,7 @@ function createTestInlineClientDriver(endpoint, encoding) {
       return response;
     },
     async openWebSocket(path, actorId, encoding2, params) {
-      const WebSocket2 = await _chunk4V7MS7SOcjs.importWebSocket.call(void 0, );
+      const WebSocket2 = await _chunkC64FV764cjs.importWebSocket.call(void 0, );
       const normalizedPath = path.startsWith("/") ? path.slice(1) : path;
       const wsUrl = new URL(
         `${endpoint}/.test/inline-driver/connect-websocket/ws`
@@ -142,18 +142,18 @@ function createTestInlineClientDriver(endpoint, encoding) {
       const wsProtocol = wsUrl.protocol === "https:" ? "wss:" : "ws:";
       const finalWsUrl = `${wsProtocol}//${wsUrl.host}${wsUrl.pathname}`;
       const protocols = [];
-      protocols.push(_chunkB6BP74X3cjs.WS_PROTOCOL_STANDARD);
-      protocols.push(`${_chunkB6BP74X3cjs.WS_PROTOCOL_TARGET}actor`);
+      protocols.push(_chunkKMYFL3OLcjs.WS_PROTOCOL_STANDARD);
+      protocols.push(`${_chunkKMYFL3OLcjs.WS_PROTOCOL_TARGET}actor`);
       protocols.push(
-        `${_chunkB6BP74X3cjs.WS_PROTOCOL_ACTOR}${encodeURIComponent(actorId)}`
+        `${_chunkKMYFL3OLcjs.WS_PROTOCOL_ACTOR}${encodeURIComponent(actorId)}`
       );
-      protocols.push(`${_chunkB6BP74X3cjs.WS_PROTOCOL_ENCODING}${encoding2}`);
+      protocols.push(`${_chunkKMYFL3OLcjs.WS_PROTOCOL_ENCODING}${encoding2}`);
       protocols.push(
-        `${_chunkB6BP74X3cjs.WS_TEST_PROTOCOL_PATH}${encodeURIComponent(normalizedPath)}`
+        `${_chunkKMYFL3OLcjs.WS_TEST_PROTOCOL_PATH}${encodeURIComponent(normalizedPath)}`
       );
       if (params !== void 0) {
         protocols.push(
-          `${_chunkB6BP74X3cjs.WS_PROTOCOL_CONN_PARAMS}${encodeURIComponent(JSON.stringify(params))}`
+          `${_chunkKMYFL3OLcjs.WS_PROTOCOL_CONN_PARAMS}${encodeURIComponent(JSON.stringify(params))}`
         );
       }
       logger().debug({
@@ -176,7 +176,10 @@ function createTestInlineClientDriver(endpoint, encoding) {
         encoding2,
         params
       );
-      return upgradeWebSocket(() => wsHandler)(c, _chunkX5IX3YPOcjs.noopNext.call(void 0, ));
+      return upgradeWebSocket(() => wsHandler)(c, _chunkX72X7I7Tcjs.noopNext.call(void 0, ));
+    },
+    async buildGatewayUrl(actorId) {
+      return `${endpoint}/gateway/${actorId}`;
     },
     displayInformation() {
       return { properties: {} };
@@ -220,14 +223,14 @@ async function makeInlineRequest(endpoint, encoding, method, args) {
   if ("ok" in callResponse) {
     return callResponse.ok;
   } else if ("err" in callResponse) {
-    throw new (0, _chunk5VVIFC6Mcjs.ActorError)(
+    throw new (0, _chunkUAX5E3EUcjs.ActorError)(
       callResponse.err.group,
       callResponse.err.code,
       callResponse.err.message,
       callResponse.err.metadata
     );
   } else {
-    _chunkB6BP74X3cjs.assertUnreachable.call(void 0, callResponse);
+    _chunkKMYFL3OLcjs.assertUnreachable.call(void 0, callResponse);
   }
 }
 
@@ -245,21 +248,25 @@ async function setupDriverTest(c, driverTestConfig) {
   });
   let client;
   if (driverTestConfig.clientType === "http") {
-    client = _chunk5VVIFC6Mcjs.createClient.call(void 0, {
+    client = _chunkUAX5E3EUcjs.createClient.call(void 0, {
       endpoint,
       namespace,
       runnerName,
-      encoding: driverTestConfig.encoding
+      encoding: driverTestConfig.encoding,
+      // Disable metadata lookup to prevent redirect to the wrong port.
+      // Each test starts a new server on a dynamic port, but the
+      // registry's publicEndpoint defaults to port 6420.
+      disableMetadataLookup: true
     });
   } else if (driverTestConfig.clientType === "inline") {
     const encoding = _nullishCoalesce(driverTestConfig.encoding, () => ( "bare"));
     const managerDriver = createTestInlineClientDriver(endpoint, encoding);
-    const runConfig = _chunk5VVIFC6Mcjs.ClientConfigSchema.parse({
+    const runConfig = _chunkUAX5E3EUcjs.ClientConfigSchema.parse({
       encoding
     });
-    client = _chunk5VVIFC6Mcjs.createClientWithDriver.call(void 0, managerDriver, runConfig);
+    client = _chunkUAX5E3EUcjs.createClientWithDriver.call(void 0, managerDriver, runConfig);
   } else {
-    _chunkB6BP74X3cjs.assertUnreachable.call(void 0, driverTestConfig.clientType);
+    _chunkKMYFL3OLcjs.assertUnreachable.call(void 0, driverTestConfig.clientType);
   }
   if (!driverTestConfig.HACK_skipCleanupNet) {
     c.onTestFinished(async () => await client.dispose());
@@ -783,7 +790,7 @@ function runActorConnTests(driverTestConfig) {
 
 // fixtures/driver-test-suite/hibernation.ts
 var HIBERNATION_SLEEP_TIMEOUT = 500;
-var hibernationActor = _chunkG4N7FZMMcjs.actor.call(void 0, {
+var hibernationActor = _chunk7E3RWMR6cjs.actor.call(void 0, {
   state: {
     sleepCount: 0,
     wakeCount: 0
@@ -1366,7 +1373,7 @@ function runActorScheduleTests(driverTestConfig) {
 
 // fixtures/driver-test-suite/sleep.ts
 var SLEEP_TIMEOUT = 1e3;
-var sleep = _chunkG4N7FZMMcjs.actor.call(void 0, {
+var sleep = _chunk7E3RWMR6cjs.actor.call(void 0, {
   state: { startCount: 0, sleepCount: 0 },
   onWake: (c) => {
     c.state.startCount += 1;
@@ -1395,7 +1402,7 @@ var sleep = _chunkG4N7FZMMcjs.actor.call(void 0, {
     sleepTimeout: SLEEP_TIMEOUT
   }
 });
-var sleepWithLongRpc = _chunkG4N7FZMMcjs.actor.call(void 0, {
+var sleepWithLongRpc = _chunk7E3RWMR6cjs.actor.call(void 0, {
   state: { startCount: 0, sleepCount: 0 },
   createVars: () => ({}),
   onWake: (c) => {
@@ -1413,7 +1420,7 @@ var sleepWithLongRpc = _chunkG4N7FZMMcjs.actor.call(void 0, {
     },
     longRunningRpc: async (c) => {
       c.log.info("starting long running rpc");
-      c.vars.longRunningResolve = _chunkX5IX3YPOcjs.promiseWithResolvers.call(void 0, );
+      c.vars.longRunningResolve = _chunkX72X7I7Tcjs.promiseWithResolvers.call(void 0, );
       c.broadcast("waiting");
       await c.vars.longRunningResolve.promise;
       c.log.info("finished long running rpc");
@@ -1427,7 +1434,7 @@ var sleepWithLongRpc = _chunkG4N7FZMMcjs.actor.call(void 0, {
     sleepTimeout: SLEEP_TIMEOUT
   }
 });
-var sleepWithRawHttp = _chunkG4N7FZMMcjs.actor.call(void 0, {
+var sleepWithRawHttp = _chunk7E3RWMR6cjs.actor.call(void 0, {
   state: { startCount: 0, sleepCount: 0, requestCount: 0 },
   onWake: (c) => {
     c.state.startCount += 1;
@@ -1464,7 +1471,7 @@ var sleepWithRawHttp = _chunkG4N7FZMMcjs.actor.call(void 0, {
     sleepTimeout: SLEEP_TIMEOUT
   }
 });
-var sleepWithRawWebSocket = _chunkG4N7FZMMcjs.actor.call(void 0, {
+var sleepWithRawWebSocket = _chunk7E3RWMR6cjs.actor.call(void 0, {
   state: { startCount: 0, sleepCount: 0, connectionCount: 0 },
   onWake: (c) => {
     c.state.startCount += 1;
@@ -1527,7 +1534,7 @@ var sleepWithRawWebSocket = _chunkG4N7FZMMcjs.actor.call(void 0, {
     sleepTimeout: SLEEP_TIMEOUT
   }
 });
-var sleepWithNoSleepOption = _chunkG4N7FZMMcjs.actor.call(void 0, {
+var sleepWithNoSleepOption = _chunk7E3RWMR6cjs.actor.call(void 0, {
   state: { startCount: 0, sleepCount: 0 },
   onWake: (c) => {
     c.state.startCount += 1;
@@ -2258,6 +2265,44 @@ function runActorInspectorTests(driverTestConfig) {
   });
 }
 
+// src/driver-test-suite/tests/actor-kv.ts
+
+function runActorKvTests(driverTestConfig) {
+  _vitest.describe.call(void 0, "Actor KV Tests", () => {
+    _vitest.test.call(void 0, "supports text encoding and decoding", async (c) => {
+      const { client } = await setupDriverTest(c, driverTestConfig);
+      const kvHandle = client.kvActor.getOrCreate(["kv-text"]);
+      await kvHandle.putText("greeting", "hello");
+      const value = await kvHandle.getText("greeting");
+      _vitest.expect.call(void 0, value).toBe("hello");
+      await kvHandle.putText("prefix-a", "alpha");
+      await kvHandle.putText("prefix-b", "beta");
+      const results = await kvHandle.listText("prefix-");
+      const sorted = results.sort((a, b) => a.key.localeCompare(b.key));
+      _vitest.expect.call(void 0, sorted).toEqual([
+        { key: "prefix-a", value: "alpha" },
+        { key: "prefix-b", value: "beta" }
+      ]);
+    });
+    _vitest.test.call(void 0, 
+      "supports arrayBuffer encoding and decoding",
+      async (c) => {
+        const { client } = await setupDriverTest(c, driverTestConfig);
+        const kvHandle = client.kvActor.getOrCreate(["kv-array-buffer"]);
+        const values = await kvHandle.roundtripArrayBuffer("bytes", [
+          4,
+          8,
+          15,
+          16,
+          23,
+          42
+        ]);
+        _vitest.expect.call(void 0, values).toEqual([4, 8, 15, 16, 23, 42]);
+      }
+    );
+  });
+}
+
 // src/driver-test-suite/tests/actor-metadata.ts
 
 function runActorMetadataTests(driverTestConfig) {
@@ -3741,6 +3786,7 @@ function runDriverTests(driverTestConfigPartial) {
           runActorOnStateChangeTests(driverTestConfig);
           runActorErrorHandlingTests(driverTestConfig);
           runActorInlineClientTests(driverTestConfig);
+          runActorKvTests(driverTestConfig);
           runRawHttpTests(driverTestConfig);
           runRawHttpRequestPropertiesTests(driverTestConfig);
           runRawWebSocketTests(driverTestConfig);
@@ -3778,7 +3824,7 @@ async function createTestRuntime(registryPath, driverFactory) {
     const parsedConfig = registry.parseConfig();
     const managerDriver = (_a = driver.manager) == null ? void 0 : _a.call(driver, parsedConfig);
     _invariant2.default.call(void 0, managerDriver, "missing manager driver");
-    const { router } = _chunkB6BP74X3cjs.buildManagerRouter.call(void 0, 
+    const { router } = _chunkKMYFL3OLcjs.buildManagerRouter.call(void 0, 
       parsedConfig,
       managerDriver,
       () => upgradeWebSocket
@@ -3786,7 +3832,7 @@ async function createTestRuntime(registryPath, driverFactory) {
     const nodeWebSocket = _nodews.createNodeWebSocket.call(void 0, { app: router });
     upgradeWebSocket = nodeWebSocket.upgradeWebSocket;
     managerDriver.setGetUpgradeWebSocket(() => upgradeWebSocket);
-    const port = await _chunk22NKW7F5cjs.getPort.call(void 0, );
+    const port = await _chunkMZPYVTVGcjs.getPort.call(void 0, );
     const server = _nodeserver.serve.call(void 0, {
       fetch: router.fetch,
       hostname: "127.0.0.1",