@kya-os/mcp-i-core 1.2.3-canary.7 → 1.3.1

package/src/delegation/__tests__/delegation-graph.test.ts

@@ -0,0 +1,584 @@
+import { describe, it, expect, beforeEach, vi } from "vitest";
+import {
+  DelegationGraphManager,
+  createDelegationGraph,
+  type DelegationNode,
+  type DelegationGraphStorageProvider,
+} from "../delegation-graph.js";
+
+describe("DelegationGraphManager", () => {
+  let storage: DelegationGraphStorageProvider;
+  let graph: DelegationGraphManager;
+
+  // Simple in-memory storage for testing
+  const createMockStorage = (): DelegationGraphStorageProvider => {
+    const nodes = new Map<string, DelegationNode>();
+
+    return {
+      getNode: vi.fn(async (id: string) => {
+        return nodes.get(id) || null;
+      }),
+      setNode: vi.fn(async (node: DelegationNode) => {
+        nodes.set(node.id, { ...node });
+      }),
+      getChildren: vi.fn(async (id: string) => {
+        const node = nodes.get(id);
+        if (!node) return [];
+        return node.children
+          .map((childId) => nodes.get(childId))
+          .filter((n): n is DelegationNode => n !== undefined);
+      }),
+      getChain: vi.fn(async (id: string) => {
+        const chain: DelegationNode[] = [];
+        let currentId: string | null = id;
+
+        while (currentId) {
+          const node = nodes.get(currentId);
+          if (!node) break;
+          chain.unshift(node);
+          currentId = node.parentId;
+        }
+
+        return chain;
+      }),
+      getDescendants: vi.fn(async (id: string) => {
+        const descendants: DelegationNode[] = [];
+        const queue: string[] = [id];
+
+        while (queue.length > 0) {
+          const currentId = queue.shift()!;
+          const node = nodes.get(currentId);
+          if (!node) continue;
+
+          for (const childId of node.children) {
+            const child = nodes.get(childId);
+            if (child) {
+              descendants.push(child);
+              queue.push(childId);
+            }
+          }
+        }
+
+        return descendants;
+      }),
+      deleteNode: vi.fn(async (id: string) => {
+        nodes.delete(id);
+      }),
+    };
+  };
+
+  beforeEach(() => {
+    storage = createMockStorage();
+    graph = new DelegationGraphManager(storage);
+  });
+
+  describe("constructor", () => {
+    it("should create graph manager with storage", () => {
+      const manager = new DelegationGraphManager(storage);
+      expect(manager).toBeInstanceOf(DelegationGraphManager);
+    });
+  });
+
+  describe("createDelegationGraph", () => {
+    it("should create a graph manager instance", () => {
+      const manager = createDelegationGraph(storage);
+      expect(manager).toBeInstanceOf(DelegationGraphManager);
+    });
+  });
+
+  describe("registerDelegation", () => {
+    it("should register a root delegation (no parent)", async () => {
+      const node = await graph.registerDelegation({
+        id: "del-001",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      expect(node.id).toBe("del-001");
+      expect(node.parentId).toBeNull();
+      expect(node.children).toEqual([]);
+      expect(node.issuerDid).toBe("did:web:example.com:issuer");
+      expect(node.subjectDid).toBe("did:web:example.com:subject");
+      expect(storage.setNode).toHaveBeenCalled();
+    });
+
+    it("should register a child delegation and link to parent", async () => {
+      // Register parent first
+      await graph.registerDelegation({
+        id: "del-parent",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      // Clear call count
+      vi.clearAllMocks();
+
+      // Register child
+      const childNode = await graph.registerDelegation({
+        id: "del-child",
+        parentId: "del-parent",
+        issuerDid: "did:web:example.com:subject",
+        subjectDid: "did:web:example.com:subject2",
+      });
+
+      expect(childNode.parentId).toBe("del-parent");
+      // setNode is called: 1) for child, 2) for parent (to add child to children list)
+      expect(storage.setNode).toHaveBeenCalledTimes(2);
+    });
+
+    it("should include credentialStatusId if provided", async () => {
+      const node = await graph.registerDelegation({
+        id: "del-001",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+        credentialStatusId: "https://example.com/status#123",
+      });
+
+      expect(node.credentialStatusId).toBe("https://example.com/status#123");
+    });
+  });
+
+  describe("getNode", () => {
+    it("should return node if exists", async () => {
+      await graph.registerDelegation({
+        id: "del-001",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      const node = await graph.getNode("del-001");
+      expect(node).not.toBeNull();
+      expect(node?.id).toBe("del-001");
+    });
+
+    it("should return null if node does not exist", async () => {
+      const node = await graph.getNode("non-existent");
+      expect(node).toBeNull();
+    });
+  });
+
+  describe("getChildren", () => {
+    it("should return empty array for node with no children", async () => {
+      await graph.registerDelegation({
+        id: "del-001",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      const children = await graph.getChildren("del-001");
+      expect(children).toEqual([]);
+    });
+
+    it("should return all direct children", async () => {
+      // Register parent
+      await graph.registerDelegation({
+        id: "del-parent",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      // Register children
+      await graph.registerDelegation({
+        id: "del-child1",
+        parentId: "del-parent",
+        issuerDid: "did:web:example.com:subject",
+        subjectDid: "did:web:example.com:subject2",
+      });
+
+      await graph.registerDelegation({
+        id: "del-child2",
+        parentId: "del-parent",
+        issuerDid: "did:web:example.com:subject",
+        subjectDid: "did:web:example.com:subject3",
+      });
+
+      const children = await graph.getChildren("del-parent");
+      expect(children.length).toBe(2);
+      expect(children.map((c) => c.id)).toContain("del-child1");
+      expect(children.map((c) => c.id)).toContain("del-child2");
+    });
+  });
+
+  describe("getDescendants", () => {
+    it("should return all descendants recursively", async () => {
+      // Create a tree: parent -> child1 -> grandchild
+      await graph.registerDelegation({
+        id: "del-parent",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      await graph.registerDelegation({
+        id: "del-child1",
+        parentId: "del-parent",
+        issuerDid: "did:web:example.com:subject",
+        subjectDid: "did:web:example.com:subject2",
+      });
+
+      await graph.registerDelegation({
+        id: "del-grandchild",
+        parentId: "del-child1",
+        issuerDid: "did:web:example.com:subject2",
+        subjectDid: "did:web:example.com:subject3",
+      });
+
+      const descendants = await graph.getDescendants("del-parent");
+      expect(descendants.length).toBe(2);
+      expect(descendants.map((d) => d.id)).toContain("del-child1");
+      expect(descendants.map((d) => d.id)).toContain("del-grandchild");
+    });
+
+    it("should return empty array for node with no descendants", async () => {
+      await graph.registerDelegation({
+        id: "del-001",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      const descendants = await graph.getDescendants("del-001");
+      expect(descendants).toEqual([]);
+    });
+  });
+
+  describe("getChain", () => {
+    it("should return chain from root to node", async () => {
+      // Create chain: root -> child -> grandchild
+      await graph.registerDelegation({
+        id: "del-root",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      await graph.registerDelegation({
+        id: "del-child",
+        parentId: "del-root",
+        issuerDid: "did:web:example.com:subject",
+        subjectDid: "did:web:example.com:subject2",
+      });
+
+      await graph.registerDelegation({
+        id: "del-grandchild",
+        parentId: "del-child",
+        issuerDid: "did:web:example.com:subject2",
+        subjectDid: "did:web:example.com:subject3",
+      });
+
+      const chain = await graph.getChain("del-grandchild");
+      expect(chain.length).toBe(3);
+      expect(chain[0].id).toBe("del-root");
+      expect(chain[1].id).toBe("del-child");
+      expect(chain[2].id).toBe("del-grandchild");
+    });
+
+    it("should return single node for root", async () => {
+      await graph.registerDelegation({
+        id: "del-root",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      const chain = await graph.getChain("del-root");
+      expect(chain.length).toBe(1);
+      expect(chain[0].id).toBe("del-root");
+    });
+  });
+
+  describe("isAncestor", () => {
+    it("should return true if ancestor is parent", async () => {
+      await graph.registerDelegation({
+        id: "del-parent",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      await graph.registerDelegation({
+        id: "del-child",
+        parentId: "del-parent",
+        issuerDid: "did:web:example.com:subject",
+        subjectDid: "did:web:example.com:subject2",
+      });
+
+      const isAncestor = await graph.isAncestor("del-parent", "del-child");
+      expect(isAncestor).toBe(true);
+    });
+
+    it("should return true if ancestor is grandparent", async () => {
+      await graph.registerDelegation({
+        id: "del-grandparent",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      await graph.registerDelegation({
+        id: "del-parent",
+        parentId: "del-grandparent",
+        issuerDid: "did:web:example.com:subject",
+        subjectDid: "did:web:example.com:subject2",
+      });
+
+      await graph.registerDelegation({
+        id: "del-child",
+        parentId: "del-parent",
+        issuerDid: "did:web:example.com:subject2",
+        subjectDid: "did:web:example.com:subject3",
+      });
+
+      const isAncestor = await graph.isAncestor("del-grandparent", "del-child");
+      expect(isAncestor).toBe(true);
+    });
+
+    it("should return false if not ancestor", async () => {
+      await graph.registerDelegation({
+        id: "del-1",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      await graph.registerDelegation({
+        id: "del-2",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer2",
+        subjectDid: "did:web:example.com:subject2",
+      });
+
+      const isAncestor = await graph.isAncestor("del-1", "del-2");
+      expect(isAncestor).toBe(false);
+    });
+
+    it("should return false if nodes are reversed", async () => {
+      await graph.registerDelegation({
+        id: "del-parent",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      await graph.registerDelegation({
+        id: "del-child",
+        parentId: "del-parent",
+        issuerDid: "did:web:example.com:subject",
+        subjectDid: "did:web:example.com:subject2",
+      });
+
+      const isAncestor = await graph.isAncestor("del-child", "del-parent");
+      expect(isAncestor).toBe(false);
+    });
+  });
+
+  describe("getDepth", () => {
+    it("should return 0 for root node", async () => {
+      await graph.registerDelegation({
+        id: "del-root",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      const depth = await graph.getDepth("del-root");
+      expect(depth).toBe(0);
+    });
+
+    it("should return 1 for immediate child", async () => {
+      await graph.registerDelegation({
+        id: "del-root",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      await graph.registerDelegation({
+        id: "del-child",
+        parentId: "del-root",
+        issuerDid: "did:web:example.com:subject",
+        subjectDid: "did:web:example.com:subject2",
+      });
+
+      const depth = await graph.getDepth("del-child");
+      expect(depth).toBe(1);
+    });
+
+    it("should return correct depth for nested nodes", async () => {
+      await graph.registerDelegation({
+        id: "del-root",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      await graph.registerDelegation({
+        id: "del-child",
+        parentId: "del-root",
+        issuerDid: "did:web:example.com:subject",
+        subjectDid: "did:web:example.com:subject2",
+      });
+
+      await graph.registerDelegation({
+        id: "del-grandchild",
+        parentId: "del-child",
+        issuerDid: "did:web:example.com:subject2",
+        subjectDid: "did:web:example.com:subject3",
+      });
+
+      const depth = await graph.getDepth("del-grandchild");
+      expect(depth).toBe(2);
+    });
+  });
+
+  describe("validateChain", () => {
+    it("should validate correct chain", async () => {
+      await graph.registerDelegation({
+        id: "del-root",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      await graph.registerDelegation({
+        id: "del-child",
+        parentId: "del-root",
+        issuerDid: "did:web:example.com:subject", // Child's issuer = parent's subject
+        subjectDid: "did:web:example.com:subject2",
+      });
+
+      const result = await graph.validateChain("del-child");
+      expect(result.valid).toBe(true);
+    });
+
+    it("should invalidate chain with mismatched issuer/subject", async () => {
+      await graph.registerDelegation({
+        id: "del-root",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      await graph.registerDelegation({
+        id: "del-child",
+        parentId: "del-root",
+        issuerDid: "did:web:example.com:wrong-issuer", // Wrong issuer
+        subjectDid: "did:web:example.com:subject2",
+      });
+
+      const result = await graph.validateChain("del-child");
+      expect(result.valid).toBe(false);
+      expect(result.reason).toContain("Invalid chain");
+    });
+
+    // Note: parentId mismatch validation is complex - covered by issuer/subject mismatch test above
+
+    it("should return invalid for non-existent delegation", async () => {
+      const result = await graph.validateChain("non-existent");
+      expect(result.valid).toBe(false);
+      expect(result.reason).toBe("Delegation not found");
+    });
+  });
+
+  describe("removeDelegation", () => {
+    it("should remove delegation from graph", async () => {
+      await graph.registerDelegation({
+        id: "del-001",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      await graph.removeDelegation("del-001");
+
+      const node = await graph.getNode("del-001");
+      expect(node).toBeNull();
+      expect(storage.deleteNode).toHaveBeenCalledWith("del-001");
+    });
+
+    it("should remove child from parent's children list", async () => {
+      await graph.registerDelegation({
+        id: "del-parent",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      await graph.registerDelegation({
+        id: "del-child",
+        parentId: "del-parent",
+        issuerDid: "did:web:example.com:subject",
+        subjectDid: "did:web:example.com:subject2",
+      });
+
+      await graph.removeDelegation("del-child");
+
+      const parent = await graph.getNode("del-parent");
+      expect(parent?.children).not.toContain("del-child");
+    });
+
+    it("should handle removal of non-existent delegation gracefully", async () => {
+      await expect(graph.removeDelegation("non-existent")).resolves.not.toThrow();
+    });
+  });
+
+  describe("edge cases", () => {
+    it("should handle multiple children correctly", async () => {
+      await graph.registerDelegation({
+        id: "del-parent",
+        parentId: null,
+        issuerDid: "did:web:example.com:issuer",
+        subjectDid: "did:web:example.com:subject",
+      });
+
+      await graph.registerDelegation({
+        id: "del-child1",
+        parentId: "del-parent",
+        issuerDid: "did:web:example.com:subject",
+        subjectDid: "did:web:example.com:subject2",
+      });
+
+      await graph.registerDelegation({
+        id: "del-child2",
+        parentId: "del-parent",
+        issuerDid: "did:web:example.com:subject",
+        subjectDid: "did:web:example.com:subject3",
+      });
+
+      await graph.registerDelegation({
+        id: "del-child3",
+        parentId: "del-parent",
+        issuerDid: "did:web:example.com:subject",
+        subjectDid: "did:web:example.com:subject4",
+      });
+
+      const children = await graph.getChildren("del-parent");
+      expect(children.length).toBe(3);
+    });
+
+    it("should handle deep chains correctly", async () => {
+      // Create a chain of depth 5
+      const chain: string[] = [];
+      for (let i = 0; i < 5; i++) {
+        const id = `del-${i}`;
+        chain.push(id);
+        await graph.registerDelegation({
+          id,
+          parentId: i === 0 ? null : chain[i - 1],
+          issuerDid: `did:web:example.com:issuer${i}`,
+          subjectDid: `did:web:example.com:subject${i + 1}`,
+        });
+      }
+
+      const depth = await graph.getDepth(chain[4]);
+      expect(depth).toBe(4);
+    });
+  });
+});
+