import { describe, expect, it } from "vitest";
import { APPS_WORKSPACE_RESOURCE_ID, type InfraAsCodeIntent } from "../notion-as-code/intents.js";
import { validateCustomBlockReferences } from "./validate-custom-block-references.js";

function page(content: string): InfraAsCodeIntent {
	return {
		type: "page",
		resourceId: "home",
		parent: { type: "resourceId", resourceId: APPS_WORKSPACE_RESOURCE_ID },
		content,
	};
}

describe("App page custom block references", () => {
	it("accepts declared keys and workspace definition IDs", () => {
		expect(() =>
			validateCustomBlockReferences(
				[
					page(
						'<custom-block definition-key="board" />\n<custom-block definition-id="f2ec132c-a7dc-4a2f-9922-6275c5653420" />',
					),
				],
				new Set(["board"]),
			),
		).not.toThrow();
	});
	it("reports the page and undeclared key, including nested placements", () => {
		expect(() =>
			validateCustomBlockReferences(
				[page('- <custom-block definition-key="missing" />')],
				new Set(),
			),
		).toThrow('Page "home" references undeclared custom block key "missing"');
	});
	it("ignores fenced and inline code examples", () => {
		expect(() =>
			validateCustomBlockReferences(
				[
					page(
						'```xml\n<custom-block definition-key="example" />\n```\n`<custom-block definition-key="example" />`\n~~~xml\n<custom-block definition-key="example" />\n~~~',
					),
				],
				new Set(),
			),
		).not.toThrow();
	});
});