Code-MCP

import { describe, it, expect } from "vitest"; import { generateDiagramHandler, visualizeSystemHandler, } from "./visualization.js"; describe("Visualization Tools", () => { describe("generateDiagramHandler", () => { it("should generate flowchart", () => { const result = generateDiagramHandler({ type: "flowchart", description: "flow", }); expect(result.content[0].text).toContain("graph TB"); }); }); describe("visualizeSystemHandler", () => { it("should generate system graph", () => { const result = visualizeSystemHandler({ components: ["Web", "API"], relationships: ["Web -> API"], }); expect(result.content[0].text).toContain("Web[Web]"); expect(result.content[0].text).toContain("Web --> API"); }); }); });