cojson 0.9.18 → 0.9.23

package/src/tests/testUtils.ts

@@ -17,6 +17,12 @@ import { expectGroup } from "../typeUtils/expectGroup.js";
 
 const Crypto = await WasmCrypto.create();
 
+const syncServer: {
+  current: undefined | LocalNode;
+} = {
+  current: undefined,
+};
+
 export function randomAnonymousAccountAndSessionID(): [
   ControlledAgent,
   SessionID,
@@ -33,25 +39,6 @@ export function createTestNode() {
   return new LocalNode(admin, session, Crypto);
 }
 
-export function connectTwoPeers(
-  a: LocalNode,
-  b: LocalNode,
-  aRole: "client" | "server",
-  bRole: "client" | "server",
-) {
-  const [aAsPeer, bAsPeer] = connectedPeers(
-    "peer:" + a.account.id,
-    "peer:" + b.account.id,
-    {
-      peer1role: aRole,
-      peer2role: bRole,
-    },
-  );
-
-  a.syncManager.addPeer(bAsPeer);
-  b.syncManager.addPeer(aAsPeer);
-}
-
 export async function createTwoConnectedNodes(
   node1Role: Peer["role"],
   node2Role: Peer["role"],
@@ -149,6 +136,25 @@ export async function createThreeConnectedNodes(
   };
 }
 
+export function connectTwoPeers(
+  a: LocalNode,
+  b: LocalNode,
+  aRole: "client" | "server",
+  bRole: "client" | "server",
+) {
+  const [aAsPeer, bAsPeer] = connectedPeers(
+    "peer:" + a.account.id,
+    "peer:" + b.account.id,
+    {
+      peer1role: aRole,
+      peer2role: bRole,
+    },
+  );
+
+  a.syncManager.addPeer(bAsPeer);
+  b.syncManager.addPeer(aAsPeer);
+}
+
 export function newGroup() {
   const [admin, sessionID] = randomAnonymousAccountAndSessionID();
 
@@ -368,3 +374,90 @@ export function createTestMetricReader() {
 export function tearDownTestMetricReader() {
   metrics.disable();
 }
+
+export function setupSyncServer() {
+  syncServer.current = createTestNode();
+  return syncServer.current;
+}
+
+export async function createConnectedTestAgentNode(opts = { connected: true }) {
+  if (!syncServer.current) {
+    throw new Error("Sync server not initialized");
+  }
+
+  const [admin, session] = randomAnonymousAccountAndSessionID();
+  const node = new LocalNode(admin, session, Crypto);
+
+  const { nodeToServerPeer, serverToNodePeer, messages, addServerPeer } =
+    connectNodeToSyncServer(node, opts.connected);
+
+  return { node, nodeToServerPeer, serverToNodePeer, messages, addServerPeer };
+}
+
+export async function createConnectedTestNode(opts = { connected: true }) {
+  if (!syncServer.current) {
+    throw new Error("Sync server not initialized");
+  }
+
+  const ctx = await LocalNode.withNewlyCreatedAccount({
+    peersToLoadFrom: [],
+    crypto: Crypto,
+    creationProps: { name: "Client" },
+  });
+
+  const { nodeToServerPeer, serverToNodePeer, messages, addServerPeer } =
+    connectNodeToSyncServer(ctx.node, opts.connected);
+
+  return {
+    node: ctx.node,
+    accountID: ctx.accountID,
+    nodeToServerPeer,
+    serverToNodePeer,
+    messages,
+    addServerPeer,
+  };
+}
+
+export function connectNodeToSyncServer(node: LocalNode, connected = true) {
+  if (!syncServer.current) {
+    throw new Error("Sync server not initialized");
+  }
+
+  const [nodeToServerPeer, serverToNodePeer] = connectedPeers(
+    syncServer.current.account.id,
+    node.account.id,
+    {
+      peer1role: "server",
+      peer2role: "client",
+    },
+  );
+
+  const messages: {
+    from: "client" | "server";
+    msg: SyncMessage;
+  }[] = [];
+
+  const serverPush = serverToNodePeer.outgoing.push;
+  serverToNodePeer.outgoing.push = (msg) => {
+    messages.push({ from: "server", msg });
+    return serverPush.call(serverToNodePeer.outgoing, msg);
+  };
+
+  const clientPush = nodeToServerPeer.outgoing.push;
+  nodeToServerPeer.outgoing.push = (msg) => {
+    messages.push({ from: "client", msg });
+    return clientPush.call(nodeToServerPeer.outgoing, msg);
+  };
+
+  syncServer.current.syncManager.addPeer(serverToNodePeer);
+  if (connected) {
+    node.syncManager.addPeer(nodeToServerPeer);
+  }
+
+  return {
+    nodeToServerPeer,
+    serverToNodePeer,
+    messages,
+    addServerPeer: () => node.syncManager.addPeer(nodeToServerPeer),
+  };
+}