59 lines
1.9 KiB
TypeScript
59 lines
1.9 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
|
|
import { nodeTypes } from "@/components/canvas/node-types";
|
|
import { CANVAS_NODE_TEMPLATES } from "@/lib/canvas-node-templates";
|
|
import {
|
|
NODE_CATALOG,
|
|
NODE_CATEGORY_META,
|
|
catalogEntriesByCategory,
|
|
isNodePaletteEnabled,
|
|
} from "@/lib/canvas-node-catalog";
|
|
import { NODE_DEFAULTS, NODE_HANDLE_MAP } from "@/lib/canvas-utils";
|
|
|
|
describe("canvas agent config", () => {
|
|
it("registers the agent node type", () => {
|
|
expect(nodeTypes.agent).toBeTypeOf("function");
|
|
});
|
|
|
|
it("adds a campaign distributor palette template", () => {
|
|
expect(CANVAS_NODE_TEMPLATES.find((template) => template.type === "agent")?.label).toBe(
|
|
"Campaign Distributor",
|
|
);
|
|
});
|
|
|
|
it("enables the agent in the catalog", () => {
|
|
const entry = NODE_CATALOG.find((item) => item.type === "agent");
|
|
expect(entry).toBeDefined();
|
|
expect(entry && isNodePaletteEnabled(entry)).toBe(true);
|
|
});
|
|
|
|
it("moves agent nodes into an Agents category", () => {
|
|
expect(NODE_CATEGORY_META.agents.label).toBe("Agents");
|
|
|
|
const byCategory = catalogEntriesByCategory();
|
|
const agentsEntries = byCategory.get("agents") ?? [];
|
|
const aiOutputEntries = byCategory.get("ai-output") ?? [];
|
|
|
|
expect(agentsEntries.map((entry) => entry.type)).toEqual(["agent", "agent-output"]);
|
|
expect(agentsEntries[0]).toMatchObject({
|
|
label: "Campaign Orchestrator",
|
|
category: "agents",
|
|
});
|
|
|
|
expect(aiOutputEntries.map((entry) => entry.type)).toEqual([
|
|
"prompt",
|
|
"video-prompt",
|
|
"ai-text",
|
|
]);
|
|
expect(NODE_CATALOG.find((entry) => entry.type === "ai-video")?.category).toBe("source");
|
|
});
|
|
|
|
it("keeps the agent input-only in MVP", () => {
|
|
expect(NODE_HANDLE_MAP.agent?.target).toBe("agent-in");
|
|
expect(NODE_HANDLE_MAP.agent?.source).toBeUndefined();
|
|
expect(NODE_DEFAULTS.agent?.data).toMatchObject({
|
|
templateId: "campaign-distributor",
|
|
});
|
|
});
|
|
});
|