perf(canvas): reduce Convex hot-path query load

tests/canvas-graph-query-cache.test.ts

@@ -0,0 +1,66 @@
+import { describe, expect, it, vi } from "vitest";
+
+vi.mock("@/convex/_generated/api", () => ({
+  api: {
+    canvasGraph: { get: "canvasGraph.get" },
+  },
+}));
+
+import {
+  getCanvasGraphEdgesFromQuery,
+  getCanvasGraphNodesFromQuery,
+  setCanvasGraphEdgesInQuery,
+  setCanvasGraphNodesInQuery,
+} from "@/components/canvas/canvas-graph-query-cache";
+
+describe("canvas graph query cache helpers", () => {
+  it("returns cached nodes and edges from the shared graph query", () => {
+    const graph = {
+      nodes: [{ _id: "node_1" }],
+      edges: [{ _id: "edge_1" }],
+    };
+    const localStore = {
+      getQuery: vi.fn((_query, args) =>
+        args.canvasId === "canvas_1" ? graph : undefined,
+      ),
+    };
+
+    expect(getCanvasGraphNodesFromQuery(localStore as never, { canvasId: "canvas_1" as never })).toEqual(graph.nodes);
+    expect(getCanvasGraphEdgesFromQuery(localStore as never, { canvasId: "canvas_1" as never })).toEqual(graph.edges);
+  });
+
+  it("preserves the sibling collection when replacing nodes or edges", () => {
+    const graph = {
+      nodes: [{ _id: "node_1" }],
+      edges: [{ _id: "edge_1" }],
+    };
+    const localStore = {
+      getQuery: vi.fn((_query, args) =>
+        Object.keys(args).length === 1 && args.canvasId === "canvas_1"
+          ? graph
+          : undefined,
+      ),
+      setQuery: vi.fn(),
+    };
+
+    setCanvasGraphNodesInQuery(localStore as never, {
+      canvasId: "canvas_1" as never,
+      nodes: [{ _id: "node_2" }] as never,
+    });
+    setCanvasGraphEdgesInQuery(localStore as never, {
+      canvasId: "canvas_1" as never,
+      edges: [{ _id: "edge_2" }] as never,
+    });
+
+    expect(localStore.getQuery).toHaveBeenNthCalledWith(1, "canvasGraph.get", { canvasId: "canvas_1" });
+    expect(localStore.getQuery).toHaveBeenNthCalledWith(2, "canvasGraph.get", { canvasId: "canvas_1" });
+    expect(localStore.setQuery).toHaveBeenNthCalledWith(1, "canvasGraph.get", { canvasId: "canvas_1" }, {
+      nodes: [{ _id: "node_2" }],
+      edges: [{ _id: "edge_1" }],
+    });
+    expect(localStore.setQuery).toHaveBeenNthCalledWith(2, "canvasGraph.get", { canvasId: "canvas_1" }, {
+      nodes: [{ _id: "node_1" }],
+      edges: [{ _id: "edge_2" }],
+    });
+  });
+});