rivetkit 2.0.39 → 2.0.41

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 _chunkMZPYVTVGcjs = require('../chunk-MZPYVTVG.cjs');
+var _chunkQDVKDX7Gcjs = require('../chunk-QDVKDX7G.cjs');
 
 
-var _chunk7E3RWMR6cjs = require('../chunk-7E3RWMR6.cjs');
-require('../chunk-DY4H3ASE.cjs');
+var _chunk4UHJ3ADWcjs = require('../chunk-4UHJ3ADW.cjs');
+require('../chunk-GTQKDCM4.cjs');
 
 
 
 
 
-var _chunkUAX5E3EUcjs = require('../chunk-UAX5E3EU.cjs');
+var _chunkWYZLRPQMcjs = require('../chunk-WYZLRPQM.cjs');
 
 
-var _chunkC64FV764cjs = require('../chunk-C64FV764.cjs');
+var _chunkDSNSFYDLcjs = require('../chunk-DSNSFYDL.cjs');
 
 
 
@@ -24,13 +24,13 @@ var _chunkC64FV764cjs = require('../chunk-C64FV764.cjs');
 
 
 
-var _chunkKMYFL3OLcjs = require('../chunk-KMYFL3OL.cjs');
+var _chunkB5UXC4QVcjs = require('../chunk-B5UXC4QV.cjs');
 require('../chunk-KDFWJKMJ.cjs');
 
 
 
 
-var _chunkX72X7I7Tcjs = require('../chunk-X72X7I7T.cjs');
+var _chunkS6F7EKC7cjs = require('../chunk-S6F7EKC7.cjs');
 
 
 
