vscode-cowriting-plugin/test/previewAnnotations.test.ts

import MarkdownIt from "markdown-it";
import { describe, expect, it } from "vitest";
import { cowritingMarkdownItPlugin } from "../src/previewAnnotations";

function render(src: string, inputs: any): string {
  const md = new MarkdownIt();
  cowritingMarkdownItPlugin(md, { inputsFor: () => inputs });
  return md.render(src);
}

const base = { baselineReason: "entered", proposals: [], enabled: true };

describe("preview annotations (PUC-3/D21)", () => {
  it("renders clean when disabled", () => {
    const html = render("hello brave world\n", { ...base, enabled: false, baselineText: "hello world\n", spans: [] });
    expect(html).not.toContain("cw-");
  });
  it("colors machine spans blue and human insertions green vs baseline", () => {
    const html = render("hello brave new world\n", {
      ...base,
      baselineText: "hello world\n",
      spans: [{ start: 6, end: 12, author: "claude" }, { start: 12, end: 16, author: "human" }],
    });
    expect(html).toContain('class="cw-ins-claude"');
    expect(html).toContain('class="cw-ins-human"');
  });
  it("strikes deletions vs baseline", () => {
    const html = render("hello world\n", { ...base, baselineText: "hello cruel world\n", spans: [] });
    expect(html).toContain("cw-del");
    expect(html).toContain("cruel");
  });
  it("pinned baseline renders clean even with spans (pin→clean, INV-33/#48)", () => {
    const html = render("hello brave world\n", {
      ...base, baselineReason: "pinned", baselineText: "hello brave world\n",
      spans: [{ start: 6, end: 12, author: "claude" }],
    });
    expect(html).not.toContain("cw-ins");
  });
  it("survives intra-emphasis boundaries (#33 discipline)", () => {
    const html = render("a *bold claim* here\n", {
      ...base, baselineText: "a here\n", spans: [{ start: 2, end: 14, author: "claude" }],
    });
    expect(html).toContain("cw-ins-claude");
    expect(html).not.toMatch(/[-]/); // no PUA sentinel leaks
  });
  // Task 8 (coverage moved from the deleted f10Review.test.ts PUC-3/9): a
  // pending F4 proposal is optimistically applied in the buffer (F12, INV-48),
  // so the preview renders its applied text ins-claude and reinserts the
  // pre-apply original as a deletion — proposals are agent-authored by
  // definition, so this is unconditional (no `spans` entry needed).
  it("a pending proposal renders its applied text cw-ins-claude and the original as a deletion", () => {
    const src = "hello CLAUDE world\n";
    const html = render(src, {
      ...base,
      baselineText: "hello world\n",
      spans: [],
      proposals: [
        { id: "p1", anchorStart: 6, anchorEnd: 12, replaced: "world", replacement: "CLAUDE" },
      ],
    });
    expect(html).toContain('class="cw-ins-claude"');
    expect(html).toContain("CLAUDE");
    expect(html).toContain("cw-del");
    expect(html).toContain("world");
  });
  // Task 8 (INV-32/33 coverage moved from f10Review.test.ts): a block with no
  // spans and no baseline divergence carries no cw- marks at all, even when
  // annotations are enabled — "changed since baseline" is the only trigger.
  it("an unchanged block (baseline == src, no spans) renders with no cw- marks", () => {
    const html = render("hello unchanged world\n", {
      ...base, baselineText: "hello unchanged world\n", spans: [],
    });
    expect(html).not.toContain("cw-");
  });
});

