jazz-run 0.14.27 → 0.15.0

package/.turbo/turbo-build.log

@@ -1,4 +1,4 @@
 
-> jazz-run@0.14.27 build /home/runner/_work/jazz/jazz/packages/jazz-run
+> jazz-run@0.15.0 build /home/runner/_work/jazz/jazz/packages/jazz-run
 > rm -rf ./dist && tsc --sourceMap --outDir dist && chmod +x ./dist/index.js
 

package/CHANGELOG.md

@@ -1,5 +1,26 @@
 # jazz-run
 
+## 0.15.0
+
+### Patch Changes
+
+- Updated dependencies [1378a1f]
+- Updated dependencies [0fa051a]
+  - jazz-tools@0.15.0
+  - cojson@0.15.0
+  - cojson-storage-sqlite@0.15.0
+  - cojson-transport-ws@0.15.0
+
+## 0.14.28
+
+### Patch Changes
+
+- Updated dependencies [06c5a1c]
+  - jazz-tools@0.14.28
+  - cojson@0.14.28
+  - cojson-storage-sqlite@0.14.28
+  - cojson-transport-ws@0.14.28
+
 ## 0.14.27
 
 ### Patch Changes

package/dist/tests/startSyncServer.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=startSyncServer.test.d.ts.map

package/dist/tests/startSyncServer.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"startSyncServer.test.d.ts","sourceRoot":"","sources":["../../src/tests/startSyncServer.test.ts"],"names":[],"mappings":""}

package/dist/tests/startSyncServer.test.js

@@ -0,0 +1,63 @@
+import { randomUUID } from "crypto";
+import { tmpdir } from "os";
+import { join } from "path";
+import { co, z } from "jazz-tools";
+import { startWorker } from "jazz-tools/worker";
+import { describe, expect, test } from "vitest";
+import { createWorkerAccount } from "../createWorkerAccount.js";
+import { startSyncServer } from "../startSyncServer.js";
+const TestMap = co.map({
+    value: z.string(),
+});
+describe("startSyncServer", () => {
+    test("persists values in storage and loads them after restart", async () => {
+        // Create a temporary database file
+        const dbPath = join(tmpdir(), `test-${randomUUID()}.db`);
+        // Start first server instance
+        const server1 = await startSyncServer({
+            port: "0", // Random available port
+            inMemory: false,
+            db: dbPath,
+        });
+        const port = server1.address().port;
+        const syncServer = `ws://localhost:${port}`;
+        // Create worker account and start first worker
+        const { accountID, agentSecret } = await createWorkerAccount({
+            name: "test-worker",
+            peer: syncServer,
+        });
+        const worker1 = await startWorker({
+            accountID,
+            accountSecret: agentSecret,
+            syncServer,
+        });
+        // Create and sync test data
+        const map = TestMap.create({ value: "testValue" });
+        // Close first server
+        await worker1.done();
+        server1.close();
+        // Start second server instance with same DB
+        const server2 = await startSyncServer({
+            port: "0",
+            inMemory: false,
+            db: dbPath,
+        });
+        const port2 = server2.address().port;
+        const syncServer2 = `ws://localhost:${port2}`;
+        // Start second worker with same account
+        const worker2 = await startWorker({
+            accountID,
+            accountSecret: agentSecret,
+            syncServer: syncServer2,
+        });
+        // Try to load the previously created map
+        const loadedMap = await TestMap.load(map.id, {});
+        // Verify the data persisted
+        expect(loadedMap).not.toBe(null);
+        expect(loadedMap?.value).toBe("testValue");
+        // Cleanup
+        await worker2.done();
+        server2.close();
+    });
+});
+//# sourceMappingURL=startSyncServer.test.js.map

