78 lines
2.3 KiB
TypeScript
78 lines
2.3 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import { isUiEnvelopeAllowed, parseUiEnvelope, parseUiRegistry } from "./validator";
|
|
import type { UIRegistry } from "./types";
|
|
|
|
const registry: UIRegistry = {
|
|
schema_version: "agent-ui-registry@1",
|
|
chart_grammars: ["echarts-safe-subset"],
|
|
components: [
|
|
{ id: "HistoryPanel", supportedSurfaces: ["side_panel", "canvas"] },
|
|
{ id: "ScadaPanel", supportedSurfaces: ["side_panel", "canvas"] }
|
|
],
|
|
actions: [{ id: "zoom_to_map", supportedSurfaces: ["map_overlay"] }]
|
|
};
|
|
|
|
describe("UIEnvelope validation", () => {
|
|
it("rejects registered components outside the frontend allowlist", () => {
|
|
const envelope = parseUiEnvelope({
|
|
kind: "registered_component",
|
|
schemaVersion: "agent-ui@1",
|
|
component: "UnsafePanel",
|
|
surface: "side_panel",
|
|
props: {}
|
|
});
|
|
|
|
expect(envelope).not.toBeNull();
|
|
expect(envelope ? isUiEnvelopeAllowed(envelope, registry) : false).toBe(false);
|
|
});
|
|
|
|
it("rejects registered components on unsupported surfaces", () => {
|
|
const envelope = parseUiEnvelope({
|
|
kind: "registered_component",
|
|
schemaVersion: "agent-ui@1",
|
|
component: "HistoryPanel",
|
|
surface: "chat_inline",
|
|
props: {}
|
|
});
|
|
|
|
expect(envelope).not.toBeNull();
|
|
expect(envelope ? isUiEnvelopeAllowed(envelope, registry) : false).toBe(false);
|
|
});
|
|
|
|
it("accepts only the echarts-safe-subset chart grammar", () => {
|
|
expect(
|
|
parseUiEnvelope({
|
|
kind: "chart",
|
|
schemaVersion: "agent-ui@1",
|
|
grammar: "vega-lite",
|
|
surface: "chat_inline",
|
|
spec: {},
|
|
data: {}
|
|
})
|
|
).toBeNull();
|
|
|
|
const envelope = parseUiEnvelope({
|
|
kind: "chart",
|
|
schemaVersion: "agent-ui@1",
|
|
grammar: "echarts-safe-subset",
|
|
surface: "chat_inline",
|
|
spec: {},
|
|
data: {}
|
|
});
|
|
|
|
expect(envelope).not.toBeNull();
|
|
expect(envelope ? isUiEnvelopeAllowed(envelope, registry) : false).toBe(true);
|
|
});
|
|
|
|
it("filters invalid registry surfaces before allowlist checks", () => {
|
|
const parsed = parseUiRegistry({
|
|
schema_version: "agent-ui-registry@1",
|
|
chart_grammars: ["echarts-safe-subset"],
|
|
components: [{ id: "HistoryPanel", supportedSurfaces: ["side_panel", "unsafe_surface"] }],
|
|
actions: []
|
|
});
|
|
|
|
expect(parsed?.components[0]?.supportedSurfaces).toEqual(["side_panel"]);
|
|
});
|
|
});
|