@kya-os/mcp-i 0.1.0-alpha.3.9 → 1.2.0

package/dist/runtime/__tests__/well-known.test.js

@@ -0,0 +1,312 @@
+"use strict";
+/**
+ * Tests for Well-Known Endpoints
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const well_known_js_1 = require("../well-known.js");
+(0, vitest_1.describe)("WellKnownManager", () => {
+    let wellKnownManager;
+    let mockIdentity;
+    let mockConfig;
+    (0, vitest_1.beforeEach)(() => {
+        mockIdentity = {
+            did: "did:web:example.com:agents:test-agent",
+            keyId: "key-test-123",
+            privateKey: "test-private-key",
+            publicKey: Buffer.from("test-public-key-32-bytes-long!!!").toString("base64"),
+            createdAt: new Date().toISOString(),
+        };
+        mockConfig = {
+            environment: "development",
+            baseUrl: "https://example.com",
+            agentMetadata: {
+                name: "Test Agent",
+                description: "A test XMCP-I agent",
+                version: "1.0.0",
+            },
+            registryUrls: {
+                kta: "https://knowthat.ai/agents/test-agent",
+                mcp: "https://registry.mcp.dev/agents/test-agent",
+            },
+        };
+        wellKnownManager = new well_known_js_1.WellKnownManager(mockIdentity, mockConfig);
+    });
+    (0, vitest_1.describe)("DID Document Generation", () => {
+        (0, vitest_1.it)("should generate valid DID document", () => {
+            const didDocument = wellKnownManager.generateDIDDocument();
+            (0, vitest_1.expect)(didDocument["@context"]).toContain("https://www.w3.org/ns/did/v1");
+            (0, vitest_1.expect)(didDocument.id).toBe(mockIdentity.did);
+            (0, vitest_1.expect)(didDocument.verificationMethod).toHaveLength(1);
+            (0, vitest_1.expect)(didDocument.verificationMethod[0].id).toBe("#key-test-123");
+            (0, vitest_1.expect)(didDocument.verificationMethod[0].type).toBe("Ed25519VerificationKey2020");
+            (0, vitest_1.expect)(didDocument.verificationMethod[0].controller).toBe(mockIdentity.did);
+            (0, vitest_1.expect)(didDocument.verificationMethod[0].publicKeyMultibase).toMatch(/^z/);
+            (0, vitest_1.expect)(didDocument.authentication).toEqual(["#key-test-123"]);
+            (0, vitest_1.expect)(didDocument.assertionMethod).toEqual(["#key-test-123"]);
+        });
+        (0, vitest_1.it)("should include proper verification method structure", () => {
+            const didDocument = wellKnownManager.generateDIDDocument();
+            const verificationMethod = didDocument.verificationMethod[0];
+            (0, vitest_1.expect)(verificationMethod.id).toBe(`#${mockIdentity.keyId}`);
+            (0, vitest_1.expect)(verificationMethod.type).toBe("Ed25519VerificationKey2020");
+            (0, vitest_1.expect)(verificationMethod.controller).toBe(mockIdentity.did);
+            (0, vitest_1.expect)(verificationMethod.publicKeyMultibase).toBeTruthy();
+        });
+        (0, vitest_1.it)("should encode public key as multibase", () => {
+            const didDocument = wellKnownManager.generateDIDDocument();
+            const publicKeyMultibase = didDocument.verificationMethod[0].publicKeyMultibase;
+            (0, vitest_1.expect)(publicKeyMultibase).toMatch(/^z[1-9A-HJ-NP-Za-km-z]+$/); // Base58btc format
+        });
+    });
+    (0, vitest_1.describe)("Agent Document Generation", () => {
+        (0, vitest_1.it)("should generate valid agent document", () => {
+            const agentDocument = wellKnownManager.generateAgentDocument();
+            (0, vitest_1.expect)(agentDocument.id).toBe(mockIdentity.did);
+            (0, vitest_1.expect)(agentDocument.capabilities["mcp-i"]).toEqual([
+                "handshake",
+                "signing",
+                "verification",
+            ]);
+            (0, vitest_1.expect)(agentDocument.registry?.kta).toBe("https://knowthat.ai/agents/test-agent");
+            (0, vitest_1.expect)(agentDocument.registry?.mcp).toBe("https://registry.mcp.dev/agents/test-agent");
+            (0, vitest_1.expect)(agentDocument.metadata?.name).toBe("Test Agent");
+            (0, vitest_1.expect)(agentDocument.metadata?.description).toBe("A test XMCP-I agent");
+            (0, vitest_1.expect)(agentDocument.metadata?.version).toBe("1.0.0");
+        });
+        (0, vitest_1.it)("should include exact MCP-I capability triplet", () => {
+            const agentDocument = wellKnownManager.generateAgentDocument();
+            const capabilities = agentDocument.capabilities["mcp-i"];
+            (0, vitest_1.expect)(capabilities).toHaveLength(3);
+            (0, vitest_1.expect)(capabilities).toContain("handshake");
+            (0, vitest_1.expect)(capabilities).toContain("signing");
+            (0, vitest_1.expect)(capabilities).toContain("verification");
+            (0, vitest_1.expect)(capabilities).toEqual(["handshake", "signing", "verification"]);
+        });
+        (0, vitest_1.it)("should work without optional metadata", () => {
+            const minimalConfig = {
+                environment: "development",
+            };
+            const minimalManager = new well_known_js_1.WellKnownManager(mockIdentity, minimalConfig);
+            const agentDocument = minimalManager.generateAgentDocument();
+            (0, vitest_1.expect)(agentDocument.id).toBe(mockIdentity.did);
+            (0, vitest_1.expect)(agentDocument.capabilities["mcp-i"]).toEqual([
+                "handshake",
+                "signing",
+                "verification",
+            ]);
+            (0, vitest_1.expect)(agentDocument.registry).toBeUndefined();
+            (0, vitest_1.expect)(agentDocument.metadata).toBeUndefined();
+        });
+    });
+    (0, vitest_1.describe)("HTTP Headers", () => {
+        (0, vitest_1.it)("should return correct headers for DID document in development", () => {
+            const headers = wellKnownManager.getDIDDocumentHeaders();
+            (0, vitest_1.expect)(headers["Content-Type"]).toBe("application/did+json");
+            (0, vitest_1.expect)(headers["Cache-Control"]).toBe("no-store");
+        });
+        (0, vitest_1.it)("should return correct headers for DID document in production", () => {
+            const prodConfig = { ...mockConfig, environment: "production" };
+            const prodManager = new well_known_js_1.WellKnownManager(mockIdentity, prodConfig);
+            const headers = prodManager.getDIDDocumentHeaders();
+            (0, vitest_1.expect)(headers["Content-Type"]).toBe("application/did+json");
+            (0, vitest_1.expect)(headers["Cache-Control"]).toBe("public, max-age=300");
+        });
+        (0, vitest_1.it)("should return correct headers for agent document in development", () => {
+            const headers = wellKnownManager.getAgentDocumentHeaders();
+            (0, vitest_1.expect)(headers["Content-Type"]).toBe("application/json");
+            (0, vitest_1.expect)(headers["Cache-Control"]).toBe("no-store");
+        });
+        (0, vitest_1.it)("should return correct headers for agent document in production", () => {
+            const prodConfig = { ...mockConfig, environment: "production" };
+            const prodManager = new well_known_js_1.WellKnownManager(mockIdentity, prodConfig);
+            const headers = prodManager.getAgentDocumentHeaders();
+            (0, vitest_1.expect)(headers["Content-Type"]).toBe("application/json");
+            (0, vitest_1.expect)(headers["Cache-Control"]).toBe("public, max-age=300");
+        });
+    });
+    (0, vitest_1.describe)("Configuration Management", () => {
+        (0, vitest_1.it)("should update configuration", () => {
+            const newConfig = {
+                environment: "production",
+                agentMetadata: {
+                    name: "Updated Agent",
+                },
+            };
+            wellKnownManager.updateConfig(newConfig);
+            const config = wellKnownManager.getConfig();
+            (0, vitest_1.expect)(config.environment).toBe("production");
+            (0, vitest_1.expect)(config.agentMetadata?.name).toBe("Updated Agent");
+            (0, vitest_1.expect)(config.baseUrl).toBe("https://example.com"); // Should preserve existing values
+        });
+        (0, vitest_1.it)("should return current configuration", () => {
+            const config = wellKnownManager.getConfig();
+            (0, vitest_1.expect)(config.environment).toBe("development");
+            (0, vitest_1.expect)(config.baseUrl).toBe("https://example.com");
+            (0, vitest_1.expect)(config.agentMetadata?.name).toBe("Test Agent");
+        });
+    });
+});
+(0, vitest_1.describe)("Well-Known Handler", () => {
+    let handler;
+    let mockIdentity;
+    let mockConfig;
+    (0, vitest_1.beforeEach)(() => {
+        mockIdentity = {
+            did: "did:web:example.com:agents:test-agent",
+            keyId: "key-test-123",
+            privateKey: "test-private-key",
+            publicKey: Buffer.from("test-public-key-32-bytes-long!!!").toString("base64"),
+            createdAt: new Date().toISOString(),
+        };
+        mockConfig = {
+            environment: "development",
+            baseUrl: "https://example.com",
+        };
+        handler = (0, well_known_js_1.createWellKnownHandler)(mockIdentity, mockConfig);
+    });
+    (0, vitest_1.describe)("DID Document Handler", () => {
+        (0, vitest_1.it)("should handle DID document request successfully", () => {
+            const response = handler.handleDIDDocument();
+            (0, vitest_1.expect)(response.status).toBe(200);
+            (0, vitest_1.expect)(response.headers["Content-Type"]).toBe("application/did+json");
+            (0, vitest_1.expect)(response.body).toBeTruthy();
+            const didDocument = JSON.parse(response.body);
+            (0, vitest_1.expect)(didDocument.id).toBe(mockIdentity.did);
+            (0, vitest_1.expect)(didDocument.verificationMethod).toHaveLength(1);
+        });
+        (0, vitest_1.it)("should return valid JSON in response body", () => {
+            const response = handler.handleDIDDocument();
+            (0, vitest_1.expect)(() => JSON.parse(response.body)).not.toThrow();
+            const didDocument = JSON.parse(response.body);
+            (0, vitest_1.expect)((0, well_known_js_1.validateDIDDocument)(didDocument)).toBe(true);
+        });
+    });
+    (0, vitest_1.describe)("Agent Document Handler", () => {
+        (0, vitest_1.it)("should handle agent document request successfully", () => {
+            const response = handler.handleAgentDocument();
+            (0, vitest_1.expect)(response.status).toBe(200);
+            (0, vitest_1.expect)(response.headers["Content-Type"]).toBe("application/json");
+            (0, vitest_1.expect)(response.body).toBeTruthy();
+            const agentDocument = JSON.parse(response.body);
+            (0, vitest_1.expect)(agentDocument.id).toBe(mockIdentity.did);
+            (0, vitest_1.expect)(agentDocument.capabilities["mcp-i"]).toEqual([
+                "handshake",
+                "signing",
+                "verification",
+            ]);
+        });
+        (0, vitest_1.it)("should return valid JSON in response body", () => {
+            const response = handler.handleAgentDocument();
+            (0, vitest_1.expect)(() => JSON.parse(response.body)).not.toThrow();
+            const agentDocument = JSON.parse(response.body);
+            (0, vitest_1.expect)((0, well_known_js_1.validateAgentDocument)(agentDocument)).toBe(true);
+        });
+    });
+});
+(0, vitest_1.describe)("Validation Functions", () => {
+    (0, vitest_1.describe)("validateDIDDocument", () => {
+        (0, vitest_1.it)("should validate correct DID document", () => {
+            const validDIDDocument = {
+                "@context": ["https://www.w3.org/ns/did/v1"],
+                id: "did:web:example.com:agents:test-agent",
+                verificationMethod: [
+                    {
+                        id: "#key-test-123",
+                        type: "Ed25519VerificationKey2020",
+                        controller: "did:web:example.com:agents:test-agent",
+                        publicKeyMultibase: "z6MkhaXgBZDvotDkL5257faiztiGiC2QtKLGpbnnEGta2doK",
+                    },
+                ],
+                authentication: ["#key-test-123"],
+                assertionMethod: ["#key-test-123"],
+            };
+            (0, vitest_1.expect)((0, well_known_js_1.validateDIDDocument)(validDIDDocument)).toBe(true);
+        });
+        (0, vitest_1.it)("should reject invalid DID documents", () => {
+            const invalidDocuments = [
+                null,
+                undefined,
+                {},
+                { "@context": "invalid" },
+                { "@context": [], id: "not-a-did" },
+                {
+                    "@context": [],
+                    id: "did:web:example.com",
+                    verificationMethod: "invalid",
+                },
+            ];
+            for (const doc of invalidDocuments) {
+                (0, vitest_1.expect)((0, well_known_js_1.validateDIDDocument)(doc)).toBe(false);
+            }
+        });
+    });
+    (0, vitest_1.describe)("validateAgentDocument", () => {
+        (0, vitest_1.it)("should validate correct agent document", () => {
+            const validAgentDocument = {
+                id: "did:web:example.com:agents:test-agent",
+                capabilities: {
+                    "mcp-i": ["handshake", "signing", "verification"],
+                },
+            };
+            (0, vitest_1.expect)((0, well_known_js_1.validateAgentDocument)(validAgentDocument)).toBe(true);
+        });
+        (0, vitest_1.it)("should reject agent documents with incorrect capabilities", () => {
+            const invalidDocuments = [
+                {
+                    id: "did:web:example.com:agents:test-agent",
+                    capabilities: {
+                        "mcp-i": ["handshake", "signing"], // Missing verification
+                    },
+                },
+                {
+                    id: "did:web:example.com:agents:test-agent",
+                    capabilities: {
+                        "mcp-i": ["handshake", "signing", "verification", "extra"], // Extra capability
+                    },
+                },
+                {
+                    id: "did:web:example.com:agents:test-agent",
+                    capabilities: {
+                        "mcp-i": ["wrong", "capabilities", "list"],
+                    },
+                },
+            ];
+            for (const doc of invalidDocuments) {
+                (0, vitest_1.expect)((0, well_known_js_1.validateAgentDocument)(doc)).toBe(false);
+            }
+        });
+        (0, vitest_1.it)("should reject invalid agent documents", () => {
+            const invalidDocuments = [
+                null,
+                undefined,
+                {},
+                { id: "not-a-did" },
+                { id: "did:web:example.com", capabilities: "invalid" },
+            ];
+            for (const doc of invalidDocuments) {
+                (0, vitest_1.expect)((0, well_known_js_1.validateAgentDocument)(doc)).toBe(false);
+            }
+        });
+    });
+});
+(0, vitest_1.describe)("Utility Functions", () => {
+    (0, vitest_1.describe)("extractDIDFromPath", () => {
+        (0, vitest_1.it)("should extract DID from root path", () => {
+            const did = (0, well_known_js_1.extractDIDFromPath)("/.well-known/did.json", "https://example.com");
+            (0, vitest_1.expect)(did).toBe("did:web:example.com");
+        });
+        (0, vitest_1.it)("should extract DID from nested path", () => {
+            const did = (0, well_known_js_1.extractDIDFromPath)("/agents/my-agent/.well-known/did.json", "https://example.com");
+            (0, vitest_1.expect)(did).toBe("did:web:example.com:agents:my-agent");
+        });
+        (0, vitest_1.it)("should handle complex paths", () => {
+            const did = (0, well_known_js_1.extractDIDFromPath)("/users/alice/agents/assistant/.well-known/did.json", "https://example.com");
+            (0, vitest_1.expect)(did).toBe("did:web:example.com:users:alice:agents:assistant");
+        });
+        (0, vitest_1.it)("should return null for invalid URLs", () => {
+            const did = (0, well_known_js_1.extractDIDFromPath)("/.well-known/did.json", "invalid-url");
+            (0, vitest_1.expect)(did).toBeNull();
+        });
+    });
+});