cojson-transport-ws 0.15.8 → 0.15.9

package/src/tests/syncServer.ts

@@ -88,5 +88,6 @@ export const startSyncServer = async (port?: number) => {
     syncServer,
     port: actualPort,
     localNode,
+    wss,
   };
 };

package/src/tests/{integration.test.ts → webSocket.integration.test.ts}

@@ -1,3 +1,4 @@
+import { assert } from "node:console";
 import { ControlledAgent, type CryptoProvider, LocalNode } from "cojson";
 import { WasmCrypto } from "cojson/crypto/WasmCrypto";
 import { afterEach, beforeEach, describe, expect, test } from "vitest";
@@ -7,7 +8,7 @@ import { startSyncServer } from "./syncServer";
 import { waitFor } from "./utils";
 
 describe("WebSocket Peer Integration", () => {
-  let server: any;
+  let server: Awaited<ReturnType<typeof startSyncServer>>;
   let syncServerUrl: string;
   let crypto: CryptoProvider;
 
@@ -93,7 +94,11 @@ describe("WebSocket Peer Integration", () => {
     const serverNode = server.localNode;
     const serverMap = await serverNode.load(map.id);
 
-    expect(serverMap.get("testKey")).toBe("testValue");
+    if (serverMap === "unavailable") {
+      throw new Error("Server map is unavailable");
+    }
+
+    expect(serverMap.get("testKey")?.toString()).toBe("testValue");
   });
 
   test("should handle disconnection and cleanup", async () => {
@@ -161,4 +166,34 @@ describe("WebSocket Peer Integration", () => {
 
     expect(ws.readyState).toBe(WebSocket.CLOSED);
   });
+
+  test("calling terminate on the server should close the connection", async () => {
+    const ws = new WebSocket(syncServerUrl);
+    let disconnectCalled = false;
+
+    createWebSocketPeer({
+      id: "test-client",
+      websocket: ws,
+      role: "server",
+      onClose: () => {
+        disconnectCalled = true;
+      },
+    });
+
+    await waitFor(() => {
+      expect(server.wss.clients.size).toBe(1);
+    });
+
+    const peerOnServer = server.localNode.syncManager.getPeers()[0];
+
+    for (const client of server.wss.clients) {
+      client.terminate();
+    }
+
+    await waitFor(() => {
+      expect(disconnectCalled).toBe(true);
+    });
+
+    expect(peerOnServer?.closed).toBe(true);
+  });
 });

package/src/utils.ts

@@ -0,0 +1,30 @@
+import type { AnyWebSocket } from "./types.js";
+
+export const BUFFER_LIMIT = 100_000;
+export const BUFFER_LIMIT_POLLING_INTERVAL = 10;
+
+export function isWebSocketOpen(websocket: AnyWebSocket) {
+  return websocket.readyState === 1;
+}
+
+export function hasWebSocketTooMuchBufferedData(websocket: AnyWebSocket) {
+  return websocket.bufferedAmount > BUFFER_LIMIT && isWebSocketOpen(websocket);
+}
+
+export function waitForWebSocketOpen(websocket: AnyWebSocket) {
+  return new Promise<void>((resolve) => {
+    if (websocket.readyState === 1) {
+      resolve();
+    } else {
+      websocket.addEventListener("open", () => resolve(), { once: true });
+    }
+  });
+}
+
+export async function waitForWebSocketBufferedAmount(websocket: AnyWebSocket) {
+  while (hasWebSocketTooMuchBufferedData(websocket)) {
+    await new Promise<void>((resolve) =>
+      setTimeout(resolve, BUFFER_LIMIT_POLLING_INTERVAL),
+    );
+  }
+}

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

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

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

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

package/dist/tests/BatchedOutgoingMessages.test.js

@@ -1,112 +0,0 @@
-import { afterEach, beforeEach, describe, expect, test, vi } from "vitest";
-import { BatchedOutgoingMessages, MAX_OUTGOING_MESSAGES_CHUNK_BYTES, } from "../BatchedOutgoingMessages.js";
-beforeEach(() => {
-    vi.useFakeTimers();
-});
-afterEach(() => {
-    vi.useRealTimers();
-});
-describe("BatchedOutgoingMessages", () => {
-    function setup() {
-        const sendMock = vi.fn();
-        const batchedMessages = new BatchedOutgoingMessages(sendMock);
-        return { sendMock, batchedMessages };
-    }
-    test("should batch messages and send them after a timeout", () => {
-        const { sendMock, batchedMessages } = setup();
-        const message1 = {
-            action: "known",
-            id: "co_z1",
-            header: false,
-            sessions: {},
-        };
-        const message2 = {
-            action: "known",
-            id: "co_z2",
-            header: false,
-            sessions: {},
-        };
-        batchedMessages.push(message1);
-        batchedMessages.push(message2);
-        expect(sendMock).not.toHaveBeenCalled();
-        vi.runAllTimers();
-        expect(sendMock).toHaveBeenCalledTimes(1);
-        expect(sendMock).toHaveBeenCalledWith(`${JSON.stringify(message1)}\n${JSON.stringify(message2)}`);
-    });
-    test("should send messages immediately when reaching MAX_OUTGOING_MESSAGES_CHUNK_BYTES", () => {
-        const { sendMock, batchedMessages } = setup();
-        const largeMessage = {
-            action: "known",
-            id: "co_z_large",
-            header: false,
-            sessions: {
-                // Add a large payload to exceed MAX_OUTGOING_MESSAGES_CHUNK_BYTES
-                payload: "x".repeat(MAX_OUTGOING_MESSAGES_CHUNK_BYTES),
-            },
-        };
-        batchedMessages.push(largeMessage);
-        expect(sendMock).toHaveBeenCalledTimes(1);
-        expect(sendMock).toHaveBeenCalledWith(JSON.stringify(largeMessage));
-    });
-    test("should send accumulated messages before a large message", () => {
-        const { sendMock, batchedMessages } = setup();
-        const smallMessage = {
-            action: "known",
-            id: "co_z_small",
-            header: false,
-            sessions: {},
-        };
-        const largeMessage = {
-            action: "known",
-            id: "co_z_large",
-            header: false,
-            sessions: {
-                // Add a large payload to exceed MAX_OUTGOING_MESSAGES_CHUNK_BYTES
-                payload: "x".repeat(MAX_OUTGOING_MESSAGES_CHUNK_BYTES),
-            },
-        };
-        batchedMessages.push(smallMessage);
-        batchedMessages.push(largeMessage);
-        vi.runAllTimers();
-        expect(sendMock).toHaveBeenCalledTimes(2);
-        expect(sendMock).toHaveBeenNthCalledWith(1, JSON.stringify(smallMessage));
-        expect(sendMock).toHaveBeenNthCalledWith(2, JSON.stringify(largeMessage));
-    });
-    test("should send remaining messages on close", () => {
-        const { sendMock, batchedMessages } = setup();
-        const message = {
-            action: "known",
-            id: "co_z_test",
-            header: false,
-            sessions: {},
-        };
-        batchedMessages.push(message);
-        expect(sendMock).not.toHaveBeenCalled();
-        batchedMessages.close();
-        expect(sendMock).toHaveBeenCalledTimes(1);
-        expect(sendMock).toHaveBeenCalledWith(JSON.stringify(message));
-    });
-    test("should clear timeout when pushing new messages", () => {
-        const { sendMock, batchedMessages } = setup();
-        const message1 = {
-            action: "known",
-            id: "co_z1",
-            header: false,
-            sessions: {},
-        };
-        const message2 = {
-            action: "known",
-            id: "co_z2",
-            header: false,
-            sessions: {},
-        };
-        batchedMessages.push(message1);
-        const clearTimeoutSpy = vi.spyOn(global, "clearTimeout");
-        batchedMessages.push(message2);
-        expect(clearTimeoutSpy).toHaveBeenCalled();
-        vi.runAllTimers();
-        expect(sendMock).toHaveBeenCalledTimes(1);
-        expect(sendMock).toHaveBeenCalledWith(`${JSON.stringify(message1)}\n${JSON.stringify(message2)}`);
-    });
-});
-//# sourceMappingURL=BatchedOutgoingMessages.test.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"BatchedOutgoingMessages.test.js","sourceRoot":"","sources":["../../src/tests/BatchedOutgoingMessages.test.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC3E,OAAO,EACL,uBAAuB,EACvB,iCAAiC,GAClC,MAAM,+BAA+B,CAAC;AAEvC,UAAU,CAAC,GAAG,EAAE;IACd,EAAE,CAAC,aAAa,EAAE,CAAC;AACrB,CAAC,CAAC,CAAC;AAEH,SAAS,CAAC,GAAG,EAAE;IACb,EAAE,CAAC,aAAa,EAAE,CAAC;AACrB,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,SAAS,KAAK;QACZ,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,eAAe,GAAG,IAAI,uBAAuB,CAAC,QAAQ,CAAC,CAAC;QAC9D,OAAO,EAAE,QAAQ,EAAE,eAAe,EAAE,CAAC;IACvC,CAAC;IAED,IAAI,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC/D,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,KAAK,EAAE,CAAC;QAC9C,MAAM,QAAQ,GAAgB;YAC5B,MAAM,EAAE,OAAO;YACf,EAAE,EAAE,OAAO;YACX,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE,EAAE;SACb,CAAC;QACF,MAAM,QAAQ,GAAgB;YAC5B,MAAM,EAAE,OAAO;YACf,EAAE,EAAE,OAAO;YACX,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE,EAAE;SACb,CAAC;QAEF,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/B,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAE/B,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAExC,EAAE,CAAC,YAAY,EAAE,CAAC;QAElB,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CACnC,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAC3D,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,kFAAkF,EAAE,GAAG,EAAE;QAC5F,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,KAAK,EAAE,CAAC;QAC9C,MAAM,YAAY,GAAgB;YAChC,MAAM,EAAE,OAAO;YACf,EAAE,EAAE,YAAY;YAChB,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE;gBACR,kEAAkE;gBAClE,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,iCAAiC,CAAC;aACF;SACvD,CAAC;QAEF,eAAe,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAEnC,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACnE,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,KAAK,EAAE,CAAC;QAC9C,MAAM,YAAY,GAAgB;YAChC,MAAM,EAAE,OAAO;YACf,EAAE,EAAE,YAAY;YAChB,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE,EAAE;SACb,CAAC;QACF,MAAM,YAAY,GAAgB;YAChC,MAAM,EAAE,OAAO;YACf,EAAE,EAAE,YAAY;YAChB,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE;gBACR,kEAAkE;gBAClE,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,iCAAiC,CAAC;aACF;SACvD,CAAC;QAEF,eAAe,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACnC,eAAe,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAEnC,EAAE,CAAC,YAAY,EAAE,CAAC;QAElB,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC;QAC1E,MAAM,CAAC,QAAQ,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACnD,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,KAAK,EAAE,CAAC;QAC9C,MAAM,OAAO,GAAgB;YAC3B,MAAM,EAAE,OAAO;YACf,EAAE,EAAE,WAAW;YACf,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE,EAAE;SACb,CAAC;QAEF,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9B,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAExC,eAAe,CAAC,KAAK,EAAE,CAAC;QAExB,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,gDAAgD,EAAE,GAAG,EAAE;QAC1D,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,KAAK,EAAE,CAAC;QAC9C,MAAM,QAAQ,GAAgB;YAC5B,MAAM,EAAE,OAAO;YACf,EAAE,EAAE,OAAO;YACX,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE,EAAE;SACb,CAAC;QACF,MAAM,QAAQ,GAAgB;YAC5B,MAAM,EAAE,OAAO;YACf,EAAE,EAAE,OAAO;YACX,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE,EAAE;SACb,CAAC;QAEF,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAE/B,MAAM,eAAe,GAAG,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;QAEzD,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAE/B,MAAM,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAE3C,EAAE,CAAC,YAAY,EAAE,CAAC;QAElB,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CACnC,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAC3D,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

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

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

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

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

@@ -1 +0,0 @@
-{"version":3,"file":"integration.test.js","sourceRoot":"","sources":["../../src/tests/integration.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAwC,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AACtD,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AACvE,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC/B,OAAO,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAC7D,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAElC,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;IAC1C,IAAI,MAAW,CAAC;IAChB,IAAI,aAAqB,CAAC;IAC1B,IAAI,MAAsB,CAAC;IAE3B,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,GAAG,MAAM,UAAU,CAAC,MAAM,EAAE,CAAC;QACnC,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;QACvC,MAAM,GAAG,MAAM,CAAC;QAChB,aAAa,GAAG,MAAM,CAAC,UAAU,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,CAAC,KAAK,EAAE,CAAC;IACjB,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC7E,qBAAqB;QACrB,MAAM,WAAW,GAAG,MAAM,CAAC,oBAAoB,EAAE,CAAC;QAClD,MAAM,UAAU,GAAG,IAAI,SAAS,CAC9B,WAAW,EACX,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,EACzD,MAAM,CACP,CAAC;QAEF,8BAA8B;QAC9B,MAAM,EAAE,GAAG,IAAI,SAAS,CAAC,aAAa,CAAC,CAAC;QAExC,2BAA2B;QAC3B,IAAI,qBAAqB,GAAG,KAAK,CAAC;QAElC,qCAAqC;QACrC,MAAM,IAAI,GAAG,mBAAmB,CAAC;YAC/B,EAAE,EAAE,aAAa;YACjB,SAAS,EAAE,EAAE;YACb,IAAI,EAAE,QAAQ;YACd,SAAS,EAAE,GAAG,EAAE;gBACd,qBAAqB,GAAG,IAAI,CAAC;YAC/B,CAAC;SACF,CAAC,CAAC;QAEH,UAAU,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAErC,mCAAmC;QACnC,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YAClC,MAAM,eAAe,GAAG,WAAW,CAAC,GAAG,EAAE;gBACvC,IAAI,qBAAqB,EAAE,CAAC;oBAC1B,aAAa,CAAC,eAAe,CAAC,CAAC;oBAC/B,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC,EAAE,GAAG,CAAC,CAAC;QACV,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,qBAAqB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,WAAW,GAAG,MAAM,CAAC,oBAAoB,EAAE,CAAC;QAClD,MAAM,UAAU,GAAG,IAAI,SAAS,CAC9B,WAAW,EACX,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,EACzD,MAAM,CACP,CAAC;QAEF,MAAM,EAAE,GAAG,IAAI,SAAS,CAAC,aAAa,CAAC,CAAC;QAExC,MAAM,IAAI,GAAG,mBAAmB,CAAC;YAC/B,EAAE,EAAE,aAAa;YACjB,SAAS,EAAE,EAAE;YACb,IAAI,EAAE,QAAQ;SACf,CAAC,CAAC;QAEH,UAAU,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAErC,sBAAsB;QACtB,MAAM,KAAK,GAAG,UAAU,CAAC,WAAW,EAAE,CAAC;QACvC,MAAM,GAAG,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC;QAC9B,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,WAAW,EAAE,UAAU,CAAC,CAAC;QAE5C,gBAAgB;QAChB,MAAM,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;QAE7B,iCAAiC;QACjC,MAAM,UAAU,GAAG,MAAM,CAAC,SAAS,CAAC;QACpC,MAAM,SAAS,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAEhD,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,WAAW,GAAG,MAAM,CAAC,oBAAoB,EAAE,CAAC;QAClD,MAAM,UAAU,GAAG,IAAI,SAAS,CAC9B,WAAW,EACX,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,EACzD,MAAM,CACP,CAAC;QAEF,MAAM,EAAE,GAAG,IAAI,SAAS,CAAC,aAAa,CAAC,CAAC;QACxC,IAAI,gBAAgB,GAAG,KAAK,CAAC;QAE7B,MAAM,IAAI,GAAG,mBAAmB,CAAC;YAC/B,EAAE,EAAE,aAAa;YACjB,SAAS,EAAE,EAAE;YACb,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,GAAG,EAAE;gBACZ,gBAAgB,GAAG,IAAI,CAAC;YAC1B,CAAC;SACF,CAAC,CAAC;QAEH,UAAU,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAErC,mCAAmC;QACnC,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QAEzD,mBAAmB;QACnB,MAAM,CAAC,KAAK,EAAE,CAAC;QAEf,+BAA+B;QAC/B,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QAEzD,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,WAAW,GAAG,MAAM,CAAC,oBAAoB,EAAE,CAAC;QAClD,MAAM,UAAU,GAAG,IAAI,SAAS,CAC9B,WAAW,EACX,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,EACzD,MAAM,CACP,CAAC;QAEF,MAAM,EAAE,GAAG,IAAI,SAAS,CAAC,aAAa,CAAC,CAAC;QACxC,IAAI,gBAAgB,GAAG,KAAK,CAAC;QAE7B,MAAM,IAAI,GAAG,mBAAmB,CAAC;YAC/B,EAAE,EAAE,aAAa;YACjB,SAAS,EAAE,EAAE;YACb,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,CAAC;YACd,OAAO,EAAE,GAAG,EAAE;gBACZ,gBAAgB,GAAG,IAAI,CAAC;YAC1B,CAAC;SACF,CAAC,CAAC;QAEH,UAAU,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAErC,8DAA8D;QAC9D,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/src/tests/BatchedOutgoingMessages.test.ts

@@ -1,146 +0,0 @@
-import type { SyncMessage } from "cojson";
-import type { CojsonInternalTypes } from "cojson";
-import { afterEach, beforeEach, describe, expect, test, vi } from "vitest";
-import {
-  BatchedOutgoingMessages,
-  MAX_OUTGOING_MESSAGES_CHUNK_BYTES,
-} from "../BatchedOutgoingMessages.js";
-
-beforeEach(() => {
-  vi.useFakeTimers();
-});
-
-afterEach(() => {
-  vi.useRealTimers();
-});
-
-describe("BatchedOutgoingMessages", () => {
-  function setup() {
-    const sendMock = vi.fn();
-    const batchedMessages = new BatchedOutgoingMessages(sendMock);
-    return { sendMock, batchedMessages };
-  }
-
-  test("should batch messages and send them after a timeout", () => {
-    const { sendMock, batchedMessages } = setup();
-    const message1: SyncMessage = {
-      action: "known",
-      id: "co_z1",
-      header: false,
-      sessions: {},
-    };
-    const message2: SyncMessage = {
-      action: "known",
-      id: "co_z2",
-      header: false,
-      sessions: {},
-    };
-
-    batchedMessages.push(message1);
-    batchedMessages.push(message2);
-
-    expect(sendMock).not.toHaveBeenCalled();
-
-    vi.runAllTimers();
-
-    expect(sendMock).toHaveBeenCalledTimes(1);
-    expect(sendMock).toHaveBeenCalledWith(
-      `${JSON.stringify(message1)}\n${JSON.stringify(message2)}`,
-    );
-  });
-
-  test("should send messages immediately when reaching MAX_OUTGOING_MESSAGES_CHUNK_BYTES", () => {
-    const { sendMock, batchedMessages } = setup();
-    const largeMessage: SyncMessage = {
-      action: "known",
-      id: "co_z_large",
-      header: false,
-      sessions: {
-        // Add a large payload to exceed MAX_OUTGOING_MESSAGES_CHUNK_BYTES
-        payload: "x".repeat(MAX_OUTGOING_MESSAGES_CHUNK_BYTES),
-      } as CojsonInternalTypes.CoValueKnownState["sessions"],
-    };
-
-    batchedMessages.push(largeMessage);
-
-    expect(sendMock).toHaveBeenCalledTimes(1);
-    expect(sendMock).toHaveBeenCalledWith(JSON.stringify(largeMessage));
-  });
-
-  test("should send accumulated messages before a large message", () => {
-    const { sendMock, batchedMessages } = setup();
-    const smallMessage: SyncMessage = {
-      action: "known",
-      id: "co_z_small",
-      header: false,
-      sessions: {},
-    };
-    const largeMessage: SyncMessage = {
-      action: "known",
-      id: "co_z_large",
-      header: false,
-      sessions: {
-        // Add a large payload to exceed MAX_OUTGOING_MESSAGES_CHUNK_BYTES
-        payload: "x".repeat(MAX_OUTGOING_MESSAGES_CHUNK_BYTES),
-      } as CojsonInternalTypes.CoValueKnownState["sessions"],
-    };
-
-    batchedMessages.push(smallMessage);
-    batchedMessages.push(largeMessage);
-
-    vi.runAllTimers();
-
-    expect(sendMock).toHaveBeenCalledTimes(2);
-    expect(sendMock).toHaveBeenNthCalledWith(1, JSON.stringify(smallMessage));
-    expect(sendMock).toHaveBeenNthCalledWith(2, JSON.stringify(largeMessage));
-  });
-
-  test("should send remaining messages on close", () => {
-    const { sendMock, batchedMessages } = setup();
-    const message: SyncMessage = {
-      action: "known",
-      id: "co_z_test",
-      header: false,
-      sessions: {},
-    };
-
-    batchedMessages.push(message);
-    expect(sendMock).not.toHaveBeenCalled();
-
-    batchedMessages.close();
-
-    expect(sendMock).toHaveBeenCalledTimes(1);
-    expect(sendMock).toHaveBeenCalledWith(JSON.stringify(message));
-  });
-
-  test("should clear timeout when pushing new messages", () => {
-    const { sendMock, batchedMessages } = setup();
-    const message1: SyncMessage = {
-      action: "known",
-      id: "co_z1",
-      header: false,
-      sessions: {},
-    };
-    const message2: SyncMessage = {
-      action: "known",
-      id: "co_z2",
-      header: false,
-      sessions: {},
-    };
-
-    batchedMessages.push(message1);
-
-    const clearTimeoutSpy = vi.spyOn(global, "clearTimeout");
-
-    batchedMessages.push(message2);
-
-    expect(clearTimeoutSpy).toHaveBeenCalled();
-
-    vi.runAllTimers();
-
-    expect(sendMock).toHaveBeenCalledTimes(1);
-    expect(sendMock).toHaveBeenCalledWith(
-      `${JSON.stringify(message1)}\n${JSON.stringify(message2)}`,
-    );
-  });
-});