rivetkit 2.0.29-rc.1 → 2.0.29

package/dist/tsup/driver-test-suite/mod.d.cts

@@ -1,4 +1,4 @@
-import { E as Encoding, M as Registry, D as DriverConfig } from '../config-DxlmiJS1.cjs';
+import { E as Encoding, N as Registry, D as DriverConfig } from '../config-BEu-kmmR.cjs';
 import 'zod';
 import 'hono';
 import 'pino';

package/dist/tsup/driver-test-suite/mod.d.ts

@@ -1,4 +1,4 @@
-import { E as Encoding, M as Registry, D as DriverConfig } from '../config-CcMdKDv9.js';
+import { E as Encoding, N as Registry, D as DriverConfig } from '../config-DELCUYts.js';
 import 'zod';
 import 'hono';
 import 'pino';

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

@@ -1,30 +1,30 @@
 import {
   getPort
-} from "../chunk-SDJFBUFE.js";
+} from "../chunk-H27UUFLD.js";
 import {
   actor,
   createManagerRouter
-} from "../chunk-QKDCNAPW.js";
-import {
-  configureInspectorAccessToken,
-  createActorInspectorClient,
-  createManagerInspectorClient
-} from "../chunk-EVQJYFFP.js";
+} from "../chunk-YPHTAWZP.js";
 import {
   createClient
-} from "../chunk-5RN6O2RW.js";
+} from "../chunk-POJ4HHIO.js";
 import {
   ActorError,
   createClientWithDriver
-} from "../chunk-CHPLLKNL.js";
-import "../chunk-WUFUT7CZ.js";
-import {
-  importWebSocket
-} from "../chunk-EU7VV5PU.js";
+} from "../chunk-FBPZAU3N.js";
 import {
   ClientConfigSchema,
   RunnerConfigSchema
-} from "../chunk-CGIRSKRW.js";
+} from "../chunk-5IE76QRF.js";
+import {
+  importWebSocket
+} from "../chunk-UJ6ETM45.js";
+import {
+  configureInspectorAccessToken,
+  createActorInspectorClient,
+  createManagerInspectorClient
+} from "../chunk-KMOA3GQ2.js";
+import "../chunk-EWQE7L3V.js";
 import {
   HEADER_ACTOR_ID,
   HEADER_ACTOR_QUERY,
@@ -35,11 +35,11 @@ import {
   WS_PROTOCOL_TARGET,
   WS_TEST_PROTOCOL_PATH,
   assertUnreachable
-} from "../chunk-HXWPIIEM.js";
+} from "../chunk-2LOXTB3G.js";
 import {
   getLogger,
   promiseWithResolvers
-} from "../chunk-YJXB7BPR.js";
+} from "../chunk-WXLDDJCI.js";
 import {
   INTERNAL_ERROR_CODE,
   INTERNAL_ERROR_DESCRIPTION
@@ -864,6 +864,124 @@ function runActorConnTests(driverTestConfig) {
         );
       });
     });
+    describe2("Connection State", () => {
+      test2("isConnected should be false before connection opens", async (c) => {
+        const { client } = await setupDriverTest(c, driverTestConfig);
+        const handle = client.counter.getOrCreate([
+          "test-isconnected-initial"
+        ]);
+        const connection = handle.connect();
+        expect2(connection.isConnected).toBe(false);
+        await vi2.waitFor(() => {
+          expect2(connection.isConnected).toBe(true);
+        });
+        await connection.dispose();
+      });
+      test2("onOpen should be called when connection opens", async (c) => {
+        const { client } = await setupDriverTest(c, driverTestConfig);
+        const handle = client.counter.getOrCreate(["test-onopen"]);
+        const connection = handle.connect();
+        let openCount = 0;
+        connection.onOpen(() => {
+          openCount++;
+        });
+        await vi2.waitFor(() => {
+          expect2(openCount).toBe(1);
+        });
+        expect2(connection.isConnected).toBe(true);
+        await connection.dispose();
+      });
+      test2("onClose should be called when connection closes via dispose", async (c) => {
+        const { client } = await setupDriverTest(c, driverTestConfig);
+        const handle = client.counter.getOrCreate(["test-onclose"]);
+        const connection = handle.connect();
+        let closeCount = 0;
+        connection.onClose(() => {
+          closeCount++;
+        });
+        await vi2.waitFor(() => {
+          expect2(connection.isConnected).toBe(true);
+        });
+        await connection.dispose();
+        expect2(closeCount).toBe(1);
+        expect2(connection.isConnected).toBe(false);
+      });
+      test2("should be able to unsubscribe from onOpen", async (c) => {
+        const { client } = await setupDriverTest(c, driverTestConfig);
+        const handle = client.counter.getOrCreate([
+          "test-onopen-unsub"
+        ]);
+        const connection = handle.connect();
+        let openCount = 0;
+        const unsubscribe = connection.onOpen(() => {
+          openCount++;
+        });
+        unsubscribe();
+        await vi2.waitFor(() => {
+          expect2(connection.isConnected).toBe(true);
+        });
+        expect2(openCount).toBe(0);
+        await connection.dispose();
+      });
+      test2("should be able to unsubscribe from onClose", async (c) => {
+        const { client } = await setupDriverTest(c, driverTestConfig);
+        const handle = client.counter.getOrCreate([
+          "test-onclose-unsub"
+        ]);
+        const connection = handle.connect();
+        let closeCount = 0;
+        const unsubscribe = connection.onClose(() => {
+          closeCount++;
+        });
+        await vi2.waitFor(() => {
+          expect2(connection.isConnected).toBe(true);
+        });
+        unsubscribe();
+        await connection.dispose();
+        expect2(closeCount).toBe(0);
+      });
+      test2("multiple onOpen handlers should all be called", async (c) => {
+        const { client } = await setupDriverTest(c, driverTestConfig);
+        const handle = client.counter.getOrCreate([
+          "test-multi-onopen"
+        ]);
+        const connection = handle.connect();
+        let handler1Called = false;
+        let handler2Called = false;
+        connection.onOpen(() => {
+          handler1Called = true;
+        });
+        connection.onOpen(() => {
+          handler2Called = true;
+        });
+        await vi2.waitFor(() => {
+          expect2(handler1Called).toBe(true);
+          expect2(handler2Called).toBe(true);
+        });
+        await connection.dispose();
+      });
+      test2("multiple onClose handlers should all be called", async (c) => {
+        const { client } = await setupDriverTest(c, driverTestConfig);
+        const handle = client.counter.getOrCreate([
+          "test-multi-onclose"
+        ]);
+        const connection = handle.connect();
+        let handler1Called = false;
+        let handler2Called = false;
+        connection.onClose(() => {
+          handler1Called = true;
+        });
+        connection.onClose(() => {
+          handler2Called = true;
+        });
+        await vi2.waitFor(() => {
+          expect2(connection.isConnected).toBe(true);
+        });
+        await connection.dispose();
+        expect2(handler1Called).toBe(true);
+        expect2(handler2Called).toBe(true);
+      });
+    });
   });
 }