// Task 7 §2.6 / cross-task review follow-up (plan-mandated, INV-29..31): the
// built-in preview's mermaid fence rendering reuses the F7.1 (#22) pure
// re-emit, same as the sunset webview did — a changed mermaid diagram gets
// intra-diagram change styling, not just the plain current diagram. Ports the
// substance of the E2E assertions Task 8 deleted with the webview
// (`git show 2170a0d^:test/e2e/suite/trackChangesPreview.test.ts`, "a changed
// flowchart augments the emitted mermaid source (#22, INV-29)") as pure unit
// tests here instead — no DOM/webview needed to verify SOURCE augmentation.
describe("intra-diagram mermaid diff in the built-in preview (plan T7 §2.6, INV-29)", () => {
  const flowchartBefore = "```mermaid\nflowchart LR\n  a --> b\n```\n";
  const flowchartAfter = "```mermaid\nflowchart LR\n  a --> b\n  a --> c\n```\n";

  it("a changed flowchart fence is re-emitted with cwAdded styling + a legend", () => {
    const html = render(flowchartAfter, { ...base, baselineText: flowchartBefore, spans: [] });
    expect(html).toMatch(/classDef cwAdded/);
    expect(html).toMatch(/class\s+c\s+cwAdded/);
    expect(html).toContain("cw-mermaid-legend");
    expect(html).toContain('class="mermaid"');
  });

  it("an unchanged mermaid fence passes through byte-identical (no diff markup)", () => {
    const html = render(flowchartBefore, { ...base, baselineText: flowchartBefore, spans: [] });
    expect(html).not.toContain("cwAdded");
    expect(html).not.toContain("cw-mermaid-legend");
    expect(html).toContain("flowchart LR");
    expect(html).toContain("a --> b");
  });

  it("annotations disabled: a changed mermaid fence renders the plain current diagram", () => {
    const html = render(flowchartAfter, { ...base, enabled: false, baselineText: flowchartBefore, spans: [] });
    expect(html).not.toContain("cwAdded");
    expect(html).not.toContain("cw-mermaid-legend");
    expect(html).toContain("a --> c");
  });

  it("pinned baseline with zero landed diff: mermaid fence renders the plain current diagram (INV-33/#48 parity)", () => {
    const html = render(flowchartAfter, {
      ...base, baselineReason: "pinned", baselineText: flowchartAfter, spans: [],
    });
    expect(html).not.toContain("cwAdded");
    expect(html).not.toContain("cw-mermaid-legend");
  });

  it("no baseline (no coediting context established): mermaid fence renders the plain current diagram", () => {
    const html = render(flowchartAfter, { ...base, baselineText: undefined, spans: [] });
    expect(html).not.toContain("cwAdded");
    expect(html).not.toContain("cw-mermaid-legend");
  });

  it("a non-mermaid changed fence is untouched by the mermaid queue (fence-kind gate)", () => {
    const before = "```js\nconst a = 1;\n```\n";
    const after = "```js\nconst a = 2;\n```\n";
    const html = render(after, { ...base, baselineText: before, spans: [] });
    expect(html).not.toContain("cwAdded");
    expect(html).not.toContain("cw-mermaid-legend");
    expect(html).toContain('class="language-js"');
  });

  // Code-review hardening (705de31 follow-up): the queue used to be consumed
  // POSITIONALLY per mermaid fence encountered, so a fence the queue doesn't
  // know about (e.g. a pending proposal's optimistic apply, F12/INV-48,
  // inserting a whole new mermaid fence ABOVE a changed one) shifted every
  // later fence's override by one, substituting the wrong diagram's augmented
  // source into the wrong fence instead of the documented verbatim fallback.
  // The fix matches queue entries to fences by BODY: a fence with no matching
  // entry renders verbatim without consuming anything, so a later matching
  // fence still finds its own entry.
  it("misalignment: an extra proposal-inserted mermaid fence above a changed one renders verbatim, and the changed fence still gets its own augmentation", () => {
    const extraFence = "```mermaid\nsequenceDiagram\n  A->>B: hi\n```\n";
    const src = extraFence + "\n" + flowchartAfter;
    const html = render(src, {
      ...base,
      baselineText: flowchartBefore,
      spans: [],
      proposals: [{ id: "p1", anchorStart: 0, anchorEnd: extraFence.length, replaced: "", replacement: extraFence }],
    });
    const fences = html.split('<pre class="mermaid"').slice(1); // one chunk per fence, document order
    expect(fences).toHaveLength(2);
    // the extra fence has no matching queue entry: verbatim, no wrong-diagram substitution
    expect(fences[0]).toContain("sequenceDiagram");
    expect(fences[0]).not.toContain("cwAdded");
    // the changed fence still finds its own augmentation despite the extra fence ahead of it in document order
    expect(fences[1]).toContain("classDef cwAdded");
    expect(fences[1]).toMatch(/class\s+c\s+cwAdded/);
    expect((html.match(/cw-mermaid-legend/g) ?? []).length).toBe(1); // legend only on the augmented fence
  });

  // Body-match must not collapse two changed fences that happen to render the
  // SAME current diagram into a single queue entry — each occurrence gets its
  // own independent augmentation (consume-first-match, not a body->entry map).
  it("two identical-body changed mermaid fences each get their own augmentation", () => {
    const baselineText = flowchartBefore + "\n" + flowchartBefore;
    const src = flowchartAfter + "\n" + flowchartAfter;
    const html = render(src, { ...base, baselineText, spans: [] });
    const fences = html.split('<pre class="mermaid"').slice(1);
    expect(fences).toHaveLength(2);
    for (const f of fences) {
      expect(f).toContain("classDef cwAdded");
      expect(f).toMatch(/class\s+c\s+cwAdded/);
    }
    expect((html.match(/cw-mermaid-legend/g) ?? []).length).toBe(2);
  });
});