package/dist/tests/startSyncServer.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"startSyncServer.test.js","sourceRoot":"","sources":["../../src/tests/startSyncServer.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC;AAC5B,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,EAAE,EAAE,CAAC,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAChD,OAAO,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAExD,MAAM,OAAO,GAAG,EAAE,CAAC,GAAG,CAAC;IACrB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;CAClB,CAAC,CAAC;AAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAI,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACzE,mCAAmC;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,EAAE,QAAQ,UAAU,EAAE,KAAK,CAAC,CAAC;QAEzD,8BAA8B;QAC9B,MAAM,OAAO,GAAG,MAAM,eAAe,CAAC;YACpC,IAAI,EAAE,GAAG,EAAE,wBAAwB;YACnC,QAAQ,EAAE,KAAK;YACf,EAAE,EAAE,MAAM;SACX,CAAC,CAAC;QAEH,MAAM,IAAI,GAAI,OAAO,CAAC,OAAO,EAAU,CAAC,IAAI,CAAC;QAC7C,MAAM,UAAU,GAAG,kBAAkB,IAAI,EAAE,CAAC;QAE5C,+CAA+C;QAC/C,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,MAAM,mBAAmB,CAAC;YAC3D,IAAI,EAAE,aAAa;YACnB,IAAI,EAAE,UAAU;SACjB,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC;YAChC,SAAS;YACT,aAAa,EAAE,WAAW;YAC1B,UAAU;SACX,CAAC,CAAC;QAEH,4BAA4B;QAC5B,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;QAEnD,qBAAqB;QACrB,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;QACrB,OAAO,CAAC,KAAK,EAAE,CAAC;QAEhB,4CAA4C;QAC5C,MAAM,OAAO,GAAG,MAAM,eAAe,CAAC;YACpC,IAAI,EAAE,GAAG;YACT,QAAQ,EAAE,KAAK;YACf,EAAE,EAAE,MAAM;SACX,CAAC,CAAC;QAEH,MAAM,KAAK,GAAI,OAAO,CAAC,OAAO,EAAU,CAAC,IAAI,CAAC;QAC9C,MAAM,WAAW,GAAG,kBAAkB,KAAK,EAAE,CAAC;QAE9C,wCAAwC;QACxC,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC;YAChC,SAAS;YACT,aAAa,EAAE,WAAW;YAC1B,UAAU,EAAE,WAAW;SACxB,CAAC,CAAC;QAEH,yCAAyC;QACzC,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QAEjD,4BAA4B;QAC5B,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjC,MAAM,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAE3C,UAAU;QACV,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;QACrB,OAAO,CAAC,KAAK,EAAE,CAAC;IAClB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/tests/startWorker.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=startWorker.test.d.ts.map

package/dist/tests/startWorker.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"startWorker.test.d.ts","sourceRoot":"","sources":["../../src/tests/startWorker.test.ts"],"names":[],"mappings":""}

package/dist/tests/startWorker.test.js

