import { describe, it, expect } from "vitest";
import { toolRegistry } from "../registry.js";
describe("ToolRegistry", () => {
it("has tools registered", () => {
expect(toolRegistry.size).toBeGreaterThan(0);
});
it("looks up a tool by name", () => {
const tool = toolRegistry.get("peframe");
expect(tool).toBeDefined();
expect(tool!.name).toBe("peframe");
expect(tool!.description).toBeTruthy();
expect(tool!.command).toBeTruthy();
});
it("returns undefined for unknown tool", () => {
expect(toolRegistry.get("nonexistent-tool")).toBeUndefined();
});
it("has() returns correct results", () => {
expect(toolRegistry.has("peframe")).toBe(true);
expect(toolRegistry.has("nonexistent")).toBe(false);
});
it("filters by tag", () => {
const peTools = toolRegistry.byTag("pe");
expect(peTools.length).toBeGreaterThan(0);
for (const t of peTools) {
expect(t.tags).toContain("pe");
}
});
it("filters by tier (quick includes only quick tools)", () => {
const quickTools = toolRegistry.byTier("quick");
for (const t of quickTools) {
expect(t.tier).toBe("quick");
}
});
it("filters by tier (deep includes all tiers)", () => {
const deepTools = toolRegistry.byTier("deep");
const allTools = toolRegistry.all();
expect(deepTools.length).toBe(allTools.length);
});
it("filters by tag and tier", () => {
const tools = toolRegistry.byTagAndTier("pe", "standard");
for (const t of tools) {
expect(t.tags).toContain("pe");
expect(["quick", "standard"]).toContain(t.tier);
}
});
it("all tools have required fields", () => {
for (const tool of toolRegistry.all()) {
expect(tool.name).toBeTruthy();
expect(tool.description).toBeTruthy();
expect(tool.command).toBeTruthy();
expect(["positional", "flag", "stdin"]).toContain(tool.inputStyle);
expect(["text", "json"]).toContain(tool.outputFormat);
expect(tool.timeout).toBeGreaterThan(0);
expect(["quick", "standard", "deep"]).toContain(tool.tier);
}
});
it("has memory forensics tools tagged with 'memory'", () => {
const memTools = toolRegistry.byTag("memory");
expect(memTools.length).toBeGreaterThanOrEqual(6);
const names = memTools.map((t) => t.name);
expect(names).toContain("vol3-info");
expect(names).toContain("vol3-pslist");
expect(names).toContain("vol3-pstree");
expect(names).toContain("vol3-netscan");
expect(names).toContain("vol3-cmdline");
expect(names).toContain("vol3-malfind");
});
it("vol3 tools use flag inputStyle with -f flag and suffixArgs", () => {
const vol3info = toolRegistry.get("vol3-info");
expect(vol3info).toBeDefined();
expect(vol3info!.inputStyle).toBe("flag");
expect(vol3info!.inputFlag).toBe("-f");
expect(vol3info!.suffixArgs).toEqual(["windows.info"]);
});
});