@@ -45,7 +45,7 @@ var _vitest = require('vitest');
 
 // src/driver-test-suite/log.ts
 function logger() {
-  return _chunkX72X7I7Tcjs.getLogger.call(void 0, "test-suite");
+  return _chunkS6F7EKC7cjs.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(_chunkKMYFL3OLcjs.HEADER_ACTOR_ID, actorId);
+      headers.set(_chunkB5UXC4QVcjs.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, _chunkUAX5E3EUcjs.ActorError)(
+              throw new (0, _chunkWYZLRPQMcjs.ActorError)(
                 errorData.error.code,
                 errorData.error.message,
                 errorData.error.metadata
@@ -121,7 +121,7 @@ function createTestInlineClientDriver(endpoint, encoding) {
             }
           }
         } catch (e) {
-          if (!(e instanceof _chunkUAX5E3EUcjs.ActorError)) {
+          if (!(e instanceof _chunkWYZLRPQMcjs.ActorError)) {
             return response;
           }
           throw e;
@@ -130,7 +130,7 @@ function createTestInlineClientDriver(endpoint, encoding) {
       return response;
     },
     async openWebSocket(path, actorId, encoding2, params) {
-      const WebSocket2 = await _chunkC64FV764cjs.importWebSocket.call(void 0, );
+      const WebSocket2 = await _chunkDSNSFYDLcjs.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(_chunkKMYFL3OLcjs.WS_PROTOCOL_STANDARD);
-      protocols.push(`${_chunkKMYFL3OLcjs.WS_PROTOCOL_TARGET}actor`);
+      protocols.push(_chunkB5UXC4QVcjs.WS_PROTOCOL_STANDARD);
+      protocols.push(`${_chunkB5UXC4QVcjs.WS_PROTOCOL_TARGET}actor`);
       protocols.push(
-        `${_chunkKMYFL3OLcjs.WS_PROTOCOL_ACTOR}${encodeURIComponent(actorId)}`
+        `${_chunkB5UXC4QVcjs.WS_PROTOCOL_ACTOR}${encodeURIComponent(actorId)}`
       );
-      protocols.push(`${_chunkKMYFL3OLcjs.WS_PROTOCOL_ENCODING}${encoding2}`);
+      protocols.push(`${_chunkB5UXC4QVcjs.WS_PROTOCOL_ENCODING}${encoding2}`);
       protocols.push(
-        `${_chunkKMYFL3OLcjs.WS_TEST_PROTOCOL_PATH}${encodeURIComponent(normalizedPath)}`
+        `${_chunkB5UXC4QVcjs.WS_TEST_PROTOCOL_PATH}${encodeURIComponent(normalizedPath)}`
       );
       if (params !== void 0) {
         protocols.push(
-          `${_chunkKMYFL3OLcjs.WS_PROTOCOL_CONN_PARAMS}${encodeURIComponent(JSON.stringify(params))}`
+          `${_chunkB5UXC4QVcjs.WS_PROTOCOL_CONN_PARAMS}${encodeURIComponent(JSON.stringify(params))}`
         );
       }
       logger().debug({
@@ -176,7 +176,7 @@ function createTestInlineClientDriver(endpoint, encoding) {
         encoding2,
         params
       );
-      return upgradeWebSocket(() => wsHandler)(c, _chunkX72X7I7Tcjs.noopNext.call(void 0, ));
+      return upgradeWebSocket(() => wsHandler)(c, _chunkS6F7EKC7cjs.noopNext.call(void 0, ));
     },
     async buildGatewayUrl(actorId) {
       return `${endpoint}/gateway/${actorId}`;
@@ -223,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, _chunkUAX5E3EUcjs.ActorError)(
+    throw new (0, _chunkWYZLRPQMcjs.ActorError)(
       callResponse.err.group,
       callResponse.err.code,
       callResponse.err.message,
       callResponse.err.metadata
     );
   } else {
-    _chunkKMYFL3OLcjs.assertUnreachable.call(void 0, callResponse);
+    _chunkB5UXC4QVcjs.assertUnreachable.call(void 0, callResponse);
   }
 }
 
@@ -248,7 +248,7 @@ async function setupDriverTest(c, driverTestConfig) {
   });
   let client;
   if (driverTestConfig.clientType === "http") {
-    client = _chunkUAX5E3EUcjs.createClient.call(void 0, {
+    client = _chunkWYZLRPQMcjs.createClient.call(void 0, {
       endpoint,
       namespace,
       runnerName,
@@ -261,12 +261,12 @@ async function setupDriverTest(c, driverTestConfig) {
   } else if (driverTestConfig.clientType === "inline") {
     const encoding = _nullishCoalesce(driverTestConfig.encoding, () => ( "bare"));
     const managerDriver = createTestInlineClientDriver(endpoint, encoding);
-    const runConfig = _chunkUAX5E3EUcjs.ClientConfigSchema.parse({
+    const runConfig = _chunkWYZLRPQMcjs.ClientConfigSchema.parse({
       encoding
     });
-    client = _chunkUAX5E3EUcjs.createClientWithDriver.call(void 0, managerDriver, runConfig);
+    client = _chunkWYZLRPQMcjs.createClientWithDriver.call(void 0, managerDriver, runConfig);
   } else {
-    _chunkKMYFL3OLcjs.assertUnreachable.call(void 0, driverTestConfig.clientType);
+    _chunkB5UXC4QVcjs.assertUnreachable.call(void 0, driverTestConfig.clientType);
   }
   if (!driverTestConfig.HACK_skipCleanupNet) {
     c.onTestFinished(async () => await client.dispose());
@@ -790,7 +790,7 @@ function runActorConnTests(driverTestConfig) {
 
 // fixtures/driver-test-suite/hibernation.ts
 var HIBERNATION_SLEEP_TIMEOUT = 500;
-var hibernationActor = _chunk7E3RWMR6cjs.actor.call(void 0, {
+var hibernationActor = _chunk4UHJ3ADWcjs.actor.call(void 0, {
   state: {
     sleepCount: 0,
     wakeCount: 0
@@ -1373,7 +1373,7 @@ function runActorScheduleTests(driverTestConfig) {
 
 // fixtures/driver-test-suite/sleep.ts
 var SLEEP_TIMEOUT = 1e3;
-var sleep = _chunk7E3RWMR6cjs.actor.call(void 0, {
+var sleep = _chunk4UHJ3ADWcjs.actor.call(void 0, {
   state: { startCount: 0, sleepCount: 0 },
   onWake: (c) => {
     c.state.startCount += 1;
@@ -1402,7 +1402,7 @@ var sleep = _chunk7E3RWMR6cjs.actor.call(void 0, {
     sleepTimeout: SLEEP_TIMEOUT
   }
 });
-var sleepWithLongRpc = _chunk7E3RWMR6cjs.actor.call(void 0, {
+var sleepWithLongRpc = _chunk4UHJ3ADWcjs.actor.call(void 0, {
   state: { startCount: 0, sleepCount: 0 },
   createVars: () => ({}),
   onWake: (c) => {
@@ -1420,7 +1420,7 @@ var sleepWithLongRpc = _chunk7E3RWMR6cjs.actor.call(void 0, {
     },
     longRunningRpc: async (c) => {
       c.log.info("starting long running rpc");
-      c.vars.longRunningResolve = _chunkX72X7I7Tcjs.promiseWithResolvers.call(void 0, );
+      c.vars.longRunningResolve = _chunkS6F7EKC7cjs.promiseWithResolvers.call(void 0, );
       c.broadcast("waiting");
       await c.vars.longRunningResolve.promise;
       c.log.info("finished long running rpc");
@@ -1434,7 +1434,7 @@ var sleepWithLongRpc = _chunk7E3RWMR6cjs.actor.call(void 0, {
     sleepTimeout: SLEEP_TIMEOUT
   }
 });
-var sleepWithRawHttp = _chunk7E3RWMR6cjs.actor.call(void 0, {
+var sleepWithRawHttp = _chunk4UHJ3ADWcjs.actor.call(void 0, {
   state: { startCount: 0, sleepCount: 0, requestCount: 0 },
   onWake: (c) => {
     c.state.startCount += 1;
@@ -1471,7 +1471,7 @@ var sleepWithRawHttp = _chunk7E3RWMR6cjs.actor.call(void 0, {
     sleepTimeout: SLEEP_TIMEOUT
   }
 });
-var sleepWithRawWebSocket = _chunk7E3RWMR6cjs.actor.call(void 0, {
+var sleepWithRawWebSocket = _chunk4UHJ3ADWcjs.actor.call(void 0, {
   state: { startCount: 0, sleepCount: 0, connectionCount: 0 },
   onWake: (c) => {
     c.state.startCount += 1;
@@ -1534,7 +1534,7 @@ var sleepWithRawWebSocket = _chunk7E3RWMR6cjs.actor.call(void 0, {
     sleepTimeout: SLEEP_TIMEOUT
   }
 });
-var sleepWithNoSleepOption = _chunk7E3RWMR6cjs.actor.call(void 0, {
+var sleepWithNoSleepOption = _chunk4UHJ3ADWcjs.actor.call(void 0, {
   state: { startCount: 0, sleepCount: 0 },
   onWake: (c) => {
     c.state.startCount += 1;
@@ -3668,6 +3668,34 @@ function runRawWebSocketTests(driverTestConfig) {
       _vitest.expect.call(void 0, requestInfo.url).toContain("user=test");
       ws.close();
     });
+    _vitest.test.call(void 0, "should handle query parameters on base websocket path (no subpath)", async (c) => {
+      const { client } = await setupDriverTest(c, driverTestConfig);
+      const actor2 = client.rawWebSocketActor.getOrCreate([
+        "base-path-query-params"
+      ]);
+      const ws = await actor2.websocket("?token=secret&session=123");
+      await new Promise((resolve, reject) => {
+        ws.addEventListener("open", () => resolve(), { once: true });
+        ws.addEventListener("error", reject);
+        ws.addEventListener("close", (evt) => {
+          reject(new Error(`WebSocket closed: code=${evt.code} reason=${evt.reason}`));
+        });
+      });
+      const requestInfoPromise = new Promise((resolve, reject) => {
+        ws.addEventListener("message", (event) => {
+          const data = JSON.parse(event.data);
+          if (data.type === "requestInfo") {
+            resolve(data);
+          }
+        });
+        ws.addEventListener("close", reject);
+      });
+      ws.send(JSON.stringify({ type: "getRequestInfo" }));
+      const requestInfo = await requestInfoPromise;
+      _vitest.expect.call(void 0, requestInfo.url).toContain("token=secret");
+      _vitest.expect.call(void 0, requestInfo.url).toContain("session=123");
+      ws.close();
+    });
   });
 }
 
@@ -3824,7 +3852,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 } = _chunkKMYFL3OLcjs.buildManagerRouter.call(void 0, 
+    const { router } = _chunkB5UXC4QVcjs.buildManagerRouter.call(void 0, 
       parsedConfig,
       managerDriver,
       () => upgradeWebSocket
@@ -3832,7 +3860,7 @@ async function createTestRuntime(registryPath, driverFactory) {
     const nodeWebSocket = _nodews.createNodeWebSocket.call(void 0, { app: router });
     upgradeWebSocket = nodeWebSocket.upgradeWebSocket;
     managerDriver.setGetUpgradeWebSocket(() => upgradeWebSocket);
-    const port = await _chunkMZPYVTVGcjs.getPort.call(void 0, );
+    const port = await _chunkQDVKDX7Gcjs.getPort.call(void 0, );
     const server = _nodeserver.serve.call(void 0, {
       fetch: router.fetch,
       hostname: "127.0.0.1",