feat: implement daemon connectivity updates and relay protocol versioning

2026-02-20 19:27:56 +07:00
parent 6dad3212da
commit 0350f00840
49 changed files with 4146 additions and 1351 deletions
@@ -0,0 +1,100 @@
+import { describe, expect, it, vi, beforeEach } from "vitest";
+
+const daemonClientMock = vi.hoisted(() => {
+  const createdConfigs: Array<{ clientSessionKey?: string; url?: string }> = [];
+
+  class MockDaemonClient {
+    private statusHandlers = new Set<
+      (
+        message: {
+          type: "status";
+          payload: { status: string; serverId: string; hostname: string | null };
+        }
+      ) => void
+    >();
+
+    public lastError: string | null = null;
+
+    constructor(config: { clientSessionKey?: string; url?: string }) {
+      createdConfigs.push(config);
+    }
+
+    subscribeConnectionStatus(): () => void {
+      return () => undefined;
+    }
+
+    on(
+      event: "status",
+      handler: (
+        message: {
+          type: "status";
+          payload: { status: string; serverId: string; hostname: string | null };
+        }
+      ) => void
+    ): () => void {
+      if (event === "status") {
+        this.statusHandlers.add(handler);
+      }
+      return () => {
+        this.statusHandlers.delete(handler);
+      };
+    }
+
+    async connect(): Promise<void> {
+      const message = {
+        type: "status" as const,
+        payload: {
+          status: "server_info",
+          serverId: "srv_probe_test",
+          hostname: "probe-host",
+        },
+      };
+      for (const handler of this.statusHandlers) {
+        handler(message);
+      }
+    }
+
+    async ping(): Promise<{ rttMs: number }> {
+      return { rttMs: 42 };
+    }
+
+    async close(): Promise<void> {
+      return;
+    }
+  }
+
+  return {
+    MockDaemonClient,
+    createdConfigs,
+  };
+});
+
+vi.mock("@server/client/daemon-client", () => ({
+  DaemonClient: daemonClientMock.MockDaemonClient,
+}));
+
+describe("test-daemon-connection probe client identity", () => {
+  beforeEach(() => {
+    daemonClientMock.createdConfigs.length = 0;
+  });
+
+  it("uses isolated probe clientSessionKey values for direct latency probes", async () => {
+    const mod = await import("./test-daemon-connection");
+
+    await mod.measureConnectionLatency({
+      id: "direct:lan:6767",
+      type: "direct",
+      endpoint: "lan:6767",
+    });
+    await mod.measureConnectionLatency({
+      id: "direct:lan:6767",
+      type: "direct",
+      endpoint: "lan:6767",
+    });
+
+    const [first, second] = daemonClientMock.createdConfigs;
+    expect(first?.clientSessionKey).toMatch(/^clsk_probe_/);
+    expect(second?.clientSessionKey).toMatch(/^clsk_probe_/);
+    expect(first?.clientSessionKey).not.toBe(second?.clientSessionKey);
+  });
+});