/**
* Unit tests for Tool Icons Utility
*
* Tests icon generation for tool categories and behavior annotations.
*/
import { describe, it, expect } from "vitest";
import {
getToolIcons,
getCategoryIcon,
getAllCategoryIcons,
} from "../icons.js";
import type { ToolGroup } from "../../types/index.js";
describe("Tool Icons Utility", () => {
describe("getToolIcons", () => {
it("should return warning icon for destructive tools", () => {
const icons = getToolIcons("core", { destructiveHint: true });
expect(icons.length).toBe(1);
expect(icons[0]?.mimeType).toBe("image/svg+xml");
// Warning icons have red color
expect(icons[0]?.src).toContain("data:image/svg+xml;base64,");
});
it("should return admin icon for VACUUM operations", () => {
const icons = getToolIcons("admin", { title: "VACUUM Tables" });
expect(icons.length).toBe(1);
expect(icons[0]?.mimeType).toBe("image/svg+xml");
});
it("should return admin icon for ANALYZE operations", () => {
const icons = getToolIcons("admin", {
title: "Analyze Table Statistics",
});
expect(icons.length).toBe(1);
});
it("should return admin icon for REINDEX operations", () => {
const icons = getToolIcons("admin", { title: "Reindex Database" });
expect(icons.length).toBe(1);
});
it("should return category icon for non-destructive tools", () => {
const icons = getToolIcons("core", { readOnlyHint: true });
expect(icons.length).toBe(1);
expect(icons[0]?.mimeType).toBe("image/svg+xml");
});
it("should return category icon when no annotations provided", () => {
const icons = getToolIcons("vector");
expect(icons.length).toBe(1);
});
it("should handle undefined annotations title", () => {
const icons = getToolIcons("postgis", { readOnlyHint: true });
expect(icons.length).toBe(1);
});
});
describe("getCategoryIcon", () => {
const toolGroups: ToolGroup[] = [
"core",
"transactions",
"jsonb",
"text",
"performance",
"admin",
"monitoring",
"backup",
"schema",
"vector",
"postgis",
"partitioning",
"stats",
"cron",
"partman",
"kcache",
"citext",
"ltree",
"pgcrypto",
"codemode",
];
it.each(toolGroups)("should return icon for %s group", (group) => {
const icons = getCategoryIcon(group);
expect(icons.length).toBe(1);
expect(icons[0]?.src).toBeDefined();
expect(icons[0]?.mimeType).toBe("image/svg+xml");
expect(icons[0]?.sizes).toEqual(["any"]);
});
});
describe("getAllCategoryIcons", () => {
it("should return icons for all tool groups", () => {
const allIcons = getAllCategoryIcons();
expect(Object.keys(allIcons).length).toBeGreaterThan(0);
});
it("should return valid icons for each group", () => {
const allIcons = getAllCategoryIcons();
for (const [, icons] of Object.entries(allIcons)) {
expect(icons.length).toBe(1);
expect(icons[0]?.src).toContain("data:image/svg+xml;base64,");
expect(icons[0]?.mimeType).toBe("image/svg+xml");
expect(icons[0]?.sizes).toEqual(["any"]);
}
});
it("should include core group", () => {
const allIcons = getAllCategoryIcons();
expect(allIcons["core"]).toBeDefined();
});
it("should include vector group", () => {
const allIcons = getAllCategoryIcons();
expect(allIcons["vector"]).toBeDefined();
});
it("should include postgis group", () => {
const allIcons = getAllCategoryIcons();
expect(allIcons["postgis"]).toBeDefined();
});
it("should include codemode group", () => {
const allIcons = getAllCategoryIcons();
expect(allIcons["codemode"]).toBeDefined();
});
});
describe("Icon Data URI Format", () => {
it("should generate valid base64 encoded SVG", () => {
const icons = getCategoryIcon("core");
const src = icons[0]?.src ?? "";
// Should be a data URI
expect(src.startsWith("data:image/svg+xml;base64,")).toBe(true);
// Extract base64 part and decode
const base64Part = src.replace("data:image/svg+xml;base64,", "");
const decodedSvg = Buffer.from(base64Part, "base64").toString("utf-8");
// Should be valid SVG
expect(decodedSvg).toContain("<svg");
expect(decodedSvg).toContain("</svg>");
expect(decodedSvg).toContain("viewBox");
});
it("should include stroke properties in SVG", () => {
const icons = getCategoryIcon("transactions");
const src = icons[0]?.src ?? "";
const base64Part = src.replace("data:image/svg+xml;base64,", "");
const decodedSvg = Buffer.from(base64Part, "base64").toString("utf-8");
expect(decodedSvg).toContain("stroke=");
expect(decodedSvg).toContain("stroke-width");
});
});
});