@@ -0,0 +1,229 @@
+import { randomUUID } from "node:crypto";
+import { unlinkSync } from "node:fs";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import { Group, InboxSender, co, z, } from "jazz-tools";
+import { startWorker } from "jazz-tools/worker";
+import { afterAll, describe, expect, onTestFinished, test } from "vitest";
+import { createWorkerAccount } from "../createWorkerAccount.js";
+import { startSyncServer } from "../startSyncServer.js";
+import { waitFor } from "./utils.js";
+const dbPath = join(tmpdir(), `test-${randomUUID()}.db`);
+afterAll(() => {
+    unlinkSync(dbPath);
+});
+async function setup(AccountSchema) {
+    const { server, port } = await setupSyncServer();
+    const syncServer = `ws://localhost:${port}`;
+    const { worker, done, waitForConnection, subscribeToConnectionChange } = await setupWorker(syncServer, AccountSchema);
+    return {
+        worker,
+        done,
+        syncServer,
+        server,
+        port,
+        waitForConnection,
+        subscribeToConnectionChange,
+    };
+}
+async function setupSyncServer(defaultPort = "0") {
+    const server = await startSyncServer({
+        port: defaultPort,
+        inMemory: false,
+        db: dbPath,
+    });
+    const port = server.address().port.toString();
+    onTestFinished(() => {
+        server.close();
+    });
+    return { server, port };
+}
+async function setupWorker(syncServer, AccountSchema) {
+    const { accountID, agentSecret } = await createWorkerAccount({
+        name: "test-worker",
+        peer: syncServer,
+    });
+    return startWorker({
+        accountID: accountID,
+        accountSecret: agentSecret,
+        syncServer,
+        AccountSchema,
+    });
+}
+const TestMap = co.map({
+    value: z.string(),
+});
+describe("startWorker integration", () => {
+    test("worker connects to sync server successfully", async () => {
+        const worker1 = await setup();
+        const worker2 = await setupWorker(worker1.syncServer);
+        const group = Group.create({ owner: worker1.worker });
+        group.addMember("everyone", "reader");
+        const map = TestMap.create({
+            value: "test",
+        }, { owner: group });
+        await map.waitForSync();
+        const mapOnWorker2 = await TestMap.load(map.id, { loadAs: worker2.worker });
+        expect(mapOnWorker2?.value).toBe("test");
+        await worker1.done();
+        await worker2.done();
+    });
+    test("worker handles successfully the custom account migration", async () => {
+        const AccountRoot = co.map({
+            value: z.string(),
+        });
+        let shouldReloadPreviousAccount = false;
+        const CustomAccount = co
+            .account({
+            root: AccountRoot,
+            profile: co.profile(),
+        })
+            .withMigration((account) => {
+            if (account.root === undefined) {
+                if (shouldReloadPreviousAccount) {
+                    throw new Error("Previous account not found");
+                }
+                shouldReloadPreviousAccount = true;
+                account.root = AccountRoot.create({
+                    value: "test",
+                }, account);
+            }
+        });
+        const worker1 = await setup(CustomAccount);
+        const { root } = await worker1.worker.ensureLoaded({
+            resolve: { root: true },
+        });
+        expect(root.value).toBe("test");
+        await worker1.done();
+        const worker2 = await startWorker({
+            accountID: worker1.worker.id,
+            accountSecret: worker1.worker._raw.core.node.getCurrentAgent().agentSecret,
+            syncServer: worker1.syncServer,
+            AccountSchema: CustomAccount,
+        });
+        const { root: root2 } = await worker2.worker.ensureLoaded({
+            resolve: { root: true },
+        });
+        expect(root2.value).toBe("test");
+        await worker2.done();
+    });
+    test("waits for all coValues to sync before resolving done", async () => {
+        const worker1 = await setup();
+        const group = Group.create({ owner: worker1.worker });
+        group.addMember("everyone", "reader");
+        const map = TestMap.create({
+            value: "test",
+        }, { owner: group });
+        await worker1.done();
+        const worker2 = await setupWorker(worker1.syncServer);
+        const mapOnWorker2 = await TestMap.load(map.id, { loadAs: worker2.worker });
+        expect(mapOnWorker2?.value).toBe("test");
+        await worker2.done();
+    });
+    test("reiceves the messages from the inbox", async () => {
+        const worker1 = await setup();
+        const worker2 = await setupWorker(worker1.syncServer);
+        const group = Group.create({ owner: worker1.worker });
+        const map = TestMap.create({
+            value: "Hello!",
+        }, { owner: group });
+        worker2.experimental.inbox.subscribe(TestMap, async (value) => {
+            return TestMap.create({
+                value: value.value + " Responded from the inbox",
+            }, { owner: value._owner });
+        });
+        const sender = await InboxSender.load(worker2.worker.id, worker1.worker);
+        const resultId = await sender.sendMessage(map);
+        const result = await TestMap.load(resultId, { loadAs: worker2.worker });
+        expect(result?.value).toEqual("Hello! Responded from the inbox");
+        await worker1.done();
+        await worker2.done();
+    });
+    // Flaky test, fails randomly on CI
+    test.skip("worker reconnects when sync server is closed and reopened", async () => {
+        const worker1 = await setup();
+        const worker2 = await setupWorker(worker1.syncServer);
+        const group = Group.create({ owner: worker1.worker });
+        group.addMember("everyone", "reader");
+        const map = TestMap.create({
+            value: "initial value",
+        }, { owner: group });
+        await map.waitForSync();
+        // Close the sync server
+        worker1.server.close();
+        // Create a new value while server is down
+        const map2 = TestMap.create({
+            value: "created while offline",
+        }, { owner: group });
+        map.value = "updated while offline";
+        // Start a new sync server on the same port
+        const newServer = await startSyncServer({
+            port: worker1.port,
+            inMemory: true,
+            db: "",
+        });
+        // Wait for reconnection
+        await worker1.waitForConnection();
+        await worker2.waitForConnection();
+        await worker1.worker.waitForAllCoValuesSync();
+        // Verify both old and new values are synced
+        const mapOnWorker2 = await TestMap.load(map.id, { loadAs: worker2.worker });
+        const map2OnWorker2 = await TestMap.load(map2.id, {
+            loadAs: worker2.worker,
+        });
+        expect(mapOnWorker2?.value).toBe("updated while offline");
+        expect(map2OnWorker2?.value).toBe("created while offline");
+        // Cleanup
+        await worker2.done();
+        newServer.close();
+    });
+    test("waitForConnection resolves when connection is established", async () => {
+        const worker1 = await setup();
+        // Initially should be connected
+        await worker1.waitForConnection();
+        // Close the sync server
+        worker1.server.close();
+        // Start a new sync server on the same port
+        const newServer = await startSyncServer({
+            port: worker1.port,
+            inMemory: true,
+            db: "",
+        });
+        // Should reconnect and resolve
+        await worker1.waitForConnection();
+        // Cleanup
+        await worker1.done();
+        newServer.close();
+    });
+    test("subscribeToConnectionChange notifies on connection state changes", async () => {
+        const worker1 = await setup();
+        const connectionStates = [];
+        // Subscribe to connection changes
+        const unsubscribe = worker1.subscribeToConnectionChange((isConnected) => {
+            connectionStates.push(isConnected);
+        });
+        await waitFor(() => {
+            expect(connectionStates).toEqual([true]);
+        });
+        // Close the sync server
+        worker1.server.close();
+        await waitFor(() => {
+            expect(connectionStates).toEqual([true, false]);
+        });
+        // Start a new sync server on the same port
+        const newServer = await startSyncServer({
+            port: worker1.port,
+            inMemory: true,
+            db: "",
+        });
+        // Wait a bit for the reconnection to be detected
+        await waitFor(() => {
+            expect(connectionStates).toEqual([true, false, true]);
+        });
+        // Cleanup
+        unsubscribe();
+        await worker1.done();
+        newServer.close();
+    });
+});
+//# sourceMappingURL=startWorker.test.js.map

