openclaw/ui/src/components/markdown-document.test.ts

// Control UI tests cover full-document markdown behavior.
import { describe, expect, it } from "vitest";
import { toSanitizedMarkdownHtml } from "./markdown.ts";

function htmlFragment(html: string): HTMLElement {
  const container = document.createElement("div");
  container.innerHTML = html;
  return container;
}

describe("document markdown", () => {
  it("preserves safe remote images while sanitizing unsafe sources", () => {
    const safe = htmlFragment(
      toSanitizedMarkdownHtml("![Alt text](https://example.com/img.png)", {
        mode: "document",
      }),
    );
    const unsafe = htmlFragment(
      toSanitizedMarkdownHtml("![Alt text](javascript:alert(1))", { mode: "document" }),
    );

    expect(safe.querySelector("img")?.getAttribute("src")).toBe("https://example.com/img.png");
    expect(unsafe.querySelector("img")?.hasAttribute("src")).toBe(false);
  });

  it("parses complete documents above message limits", () => {
    const input = `# Start\n\n${"x".repeat(140_000)}\n\n## End`;
    const fragment = htmlFragment(toSanitizedMarkdownHtml(input, { mode: "document" }));

    expect(fragment.querySelector("h1")?.textContent).toBe("Start");
    expect(fragment.querySelector("h2")?.textContent).toBe("End");
    expect(fragment.textContent).not.toContain("truncated");
  });
});