more tests + half cache plugins

2026-05-07 09:52:20 -05:00
parent b0c88f8c08
commit bdc55f89fd
6 changed files with 503 additions and 4 deletions
@@ -0,0 +1,120 @@
+import { describe, expect, test } from "bun:test";
+import { pruneStaleClients } from "./stale-prune";
+import type { ClientInfo } from "../types";
+
+function makeClient(id: string, lastSeen: number, role: "client" | "viewer" = "client"): ClientInfo {
+  const calls: string[] = [];
+  return {
+    id,
+    lastSeen,
+    role,
+    ws: {
+      close(...args: any[]) { calls.push(`close:${args.join(",")}`); },
+      send() {},
+      _calls: calls,
+    },
+  } as any;
+}
+
+describe("pruneStaleClients", () => {
+  test("does nothing when all clients are fresh", () => {
+    const clients = new Map<string, ClientInfo>();
+    clients.set("a", makeClient("a", Date.now()));
+
+    const offlined: string[] = [];
+    const deleted: string[] = [];
+
+    pruneStaleClients({
+      clients,
+      staleMs: 60_000,
+      pruneBatch: 10,
+      setOnlineState: (id, online) => { if (!online) offlined.push(id); },
+      deleteClient: (id) => { deleted.push(id); },
+    });
+
+    expect(offlined).toEqual([]);
+    expect(deleted).toEqual([]);
+  });
+
+  test("closes and marks offline stale role=client entries without deleting", () => {
+    const clients = new Map<string, ClientInfo>();
+    const staleClient = makeClient("stale-c", Date.now() - 120_000, "client");
+    clients.set("stale-c", staleClient);
+
+    const offlined: string[] = [];
+    const deleted: string[] = [];
+
+    pruneStaleClients({
+      clients,
+      staleMs: 60_000,
+      pruneBatch: 10,
+      setOnlineState: (id, online) => { if (!online) offlined.push(id); },
+      deleteClient: (id) => { deleted.push(id); },
+    });
+
+    expect(offlined).toContain("stale-c");
+    expect(deleted).not.toContain("stale-c");
+    expect((staleClient.ws as any)._calls[0]).toStartWith("close:");
+  });
+
+  test("closes and deletes stale non-client (viewer) entries", () => {
+    const clients = new Map<string, ClientInfo>();
+    const viewer = makeClient("stale-v", Date.now() - 120_000, "viewer");
+    clients.set("stale-v", viewer);
+
+    const offlined: string[] = [];
+    const deleted: string[] = [];
+
+    pruneStaleClients({
+      clients,
+      staleMs: 60_000,
+      pruneBatch: 10,
+      setOnlineState: (id, online) => { if (!online) offlined.push(id); },
+      deleteClient: (id) => { deleted.push(id); },
+    });
+
+    expect(deleted).toContain("stale-v");
+    expect(offlined).toContain("stale-v");
+  });
+
+  test("respects pruneBatch limit", () => {
+    const clients = new Map<string, ClientInfo>();
+    for (let i = 0; i < 5; i++) {
+      clients.set(`s-${i}`, makeClient(`s-${i}`, Date.now() - 200_000, "viewer"));
+    }
+
+    let pruned = 0;
+
+    pruneStaleClients({
+      clients,
+      staleMs: 60_000,
+      pruneBatch: 2,
+      setOnlineState: () => { pruned++; },
+      deleteClient: () => {},
+    });
+
+    expect(pruned).toBe(2);
+  });
+
+  test("mixes fresh and stale correctly", () => {
+    const clients = new Map<string, ClientInfo>();
+    clients.set("fresh", makeClient("fresh", Date.now()));
+    clients.set("stale", makeClient("stale", Date.now() - 200_000, "viewer"));
+
+    const offlined: string[] = [];
+    const deleted: string[] = [];
+
+    pruneStaleClients({
+      clients,
+      staleMs: 60_000,
+      pruneBatch: 10,
+      setOnlineState: (id, online) => { if (!online) offlined.push(id); },
+      deleteClient: (id) => { deleted.push(id); },
+    });
+
+    expect(offlined).toContain("stale");
+    expect(offlined).not.toContain("fresh");
+    expect(deleted).toContain("stale");
+    expect(deleted).not.toContain("fresh");
+  });
+});