package/dist/tests/startWorker.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"startWorker.test.js","sourceRoot":"","sources":["../../src/tests/startWorker.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAOL,KAAK,EACL,WAAW,EAEX,EAAE,EAEF,CAAC,GACF,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAC1E,OAAO,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAErC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,EAAE,QAAQ,UAAU,EAAE,KAAK,CAAC,CAAC;AAEzD,QAAQ,CAAC,GAAG,EAAE;IACZ,UAAU,CAAC,MAAM,CAAC,CAAC;AACrB,CAAC,CAAC,CAAC;AAEH,KAAK,UAAU,KAAK,CAIlB,aAAiB;IACjB,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,eAAe,EAAE,CAAC;IAEjD,MAAM,UAAU,GAAG,kBAAkB,IAAI,EAAE,CAAC;IAE5C,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,2BAA2B,EAAE,GACpE,MAAM,WAAW,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;IAE/C,OAAO;QACL,MAAM;QACN,IAAI;QACJ,UAAU;QACV,MAAM;QACN,IAAI;QACJ,iBAAiB;QACjB,2BAA2B;KAC5B,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,eAAe,CAAC,WAAW,GAAG,GAAG;IAC9C,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC;QACnC,IAAI,EAAE,WAAW;QACjB,QAAQ,EAAE,KAAK;QACf,EAAE,EAAE,MAAM;KACX,CAAC,CAAC;IAEH,MAAM,IAAI,GAAI,MAAM,CAAC,OAAO,EAAuB,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;IAEpE,cAAc,CAAC,GAAG,EAAE;QAClB,MAAM,CAAC,KAAK,EAAE,CAAC;IACjB,CAAC,CAAC,CAAC;IAEH,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;AAC1B,CAAC;AAED,KAAK,UAAU,WAAW,CAIxB,UAAkB,EAAE,aAAiB;IACrC,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,MAAM,mBAAmB,CAAC;QAC3D,IAAI,EAAE,aAAa;QACnB,IAAI,EAAE,UAAU;KACjB,CAAC,CAAC;IAEH,OAAO,WAAW,CAAC;QACjB,SAAS,EAAE,SAAS;QACpB,aAAa,EAAE,WAAW;QAC1B,UAAU;QACV,aAAa;KACd,CAAC,CAAC;AACL,CAAC;AAED,MAAM,OAAO,GAAG,EAAE,CAAC,GAAG,CAAC;IACrB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;CAClB,CAAC,CAAC;AAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,IAAI,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,OAAO,GAAG,MAAM,KAAK,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAEtD,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QACtD,KAAK,CAAC,SAAS,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAEtC,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CACxB;YACE,KAAK,EAAE,MAAM;SACd,EACD,EAAE,KAAK,EAAE,KAAK,EAAE,CACjB,CAAC;QAEF,MAAM,GAAG,CAAC,WAAW,EAAE,CAAC;QAExB,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAE5E,MAAM,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEzC,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;QACrB,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,WAAW,GAAG,EAAE,CAAC,GAAG,CAAC;YACzB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;SAClB,CAAC,CAAC;QAEH,IAAI,2BAA2B,GAAG,KAAK,CAAC;QAExC,MAAM,aAAa,GAAG,EAAE;aACrB,OAAO,CAAC;YACP,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,EAAE,CAAC,OAAO,EAAE;SACtB,CAAC;aACD,aAAa,CAAC,CAAC,OAAO,EAAE,EAAE;YACzB,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBAC/B,IAAI,2BAA2B,EAAE,CAAC;oBAChC,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;gBAChD,CAAC;gBAED,2BAA2B,GAAG,IAAI,CAAC;gBAEnC,OAAO,CAAC,IAAI,GAAG,WAAW,CAAC,MAAM,CAC/B;oBACE,KAAK,EAAE,MAAM;iBACd,EACD,OAAO,CACR,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QAEL,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,aAAa,CAAC,CAAC;QAE3C,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,YAAY,CAAC;YACjD,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;SACxB,CAAC,CAAC;QAEH,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEhC,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;QAErB,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC;YAChC,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,EAAE;YAC5B,aAAa,EACX,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC,WAAW;YAC7D,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,aAAa,EAAE,aAAa;SAC7B,CAAC,CAAC;QAEH,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,YAAY,CAAC;YACxD,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;SACxB,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEjC,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,OAAO,GAAG,MAAM,KAAK,EAAE,CAAC;QAE9B,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QACtD,KAAK,CAAC,SAAS,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAEtC,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CACxB;YACE,KAAK,EAAE,MAAM;SACd,EACD,EAAE,KAAK,EAAE,KAAK,EAAE,CACjB,CAAC;QAEF,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;QAErB,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAEtD,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAE5E,MAAM,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEzC,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,OAAO,GAAG,MAAM,KAAK,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAEtD,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QACtD,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CACxB;YACE,KAAK,EAAE,QAAQ;SAChB,EACD,EAAE,KAAK,EAAE,KAAK,EAAE,CACjB,CAAC;QAEF,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YAC5D,OAAO,OAAO,CAAC,MAAM,CACnB;gBACE,KAAK,EAAE,KAAK,CAAC,KAAK,GAAG,2BAA2B;aACjD,EACD,EAAE,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,CACxB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,IAAI,CAGnC,OAAO,CAAC,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QAErC,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAE/C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAExE,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,OAAO,CAAC,iCAAiC,CAAC,CAAC;QAEjE,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;QACrB,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,mCAAmC;IACnC,IAAI,CAAC,IAAI,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QAChF,MAAM,OAAO,GAAG,MAAM,KAAK,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAEtD,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QACtD,KAAK,CAAC,SAAS,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAEtC,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CACxB;YACE,KAAK,EAAE,eAAe;SACvB,EACD,EAAE,KAAK,EAAE,KAAK,EAAE,CACjB,CAAC;QAEF,MAAM,GAAG,CAAC,WAAW,EAAE,CAAC;QAExB,wBAAwB;QACxB,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAEvB,0CAA0C;QAC1C,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CACzB;YACE,KAAK,EAAE,uBAAuB;SAC/B,EACD,EAAE,KAAK,EAAE,KAAK,EAAE,CACjB,CAAC;QAEF,GAAG,CAAC,KAAK,GAAG,uBAAuB,CAAC;QAEpC,2CAA2C;QAC3C,MAAM,SAAS,GAAG,MAAM,eAAe,CAAC;YACtC,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,QAAQ,EAAE,IAAI;YACd,EAAE,EAAE,EAAE;SACP,CAAC,CAAC;QAEH,wBAAwB;QACxB,MAAM,OAAO,CAAC,iBAAiB,EAAE,CAAC;QAClC,MAAM,OAAO,CAAC,iBAAiB,EAAE,CAAC;QAElC,MAAM,OAAO,CAAC,MAAM,CAAC,sBAAsB,EAAE,CAAC;QAE9C,4CAA4C;QAC5C,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAC5E,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE;YAChD,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC,CAAC;QAEH,MAAM,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAC1D,MAAM,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAE3D,UAAU;QACV,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;QACrB,SAAS,CAAC,KAAK,EAAE,CAAC;IACpB,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,OAAO,GAAG,MAAM,KAAK,EAAE,CAAC;QAE9B,gCAAgC;QAChC,MAAM,OAAO,CAAC,iBAAiB,EAAE,CAAC;QAElC,wBAAwB;QACxB,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAEvB,2CAA2C;QAC3C,MAAM,SAAS,GAAG,MAAM,eAAe,CAAC;YACtC,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,QAAQ,EAAE,IAAI;YACd,EAAE,EAAE,EAAE;SACP,CAAC,CAAC;QAEH,+BAA+B;QAC/B,MAAM,OAAO,CAAC,iBAAiB,EAAE,CAAC;QAElC,UAAU;QACV,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;QACrB,SAAS,CAAC,KAAK,EAAE,CAAC;IACpB,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAClF,MAAM,OAAO,GAAG,MAAM,KAAK,EAAE,CAAC;QAE9B,MAAM,gBAAgB,GAAc,EAAE,CAAC;QAEvC,kCAAkC;QAClC,MAAM,WAAW,GAAG,OAAO,CAAC,2BAA2B,CAAC,CAAC,WAAW,EAAE,EAAE;YACtE,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,wBAAwB;QACxB,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAEvB,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,2CAA2C;QAC3C,MAAM,SAAS,GAAG,MAAM,eAAe,CAAC;YACtC,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,QAAQ,EAAE,IAAI;YACd,EAAE,EAAE,EAAE;SACP,CAAC,CAAC;QAEH,iDAAiD;QACjD,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,UAAU;QACV,WAAW,EAAE,CAAC;QACd,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;QACrB,SAAS,CAAC,KAAK,EAAE,CAAC;IACpB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/tests/utils.d.ts

@@ -0,0 +1,2 @@
+export declare function waitFor(callback: () => boolean | void): Promise<void>;
+//# sourceMappingURL=utils.d.ts.map

package/dist/tests/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/tests/utils.ts"],"names":[],"mappings":"AAAA,wBAAgB,OAAO,CAAC,QAAQ,EAAE,MAAM,OAAO,GAAG,IAAI,iBA0BrD"}

package/dist/tests/utils.js

@@ -0,0 +1,25 @@
+export function waitFor(callback) {
+    return new Promise((resolve, reject) => {
+        const checkPassed = () => {
+            try {
+                return { ok: callback(), error: null };
+            }
+            catch (error) {
+                return { ok: false, error };
+            }
+        };
+        let retries = 0;
+        const interval = setInterval(() => {
+            const { ok, error } = checkPassed();
+            if (ok !== false) {
+                clearInterval(interval);
+                resolve();
+            }
+            if (++retries > 10) {
+                clearInterval(interval);
+                reject(error);
+            }
+        }, 100);
+    });
+}
+//# sourceMappingURL=utils.js.map

package/dist/tests/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/tests/utils.ts"],"names":[],"mappings":"AAAA,MAAM,UAAU,OAAO,CAAC,QAA8B;IACpD,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC3C,MAAM,WAAW,GAAG,GAAG,EAAE;YACvB,IAAI,CAAC;gBACH,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;YACzC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;YAC9B,CAAC;QACH,CAAC,CAAC;QAEF,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,EAAE;YAChC,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,GAAG,WAAW,EAAE,CAAC;YAEpC,IAAI,EAAE,KAAK,KAAK,EAAE,CAAC;gBACjB,aAAa,CAAC,QAAQ,CAAC,CAAC;gBACxB,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,IAAI,EAAE,OAAO,GAAG,EAAE,EAAE,CAAC;gBACnB,aAAa,CAAC,QAAQ,CAAC,CAAC;gBACxB,MAAM,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC;QACH,CAAC,EAAE,GAAG,CAAC,CAAC;IACV,CAAC,CAAC,CAAC;AACL,CAAC"}

package/package.json

@@ -3,7 +3,7 @@
   "bin": "./dist/index.js",
   "type": "module",
   "license": "MIT",
-  "version": "0.14.27",
+  "version": "0.15.0",
   "exports": {
     "./startSyncServer": {
       "import": "./dist/startSyncServer.js",
@@ -21,11 +21,11 @@
     "@effect/printer-ansi": "^0.34.5",
     "@effect/schema": "^0.71.1",
     "@effect/typeclass": "^0.25.5",
-    "cojson": "0.14.27",
-    "cojson-storage-sqlite": "0.14.27",
-    "cojson-transport-ws": "0.14.27",
+    "cojson": "0.15.0",
+    "cojson-storage-sqlite": "0.15.0",
+    "cojson-transport-ws": "0.15.0",
     "effect": "^3.6.5",
-    "jazz-tools": "0.14.27",
+    "jazz-tools": "0.15.0",
     "ws": "^8.14.2"
   },
   "devDependencies": {

package/src/tests/startSyncServer.test.ts

@@ -0,0 +1,76 @@
+import { randomUUID } from "crypto";
+import { tmpdir } from "os";
+import { join } from "path";
+import { co, z } from "jazz-tools";
+import { startWorker } from "jazz-tools/worker";
+import { describe, expect, test } from "vitest";
+import { createWorkerAccount } from "../createWorkerAccount.js";
+import { startSyncServer } from "../startSyncServer.js";
+
+const TestMap = co.map({
+  value: z.string(),
+});
+
+describe("startSyncServer", () => {
+  test("persists values in storage and loads them after restart", async () => {
+    // Create a temporary database file
+    const dbPath = join(tmpdir(), `test-${randomUUID()}.db`);
+
+    // Start first server instance
+    const server1 = await startSyncServer({
+      port: "0", // Random available port
+      inMemory: false,
+      db: dbPath,
+    });
+
+    const port = (server1.address() as any).port;
+    const syncServer = `ws://localhost:${port}`;
+
+    // Create worker account and start first worker
+    const { accountID, agentSecret } = await createWorkerAccount({
+      name: "test-worker",
+      peer: syncServer,
+    });
+
+    const worker1 = await startWorker({
+      accountID,
+      accountSecret: agentSecret,
+      syncServer,
+    });
+
+    // Create and sync test data
+    const map = TestMap.create({ value: "testValue" });
+
+    // Close first server
+    await worker1.done();
+    server1.close();
+
+    // Start second server instance with same DB
+    const server2 = await startSyncServer({
+      port: "0",
+      inMemory: false,
+      db: dbPath,
+    });
+
+    const port2 = (server2.address() as any).port;
+    const syncServer2 = `ws://localhost:${port2}`;
+
+    // Start second worker with same account
+    const worker2 = await startWorker({
+      accountID,
+      accountSecret: agentSecret,
+      syncServer: syncServer2,
+    });
+
+    // Try to load the previously created map
+    const loadedMap = await TestMap.load(map.id, {});
+
+    // Verify the data persisted
+    expect(loadedMap).not.toBe(null);
+    expect(loadedMap?.value).toBe("testValue");
+
+    // Cleanup
+    await worker2.done();
+    server2.close();
+  });
+});

package/src/tests/startWorker.test.ts

@@ -0,0 +1,352 @@
+import { randomUUID } from "node:crypto";
+import { unlinkSync } from "node:fs";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import {
+  Account,
+  AccountClass,
+  AccountSchema,
+  AnyAccountSchema,
+  CoMap,
+  CoValueFromRaw,
+  Group,
+  InboxSender,
+  Loaded,
+  co,
+  coField,
+  z,
+} from "jazz-tools";
+import { startWorker } from "jazz-tools/worker";
+import { afterAll, describe, expect, onTestFinished, test } from "vitest";
+import { createWorkerAccount } from "../createWorkerAccount.js";
+import { startSyncServer } from "../startSyncServer.js";
+import { waitFor } from "./utils.js";
+
+const dbPath = join(tmpdir(), `test-${randomUUID()}.db`);
+
+afterAll(() => {
+  unlinkSync(dbPath);
+});
+
+async function setup<
+  S extends
+    | (AccountClass<Account> & CoValueFromRaw<Account>)
+    | AnyAccountSchema,
+>(AccountSchema?: S) {
+  const { server, port } = await setupSyncServer();
+
+  const syncServer = `ws://localhost:${port}`;
+
+  const { worker, done, waitForConnection, subscribeToConnectionChange } =
+    await setupWorker(syncServer, AccountSchema);
+
+  return {
+    worker,
+    done,
+    syncServer,
+    server,
+    port,
+    waitForConnection,
+    subscribeToConnectionChange,
+  };
+}
+
+async function setupSyncServer(defaultPort = "0") {
+  const server = await startSyncServer({
+    port: defaultPort,
+    inMemory: false,
+    db: dbPath,
+  });
+
+  const port = (server.address() as { port: number }).port.toString();
+
+  onTestFinished(() => {
+    server.close();
+  });
+
+  return { server, port };
+}
+
+async function setupWorker<
+  S extends
+    | (AccountClass<Account> & CoValueFromRaw<Account>)
+    | AnyAccountSchema,
+>(syncServer: string, AccountSchema?: S) {
+  const { accountID, agentSecret } = await createWorkerAccount({
+    name: "test-worker",
+    peer: syncServer,
+  });
+
+  return startWorker({
+    accountID: accountID,
+    accountSecret: agentSecret,
+    syncServer,
+    AccountSchema,
+  });
+}
+
+const TestMap = co.map({
+  value: z.string(),
+});
+
+describe("startWorker integration", () => {
+  test("worker connects to sync server successfully", async () => {
+    const worker1 = await setup();
+    const worker2 = await setupWorker(worker1.syncServer);
+
+    const group = Group.create({ owner: worker1.worker });
+    group.addMember("everyone", "reader");
+
+    const map = TestMap.create(
+      {
+        value: "test",
+      },
+      { owner: group },
+    );
+
+    await map.waitForSync();
+
+    const mapOnWorker2 = await TestMap.load(map.id, { loadAs: worker2.worker });
+
+    expect(mapOnWorker2?.value).toBe("test");
+
+    await worker1.done();
+    await worker2.done();
+  });
+
+  test("worker handles successfully the custom account migration", async () => {
+    const AccountRoot = co.map({
+      value: z.string(),
+    });
+
+    let shouldReloadPreviousAccount = false;
+
+    const CustomAccount = co
+      .account({
+        root: AccountRoot,
+        profile: co.profile(),
+      })
+      .withMigration((account) => {
+        if (account.root === undefined) {
+          if (shouldReloadPreviousAccount) {
+            throw new Error("Previous account not found");
+          }
+
+          shouldReloadPreviousAccount = true;
+
+          account.root = AccountRoot.create(
+            {
+              value: "test",
+            },
+            account,
+          );
+        }
+      });
+
+    const worker1 = await setup(CustomAccount);
+
+    const { root } = await worker1.worker.ensureLoaded({
+      resolve: { root: true },
+    });
+
+    expect(root.value).toBe("test");
+
+    await worker1.done();
+
+    const worker2 = await startWorker({
+      accountID: worker1.worker.id,
+      accountSecret:
+        worker1.worker._raw.core.node.getCurrentAgent().agentSecret,
+      syncServer: worker1.syncServer,
+      AccountSchema: CustomAccount,
+    });
+
+    const { root: root2 } = await worker2.worker.ensureLoaded({
+      resolve: { root: true },
+    });
+
+    expect(root2.value).toBe("test");
+
+    await worker2.done();
+  });
+
+  test("waits for all coValues to sync before resolving done", async () => {
+    const worker1 = await setup();
+
+    const group = Group.create({ owner: worker1.worker });
+    group.addMember("everyone", "reader");
+
+    const map = TestMap.create(
+      {
+        value: "test",
+      },
+      { owner: group },
+    );
+
+    await worker1.done();
+
+    const worker2 = await setupWorker(worker1.syncServer);
+
+    const mapOnWorker2 = await TestMap.load(map.id, { loadAs: worker2.worker });
+
+    expect(mapOnWorker2?.value).toBe("test");
+
+    await worker2.done();
+  });
+
+  test("reiceves the messages from the inbox", async () => {
+    const worker1 = await setup();
+    const worker2 = await setupWorker(worker1.syncServer);
+
+    const group = Group.create({ owner: worker1.worker });
+    const map = TestMap.create(
+      {
+        value: "Hello!",
+      },
+      { owner: group },
+    );
+
+    worker2.experimental.inbox.subscribe(TestMap, async (value) => {
+      return TestMap.create(
+        {
+          value: value.value + " Responded from the inbox",
+        },
+        { owner: value._owner },
+      );
+    });
+
+    const sender = await InboxSender.load<
+      Loaded<typeof TestMap>,
+      Loaded<typeof TestMap>
+    >(worker2.worker.id, worker1.worker);
+
+    const resultId = await sender.sendMessage(map);
+
+    const result = await TestMap.load(resultId, { loadAs: worker2.worker });
+
+    expect(result?.value).toEqual("Hello! Responded from the inbox");
+
+    await worker1.done();
+    await worker2.done();
+  });
+
+  // Flaky test, fails randomly on CI
+  test.skip("worker reconnects when sync server is closed and reopened", async () => {
+    const worker1 = await setup();
+    const worker2 = await setupWorker(worker1.syncServer);
+
+    const group = Group.create({ owner: worker1.worker });
+    group.addMember("everyone", "reader");
+
+    const map = TestMap.create(
+      {
+        value: "initial value",
+      },
+      { owner: group },
+    );
+
+    await map.waitForSync();
+
+    // Close the sync server
+    worker1.server.close();
+
+    // Create a new value while server is down
+    const map2 = TestMap.create(
+      {
+        value: "created while offline",
+      },
+      { owner: group },
+    );
+
+    map.value = "updated while offline";
+
+    // Start a new sync server on the same port
+    const newServer = await startSyncServer({
+      port: worker1.port,
+      inMemory: true,
+      db: "",
+    });
+
+    // Wait for reconnection
+    await worker1.waitForConnection();
+    await worker2.waitForConnection();
+
+    await worker1.worker.waitForAllCoValuesSync();
+
+    // Verify both old and new values are synced
+    const mapOnWorker2 = await TestMap.load(map.id, { loadAs: worker2.worker });
+    const map2OnWorker2 = await TestMap.load(map2.id, {
+      loadAs: worker2.worker,
+    });
+
+    expect(mapOnWorker2?.value).toBe("updated while offline");
+    expect(map2OnWorker2?.value).toBe("created while offline");
+
+    // Cleanup
+    await worker2.done();
+    newServer.close();
+  });
+
+  test("waitForConnection resolves when connection is established", async () => {
+    const worker1 = await setup();
+
+    // Initially should be connected
+    await worker1.waitForConnection();
+
+    // Close the sync server
+    worker1.server.close();
+
+    // Start a new sync server on the same port
+    const newServer = await startSyncServer({
+      port: worker1.port,
+      inMemory: true,
+      db: "",
+    });
+
+    // Should reconnect and resolve
+    await worker1.waitForConnection();
+
+    // Cleanup
+    await worker1.done();
+    newServer.close();
+  });
+
+  test("subscribeToConnectionChange notifies on connection state changes", async () => {
+    const worker1 = await setup();
+
+    const connectionStates: boolean[] = [];
+
+    // Subscribe to connection changes
+    const unsubscribe = worker1.subscribeToConnectionChange((isConnected) => {
+      connectionStates.push(isConnected);
+    });
+
+    await waitFor(() => {
+      expect(connectionStates).toEqual([true]);
+    });
+
+    // Close the sync server
+    worker1.server.close();
+
+    await waitFor(() => {
+      expect(connectionStates).toEqual([true, false]);
+    });
+
+    // Start a new sync server on the same port
+    const newServer = await startSyncServer({
+      port: worker1.port,
+      inMemory: true,
+      db: "",
+    });
+
+    // Wait a bit for the reconnection to be detected
+    await waitFor(() => {
+      expect(connectionStates).toEqual([true, false, true]);
+    });
+
+    // Cleanup
+    unsubscribe();
+    await worker1.done();
+    newServer.close();
+  });
+});

package/src/tests/utils.ts

@@ -0,0 +1,27 @@
+export function waitFor(callback: () => boolean | void) {
+  return new Promise<void>((resolve, reject) => {
+    const checkPassed = () => {
+      try {
+        return { ok: callback(), error: null };
+      } catch (error) {
+        return { ok: false, error };
+      }
+    };
+
+    let retries = 0;
+
+    const interval = setInterval(() => {
+      const { ok, error } = checkPassed();
+
+      if (ok !== false) {
+        clearInterval(interval);
+        resolve();
+      }
+
+      if (++retries > 10) {
+        clearInterval(interval);
+        reject(error);
+      }
+    }, 100);
+  });
+}