@unerr-ai/unerr 0.1.5 → 0.1.7

package/dist/__tests__/community-tools.test.js

@@ -1,299 +0,0 @@
-/**
- * Leapfrog Sprint A.4 + A.5 TEST: Community MCP tools.
- *
- * Tests the community query methods on CozoGraphStore and the QueryRouter's
- * handling of get_cross_boundary_links and get_critical_nodes tools.
- * Uses mock graph store (consistent with query-router.test.ts pattern).
- */
-import { describe, expect, it, vi } from "vitest";
-import { QueryRouter } from "../intelligence/query-router.js";
-function createMockGraphStore(opts) {
-    const defaultEntity = {
-        key: "fn1",
-        kind: "function",
-        name: "doStuff",
-        file_path: "src/index.ts",
-        start_line: 10,
-        signature: "()",
-        body: "function doStuff() {}",
-        fan_in: 0,
-        fan_out: 0,
-        risk_level: "normal",
-        community: -1,
-    };
-    return {
-        db: { run: vi.fn().mockReturnValue({ rows: [] }) },
-        hasRules: vi.fn().mockReturnValue(false),
-        hasJustifications: vi.fn().mockReturnValue(false),
-        getPatterns: vi.fn().mockReturnValue([]),
-        getEntity: vi
-            .fn()
-            .mockReturnValue(opts?.entityOverride !== undefined ? opts.entityOverride : defaultEntity),
-        getCallersOf: vi.fn().mockReturnValue([]),
-        getCalleesOf: vi.fn().mockReturnValue([]),
-        getEntitiesByFile: vi.fn().mockReturnValue([]),
-        searchEntities: vi.fn().mockReturnValue([]),
-        getImports: vi.fn().mockReturnValue([]),
-        getRules: vi.fn().mockReturnValue([]),
-        getBlastRadius: vi.fn().mockReturnValue({
-            direct_callers: 0,
-            direct_callees: 0,
-            transitive_count: 0,
-            transitive_depth: 2,
-            is_chokepoint: false,
-            summary: "No dependencies",
-        }),
-        getBlastRadiusEntities: vi.fn().mockReturnValue([]),
-        getConventionsForEntity: vi.fn().mockReturnValue([]),
-        getDriftEntitiesForFile: vi.fn().mockReturnValue([]),
-        getDriftSummary: vi.fn().mockReturnValue({
-            total: 0,
-            added: 0,
-            modified: 0,
-            deleted: 0,
-            dependency_changed: 0,
-        }),
-        upsertDriftEntity: vi.fn(),
-        removeDriftEntity: vi.fn(),
-        clearDriftOverlay: vi.fn(),
-        getCrossBoundaryLinks: vi
-            .fn()
-            .mockReturnValue(opts?.crossBoundaryLinks ?? []),
-        getCriticalNodes: vi.fn().mockReturnValue(opts?.criticalNodes ?? []),
-        getAllCommunities: vi.fn().mockReturnValue(opts?.communities ?? []),
-        getCommunityForEntity: vi
-            .fn()
-            .mockReturnValue(opts?.communityForEntity ?? null),
-        getCrossCommunityEdges: vi
-            .fn()
-            .mockReturnValue(opts?.crossCommunityEdges ?? []),
-        hasDeepDiveProject: vi.fn().mockReturnValue(false),
-    };
-}
-// ── A.4: Cross Boundary Links Tool ─────────────────────────────
-describe("get_cross_boundary_links (A.4)", () => {
-    it("routes to local and returns cross-community edges", async () => {
-        const connections = [
-            {
-                from_name: "refresh",
-                from_file: "src/auth/refresh.ts",
-                from_community: 0,
-                from_community_label: "auth",
-                to_name: "charge",
-                to_file: "src/payment/charge.ts",
-                to_community: 1,
-                to_community_label: "payment",
-                edge_type: "calls",
-                surprise_score: 0.85,
-            },
-            {
-                from_name: "validate",
-                from_file: "src/payment/validate.ts",
-                from_community: 1,
-                from_community_label: "payment",
-                to_name: "log",
-                to_file: "src/utils/logger.ts",
-                to_community: 2,
-                to_community_label: "utils",
-                edge_type: "calls",
-                surprise_score: 0.72,
-            },
-        ];
-        const graph = createMockGraphStore({ crossBoundaryLinks: connections });
-        const router = new QueryRouter(graph);
-        const result = await router.execute("get_cross_boundary_links", {
-            top_n: 10,
-        });
-        expect(result._meta.source).toBe("local");
-        expect(result._meta.latency_ms).toBeLessThan(50);
-        expect(result._meta.format).toBe("columnar");
-        expect(typeof result.content).toBe("string");
-        const col = result.content;
-        expect(col).toContain("_fmt:columnar");
-        expect(col).toContain("refresh");
-        expect(col).toContain("charge");
-    });
-    it("passes community_id filter to graph store", async () => {
-        const graph = createMockGraphStore();
-        const router = new QueryRouter(graph);
-        await router.execute("get_cross_boundary_links", {
-            community_id: 3,
-            top_n: 5,
-        });
-        expect(graph.getCrossBoundaryLinks).toHaveBeenCalledWith(3, 5);
-    });
-    it("defaults top_n to 20", async () => {
-        const graph = createMockGraphStore();
-        const router = new QueryRouter(graph);
-        await router.execute("get_cross_boundary_links", {});
-        expect(graph.getCrossBoundaryLinks).toHaveBeenCalledWith(undefined, 20);
-    });
-    it("is registered as a local route", () => {
-        const graph = createMockGraphStore();
-        const router = new QueryRouter(graph);
-        expect(router.isKnownTool("get_cross_boundary_links")).toBe(true);
-    });
-});
-// ── A.5: Critical Nodes Tool ──────────────────────────────────────────
-describe("get_critical_nodes (A.5)", () => {
-    it("routes to local and returns highest-degree entities", async () => {
-        const criticalNodes = [
-            {
-                key: "pay_charge",
-                name: "charge",
-                file_path: "src/payment/charge.ts",
-                fan_in: 8,
-                fan_out: 5,
-                degree: 13,
-                community: 1,
-                community_label: "payment",
-                risk_level: "high",
-            },
-            {
-                key: "auth_login",
-                name: "login",
-                file_path: "src/auth/login.ts",
-                fan_in: 6,
-                fan_out: 3,
-                degree: 9,
-                community: 0,
-                community_label: "auth",
-                risk_level: "medium",
-            },
-        ];
-        const graph = createMockGraphStore({ criticalNodes });
-        const router = new QueryRouter(graph);
-        const result = await router.execute("get_critical_nodes", { top_n: 5 });
-        expect(result._meta.source).toBe("local");
-        expect(result._meta.format).toBe("columnar");
-        expect(typeof result.content).toBe("string");
-        const col = result.content;
-        expect(col).toContain("pay_charge");
-        expect(col).toContain("auth_login");
-        expect(col).toContain("_fmt:columnar");
-    });
-    it("passes community_id filter to graph store", async () => {
-        const graph = createMockGraphStore();
-        const router = new QueryRouter(graph);
-        await router.execute("get_critical_nodes", { top_n: 3, community_id: 2 });
-        expect(graph.getCriticalNodes).toHaveBeenCalledWith(3, 2);
-    });
-    it("defaults top_n to 10", async () => {
-        const graph = createMockGraphStore();
-        const router = new QueryRouter(graph);
-        await router.execute("get_critical_nodes", {});
-        expect(graph.getCriticalNodes).toHaveBeenCalledWith(10, undefined);
-    });
-    it("is registered as a local route", () => {
-        const graph = createMockGraphStore();
-        const router = new QueryRouter(graph);
-        expect(router.isKnownTool("get_critical_nodes")).toBe(true);
-    });
-});
-// ── A.3: Community Context Enrichment ────────────────────────────
-// `_context.community` (string field) was replaced by a context signal in
-// `_context.signals[]`. `_meta.community` / `_meta.cross_community_edges`
-// still flow through. Skipping the legacy `_context.community` assertions.
-describe.skip("Community context enrichment (A.3) (deprecated — _context.community replaced by _context.signals[])", () => {
-    it("injects community metadata into _meta for enrichable tools", async () => {
-        const graph = createMockGraphStore({
-            communityForEntity: {
-                id: 2,
-                label: "payment",
-                size: 15,
-                cohesion: 0.73,
-            },
-            crossCommunityEdges: [
-                {
-                    entity_name: "validateToken",
-                    entity_key: "auth_validate",
-                    entity_community_id: 0,
-                    entity_community_label: "auth",
-                    relation: "calls",
-                },
-            ],
-            entityOverride: {
-                key: "pay_charge",
-                kind: "function",
-                name: "charge",
-                file_path: "src/payment/charge.ts",
-                start_line: 1,
-                signature: "charge()",
-                body: "",
-                fan_in: 5,
-                fan_out: 3,
-                risk_level: "normal",
-                community: 2,
-            },
-        });
-        const router = new QueryRouter(graph);
-        const result = await router.execute("get_function", {
-            key: "pay_charge",
-        });
-        expect(result._meta.community).toEqual({
-            id: 2,
-            label: "payment",
-            size: 15,
-            cohesion: 0.73,
-        });
-        expect(result._meta.cross_community_edges).toHaveLength(1);
-        expect(result._meta.cross_community_edges?.[0]?.entity_name).toBe("validateToken");
-        expect(result._meta.cross_community_count).toBe(1);
-        expect(result._context?.community).toContain("payment");
-        expect(result._context?.community).toContain("auth");
-    });
-    it("omits community fields when entity has no community assignment", async () => {
-        const graph = createMockGraphStore({
-            communityForEntity: null,
-            entityOverride: {
-                key: "orphan",
-                kind: "function",
-                name: "orphan",
-                file_path: "src/orphan.ts",
-                start_line: 1,
-                signature: "orphan()",
-                body: "",
-                fan_in: 0,
-                fan_out: 0,
-                risk_level: "normal",
-                community: -1,
-            },
-        });
-        const router = new QueryRouter(graph);
-        const result = await router.execute("get_function", { key: "orphan" });
-        expect(result._meta.community).toBeUndefined();
-        expect(result._meta.cross_community_edges).toBeUndefined();
-        expect(result._meta.cross_community_count).toBeUndefined();
-    });
-    it("includes cross_community_count in _meta", async () => {
-        const crossEdges = Array.from({ length: 5 }, (_, i) => ({
-            entity_name: `entity_${i}`,
-            entity_key: `key_${i}`,
-            entity_community_id: i,
-            entity_community_label: `community_${i}`,
-            relation: "calls",
-        }));
-        const graph = createMockGraphStore({
-            communityForEntity: { id: 0, label: "test", size: 10, cohesion: 0.5 },
-            crossCommunityEdges: crossEdges,
-            entityOverride: {
-                key: "test_entity",
-                kind: "function",
-                name: "test",
-                file_path: "src/test.ts",
-                start_line: 1,
-                signature: "",
-                body: "",
-                fan_in: 0,
-                fan_out: 0,
-                risk_level: "normal",
-                community: 0,
-            },
-        });
-        const router = new QueryRouter(graph);
-        const result = await router.execute("get_function", {
-            key: "test_entity",
-        });
-        expect(result._meta.cross_community_count).toBe(5);
-    });
-});