import { describe, it, expect, vi } from "vitest";
import { render, screen, fireEvent } from "@testing-library/react";
import { AgentLogViewer } from "../AgentLogViewer";
import type { AgentLogEntry } from "@fusion/core";

function makeEntry(overrides: Partial<AgentLogEntry> = {}): AgentLogEntry {
  return {
    timestamp: "2026-01-01T00:00:00Z",
    taskId: "FN-001",
    text: "Hello world",
    type: "text",
    ...overrides,
  };
}

describe("AgentLogViewer", () => {
  it("shows loading message when loading with no entries", () => {
    render(<AgentLogViewer entries={[]} loading={true} />);
    expect(screen.getByText("Loading agent logs…")).toBeTruthy();
  });

  it("shows empty message when no entries and not loading", () => {
    render(<AgentLogViewer entries={[]} loading={false} />);
    expect(screen.getByText("No agent output yet.")).toBeTruthy();
  });

  it("renders text entries as spans in reverse order (newest first)", () => {
    const entries = [
      makeEntry({ text: "first chunk" }),
      makeEntry({ text: "second chunk" }),
    ];
    const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
    const textSpans = container.querySelectorAll(".agent-log-text");
    expect(textSpans).toHaveLength(2);
    // Reversed order: second chunk first, then first chunk
    expect(textSpans[0].textContent).toBe("second chunk");
    expect(textSpans[1].textContent).toBe("first chunk");
  });

  it("renders tool entries with distinct styling", () => {
    const entries = [
      makeEntry({ text: "Read", type: "tool" }),
    ];
    const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
    const toolDiv = container.querySelector(".agent-log-tool");
    expect(toolDiv).toBeTruthy();
    expect(toolDiv!.textContent).toContain("Read");
  });

  it("renders a mix of text and tool entries in reverse order", () => {
    const entries = [
      makeEntry({ text: "Starting...", type: "text" }),
      makeEntry({ text: "Bash", type: "tool" }),
      makeEntry({ text: "Done!", type: "text" }),
    ];
    const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
    // Reversed order: Done! (text), Bash (tool), Starting... (text)
    const textSpans = container.querySelectorAll(".agent-log-text");
    expect(textSpans).toHaveLength(2);
    expect(textSpans[0].textContent).toBe("Done!");
    expect(textSpans[1].textContent).toBe("Starting...");

    const toolDivs = container.querySelectorAll(".agent-log-tool");
    expect(toolDivs).toHaveLength(1);
  });

  it("renders tool entry detail when present", () => {
    const entries = [
      makeEntry({ text: "Bash", type: "tool", detail: "ls -la packages/" }),
    ];
    const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
    const detail = container.querySelector(".agent-log-tool-detail");
    expect(detail).toBeTruthy();
    expect(detail!.textContent).toContain("ls -la packages/");
  });

  it("does not render detail span when detail is absent", () => {
    const entries = [
      makeEntry({ text: "Bash", type: "tool" }),
    ];
    const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
    const detail = container.querySelector(".agent-log-tool-detail");
    expect(detail).toBeNull();
  });

  it("renders long detail text without breaking layout", () => {
    const longDetail = "a/very/long/path/".repeat(10) + "file.ts";
    const entries = [
      makeEntry({ text: "Read", type: "tool", detail: longDetail }),
    ];
    const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
    const detail = container.querySelector(".agent-log-tool-detail");
    expect(detail).toBeTruthy();
    expect(detail!.textContent).toContain(longDetail);
    // Verify the tool div still renders correctly
    const toolDiv = container.querySelector(".agent-log-tool");
    expect(toolDiv).toBeTruthy();
  });

  it("has a monospace font family", () => {
    const entries = [makeEntry()];
    const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
    const viewer = container.querySelector("[data-testid='agent-log-viewer']") as HTMLElement;
    expect(viewer.style.fontFamily).toBe("monospace");
  });

  describe("agent badge deduplication", () => {
    it("shows badge only on the first (newest) of consecutive text entries from the same agent", () => {
      const entries = [
        makeEntry({ text: "chunk 1", type: "text", agent: "executor" }),
        makeEntry({ text: "chunk 2", type: "text", agent: "executor" }),
        makeEntry({ text: "chunk 3", type: "text", agent: "executor" }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const badges = container.querySelectorAll(".agent-log-agent-badge");
      expect(badges).toHaveLength(1);
      // In reversed order, the newest (chunk 3) gets the badge
      expect(badges[0].textContent).toBe("[executor]");
    });

    it("shows badge on each agent transition in reversed order", () => {
      const entries = [
        makeEntry({ text: "hello", type: "text", agent: "triage" }),
        makeEntry({ text: "world", type: "text", agent: "triage" }),
        makeEntry({ text: "starting", type: "text", agent: "executor" }),
        makeEntry({ text: "done", type: "text", agent: "executor" }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const badges = container.querySelectorAll(".agent-log-agent-badge");
      expect(badges).toHaveLength(2);
      // Reversed order: done (executor), starting (executor), world (triage), hello (triage)
      // Badge on done (i=0) and world (transition from executor to triage)
      expect(badges[0].textContent).toBe("[executor]");
      expect(badges[1].textContent).toBe("[triage]");
    });

    it("shows badge on text, tool, and text-after-tool (same agent, type change) in reversed order", () => {
      const entries = [
        makeEntry({ text: "reading...", type: "text", agent: "executor" }),
        makeEntry({ text: "Read", type: "tool", agent: "executor" }),
        makeEntry({ text: "got it", type: "text", agent: "executor" }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const badges = container.querySelectorAll(".agent-log-agent-badge");
      // Reversed: got it (text), Read (tool), reading... (text)
      // Badge on got it (i=0), Read (always block-level), reading... (type changed from tool)
      expect(badges).toHaveLength(3);
    });

    it("shows badge only on the first (newest) of consecutive thinking entries from the same agent", () => {
      const entries = [
        makeEntry({ text: "hmm", type: "thinking", agent: "triage" }),
        makeEntry({ text: "let me think", type: "thinking", agent: "triage" }),
        makeEntry({ text: "ok", type: "thinking", agent: "triage" }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const badges = container.querySelectorAll(".agent-log-agent-badge");
      expect(badges).toHaveLength(1);
      // In reversed order, the newest (ok) gets the badge
      expect(badges[0].textContent).toBe("[triage]");
    });

    it("always shows badge on tool entries regardless of surrounding entries", () => {
      const entries = [
        makeEntry({ text: "Bash", type: "tool", agent: "executor" }),
        makeEntry({ text: "Read", type: "tool", agent: "executor" }),
        makeEntry({ text: "Write", type: "tool", agent: "executor" }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const badges = container.querySelectorAll(".agent-log-agent-badge");
      expect(badges).toHaveLength(3);
    });

    it("always shows badge on tool_result and tool_error entries", () => {
      const entries = [
        makeEntry({ text: "Bash", type: "tool", agent: "executor" }),
        makeEntry({ text: "ok", type: "tool_result", agent: "executor" }),
        makeEntry({ text: "Read", type: "tool", agent: "executor" }),
        makeEntry({ text: "not found", type: "tool_error", agent: "executor" }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const badges = container.querySelectorAll(".agent-log-agent-badge");
      expect(badges).toHaveLength(4);
    });

    it("produces no badges when entries have no agent field", () => {
      const entries = [
        makeEntry({ text: "legacy chunk 1", type: "text" }),
        makeEntry({ text: "legacy chunk 2", type: "text" }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const badges = container.querySelectorAll(".agent-log-agent-badge");
      expect(badges).toHaveLength(0);
    });
  });

  describe("model info header", () => {
    it("renders model info header with executor model when set", () => {
      const entries = [makeEntry()];
      const { container } = render(
        <AgentLogViewer
          entries={entries}
          loading={false}
          executorModel={{ provider: "anthropic", modelId: "claude-sonnet-4-5" }}
        />
      );
      const header = container.querySelector("[data-testid='agent-log-model-header']");
      expect(header).toBeTruthy();
      expect(header!.textContent).toContain("Executor:");
      expect(header!.textContent).toContain("anthropic/claude-sonnet-4-5");
      expect(header!.textContent).toContain("Validator:");
      expect(header!.textContent).toContain("Using default");
      // Verify ProviderIcon is rendered for executor
      expect(container.querySelector('[data-provider="anthropic"]')).toBeTruthy();
    });

    it("renders 'Using default' when no executor model override is set", () => {
      const entries = [makeEntry()];
      const { container } = render(
        <AgentLogViewer entries={entries} loading={false} executorModel={null} />
      );
      const header = container.querySelector("[data-testid='agent-log-model-header']");
      expect(header).toBeTruthy();
      expect(header!.textContent).toContain("Executor:");
      expect(header!.textContent).toContain("Using default");
    });

    it("renders 'Using default' when executorModel is undefined", () => {
      const entries = [makeEntry()];
      const { container } = render(
        <AgentLogViewer entries={entries} loading={false} />
      );
      const header = container.querySelector("[data-testid='agent-log-model-header']");
      expect(header).toBeTruthy();
      expect(header!.textContent).toContain("Executor:");
      expect(header!.textContent).toContain("Using default");
    });

    it("renders model info header with validator model when set", () => {
      const entries = [makeEntry()];
      const { container } = render(
        <AgentLogViewer
          entries={entries}
          loading={false}
          validatorModel={{ provider: "openai", modelId: "gpt-4o" }}
        />
      );
      const header = container.querySelector("[data-testid='agent-log-model-header']");
      expect(header).toBeTruthy();
      expect(header!.textContent).toContain("Validator:");
      expect(header!.textContent).toContain("openai/gpt-4o");
      expect(header!.textContent).toContain("Executor:");
      expect(header!.textContent).toContain("Using default");
      // Verify ProviderIcon is rendered for validator
      expect(container.querySelector('[data-provider="openai"]')).toBeTruthy();
    });

    it("renders 'Using default' when no validator model override is set", () => {
      const entries = [makeEntry()];
      const { container } = render(
        <AgentLogViewer entries={entries} loading={false} validatorModel={null} />
      );
      const header = container.querySelector("[data-testid='agent-log-model-header']");
      expect(header).toBeTruthy();
      expect(header!.textContent).toContain("Validator:");
      expect(header!.textContent).toContain("Using default");
    });

    it("renders both models when both are configured", () => {
      const entries = [makeEntry()];
      const { container } = render(
        <AgentLogViewer
          entries={entries}
          loading={false}
          executorModel={{ provider: "anthropic", modelId: "claude-opus-4" }}
          validatorModel={{ provider: "openai", modelId: "gpt-4o" }}
        />
      );
      const header = container.querySelector("[data-testid='agent-log-model-header']");
      expect(header).toBeTruthy();
      expect(header!.textContent).toContain("Executor:");
      expect(header!.textContent).toContain("anthropic/claude-opus-4");
      expect(header!.textContent).toContain("Validator:");
      expect(header!.textContent).toContain("openai/gpt-4o");
      // Verify both ProviderIcons are rendered
      expect(container.querySelector('[data-provider="anthropic"]')).toBeTruthy();
      expect(container.querySelector('[data-provider="openai"]')).toBeTruthy();
    });

    it("renders header with 'Using default' for both models when both are null/undefined", () => {
      const entries = [makeEntry()];
      const { container } = render(
        <AgentLogViewer entries={entries} loading={false} />
      );
      const header = container.querySelector("[data-testid='agent-log-model-header']");
      expect(header).toBeTruthy();
      expect(header!.textContent).toContain("Executor:");
      expect(header!.textContent).toContain("Using default");
      expect(header!.textContent).toContain("Validator:");
      expect(header!.textContent).toContain("Using default");
    });

    it("shows 'Using default' when executorModel has only provider but no modelId", () => {
      const entries = [makeEntry()];
      const { container } = render(
        <AgentLogViewer
          entries={entries}
          loading={false}
          executorModel={{ provider: "anthropic" }}
        />
      );
      const header = container.querySelector("[data-testid='agent-log-model-header']");
      expect(header).toBeTruthy();
      expect(header!.textContent).toContain("Executor:");
      expect(header!.textContent).toContain("Using default");
    });

    it("shows 'Using default' when executorModel has only modelId but no provider", () => {
      const entries = [makeEntry()];
      const { container } = render(
        <AgentLogViewer
          entries={entries}
          loading={false}
          executorModel={{ modelId: "claude-sonnet-4-5" }}
        />
      );
      const header = container.querySelector("[data-testid='agent-log-model-header']");
      expect(header).toBeTruthy();
      expect(header!.textContent).toContain("Executor:");
      expect(header!.textContent).toContain("Using default");
    });
  });

  describe("horizontal overflow prevention", () => {
    it("sets overflowX hidden on the viewer container", () => {
      const longString = "A".repeat(300);
      const entries = [makeEntry({ text: longString })];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const viewer = container.querySelector("[data-testid='agent-log-viewer']") as HTMLElement;
      expect(viewer.style.overflowX).toBe("hidden");
    });

    it("sets overflow-wrap break-word on the viewer container", () => {
      const entries = [makeEntry({ text: "x".repeat(250) })];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const viewer = container.querySelector("[data-testid='agent-log-viewer']") as HTMLElement;
      expect(viewer.style.overflowWrap).toBe("break-word");
    });

    it("renders pre elements with overflow-x auto for internal scrolling", () => {
      const longLine = "const x = " + "'a'.repeat(500)";
      const entries = [makeEntry({ text: "```\n" + longLine + "\n```" })];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const pre = container.querySelector("pre") as HTMLElement;
      expect(pre).toBeTruthy();
      expect(pre.style.overflowX).toBe("auto");
      expect(pre.style.maxWidth).toBe("100%");
    });

    it("wraps model header on narrow widths via flex-wrap", () => {
      const entries = [makeEntry()];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const header = container.querySelector("[data-testid='agent-log-model-header']") as HTMLElement;
      expect(header.style.flexWrap).toBe("wrap");
    });
  });

  describe("full-height layout", () => {
    it("does not have a fixed maxHeight constraint", () => {
      const entries = [makeEntry()];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const viewer = container.querySelector("[data-testid='agent-log-viewer']") as HTMLElement;
      // The viewer should NOT have a maxHeight of 500px (the old fixed constraint)
      expect(viewer.style.maxHeight).not.toBe("500px");
      // maxHeight should be empty (unset) so the viewer can grow to fill available space
      expect(viewer.style.maxHeight).toBe("");
    });

    it("maintains overflow-y auto for internal scrolling", () => {
      const entries = [makeEntry()];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const viewer = container.querySelector("[data-testid='agent-log-viewer']") as HTMLElement;
      // The viewer itself should remain the scroll container
      expect(viewer.style.overflowY).toBe("auto");
    });

    it("uses agent-log-viewer--streaming class when entries are present", () => {
      const entries = [makeEntry()];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const viewer = container.querySelector("[data-testid='agent-log-viewer']") as HTMLElement;
      expect(viewer.classList.contains("agent-log-viewer")).toBe(true);
      expect(viewer.classList.contains("agent-log-viewer--streaming")).toBe(true);
    });

    it("does not use streaming class on loading state", () => {
      const { container } = render(<AgentLogViewer entries={[]} loading={true} />);
      const viewer = container.querySelector("[data-testid='agent-log-viewer']") as HTMLElement;
      expect(viewer.classList.contains("agent-log-viewer")).toBe(true);
      expect(viewer.classList.contains("agent-log-viewer--streaming")).toBe(false);
    });
  });

  describe("auto-scroll behavior", () => {
    it("scrolls to top when new entries arrive and user is near the top", () => {
      const { rerender, container } = render(<AgentLogViewer entries={[makeEntry({ text: "first" })]} loading={false} />);
      const viewer = container.querySelector("[data-testid='agent-log-viewer']") as HTMLDivElement;
      
      // Simulate user being at the top
      viewer.scrollTop = 0;
      
      // Add a new entry
      rerender(<AgentLogViewer entries={[makeEntry({ text: "second" }), makeEntry({ text: "first" })]} loading={false} />);
      
      // Should have scrolled to top (newest first)
      expect(viewer.scrollTop).toBe(0);
    });

    it("does not auto-scroll when user has scrolled down", () => {
      const { rerender, container } = render(<AgentLogViewer entries={[makeEntry({ text: "first" })]} loading={false} />);
      const viewer = container.querySelector("[data-testid='agent-log-viewer']") as HTMLDivElement;
      
      // Simulate user scrolling down past the threshold
      Object.defineProperty(viewer, 'scrollTop', { value: 100, writable: true });
      
      // Add a new entry
      rerender(<AgentLogViewer entries={[makeEntry({ text: "second" }), makeEntry({ text: "first" })]} loading={false} />);
      
      // Should not have scrolled (scrollTop should remain 100)
      expect(viewer.scrollTop).toBe(100);
    });
  });

  describe("long content preservation", () => {
    it("renders very long text entries without truncation", () => {
      const longText = "A".repeat(5000);
      const entries = [makeEntry({ text: longText })];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const textSpans = container.querySelectorAll(".agent-log-text");
      expect(textSpans).toHaveLength(1);
      expect(textSpans[0].textContent).toBe(longText);
      expect(textSpans[0].textContent!.length).toBe(5000);
    });

    it("renders very long detail text without truncation", () => {
      const longDetail = "B".repeat(5000);
      const entries = [makeEntry({ text: "Read", type: "tool", detail: longDetail })];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const detail = container.querySelector(".agent-log-tool-detail");
      expect(detail).toBeTruthy();
      expect(detail!.textContent).toContain(longDetail);
      expect(detail!.textContent!.length).toBeGreaterThan(5000); // " — " prefix adds a few chars
    });

    it("renders multiline text content without truncation", () => {
      const multilineText = [
        "## Analysis",
        "",
        "After reviewing the codebase:",
        "",
        "1. First issue found",
        "2. Second issue found",
        "",
        "```typescript",
        "const x = 1;",
        "```",
        "",
        "Line " + "C".repeat(2000) + " end",
      ].join("\n");
      const entries = [makeEntry({ text: multilineText })];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const textSpans = container.querySelectorAll(".agent-log-text");
      expect(textSpans).toHaveLength(1);
      // The markdown-rendered content should still contain the essential parts
      expect(textSpans[0].textContent).toContain("Analysis");
      expect(textSpans[0].textContent).toContain("First issue found");
      expect(textSpans[0].textContent).toContain("const x = 1");
    });

    it("renders long tool_result detail without truncation", () => {
      const longDetail = "D".repeat(5000);
      const entries = [makeEntry({ text: "Bash", type: "tool_result", detail: longDetail })];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const detail = container.querySelector(".agent-log-tool-detail");
      expect(detail).toBeTruthy();
      expect(detail!.textContent).toContain(longDetail);
    });

    it("renders long tool_error detail without truncation", () => {
      const longDetail = "E".repeat(5000);
      const entries = [makeEntry({ text: "Write", type: "tool_error", detail: longDetail })];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const detail = container.querySelector(".agent-log-tool-detail");
      expect(detail).toBeTruthy();
      expect(detail!.textContent).toContain(longDetail);
    });
  });

  describe("markdown rendering", () => {
    it("renders plain text without markdown correctly", () => {
      const entries = [
        makeEntry({ text: "Hello world, this is plain text." }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const textSpans = container.querySelectorAll(".agent-log-text");
      expect(textSpans).toHaveLength(1);
      expect(textSpans[0].textContent).toBe("Hello world, this is plain text.");
    });

    it("renders inline markdown elements (bold, italic, inline code)", () => {
      const entries = [
        makeEntry({ text: "This is **bold** and *italic* with `inline code`." }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const textSpans = container.querySelectorAll(".agent-log-text");
      expect(textSpans).toHaveLength(1);
      // Check that the markdown elements are rendered
      const strong = textSpans[0].querySelector("strong");
      expect(strong).toBeTruthy();
      expect(strong!.textContent).toBe("bold");
      const em = textSpans[0].querySelector("em");
      expect(em).toBeTruthy();
      expect(em!.textContent).toBe("italic");
      const code = textSpans[0].querySelector("code");
      expect(code).toBeTruthy();
      expect(code!.textContent).toBe("inline code");
    });

    it("renders code blocks with GFM support", () => {
      const entries = [
        makeEntry({ text: "```typescript\nconst x = 1;\n```" }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const textSpans = container.querySelectorAll(".agent-log-text");
      expect(textSpans).toHaveLength(1);
      // Check that code block is rendered
      const pre = textSpans[0].querySelector("pre");
      expect(pre).toBeTruthy();
      const code = pre!.querySelector("code");
      expect(code).toBeTruthy();
      expect(code!.textContent).toContain("const x = 1");
    });

    it("renders GFM task lists", () => {
      const entries = [
        makeEntry({ text: "- [x] Completed task\n- [ ] Pending task" }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const textSpans = container.querySelectorAll(".agent-log-text");
      expect(textSpans).toHaveLength(1);
      // Check that task list is rendered
      const ul = textSpans[0].querySelector("ul");
      expect(ul).toBeTruthy();
      const taskListItems = ul!.querySelectorAll("li");
      expect(taskListItems).toHaveLength(2);
      // Check checkboxes
      const checkboxes = ul!.querySelectorAll('input[type="checkbox"]');
      expect(checkboxes).toHaveLength(2);
      expect((checkboxes[0] as HTMLInputElement).checked).toBe(true);
      expect((checkboxes[1] as HTMLInputElement).checked).toBe(false);
    });

    it("renders blockquotes", () => {
      const entries = [
        makeEntry({ text: "> This is a blockquote\n> with multiple lines" }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const textSpans = container.querySelectorAll(".agent-log-text");
      expect(textSpans).toHaveLength(1);
      // Check that blockquote is rendered
      const blockquote = textSpans[0].querySelector("blockquote");
      expect(blockquote).toBeTruthy();
      expect(blockquote!.textContent).toContain("This is a blockquote");
    });

    it("renders markdown in thinking entries", () => {
      const entries = [
        makeEntry({ text: "Let me think about **this problem**...", type: "thinking" }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const thinkingSpans = container.querySelectorAll(".agent-log-thinking");
      expect(thinkingSpans).toHaveLength(1);
      const strong = thinkingSpans[0].querySelector("strong");
      expect(strong).toBeTruthy();
      expect(strong!.textContent).toBe("this problem");
    });

    it("renders mixed content with markdown and plain text", () => {
      const entries = [
        makeEntry({ text: "The code:\n\n```js\nconsole.log('hello');\n```\n\nworks!" }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const textSpans = container.querySelectorAll(".agent-log-text");
      expect(textSpans).toHaveLength(1);
      const pre = textSpans[0].querySelector("pre");
      expect(pre).toBeTruthy();
      // Plain text before and after should be preserved
      expect(textSpans[0].textContent).toContain("The code:");
      expect(textSpans[0].textContent).toContain("works!");
    });
  });

  describe("markdown render toggle", () => {
    it("renders the toggle button in the model info header", () => {
      const entries = [makeEntry()];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const toggle = container.querySelector("[data-testid='agent-log-mode-toggle']");
      expect(toggle).toBeTruthy();
      expect(toggle!.textContent).toBe("Markdown");
    });

    it("defaults to markdown mode", () => {
      const entries = [makeEntry({ text: "**bold** text" })];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      // In markdown mode, bold should be rendered as <strong>
      const textSpans = container.querySelectorAll(".agent-log-text");
      const strong = textSpans[0].querySelector("strong");
      expect(strong).toBeTruthy();
      expect(strong!.textContent).toBe("bold");
    });

    it("has correct aria attributes on the toggle", () => {
      const entries = [makeEntry()];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const toggle = container.querySelector("[data-testid='agent-log-mode-toggle']") as HTMLButtonElement;
      expect(toggle).toBeTruthy();
      expect(toggle.getAttribute("aria-pressed")).toBe("true");
      expect(toggle.getAttribute("aria-label")).toBe("Switch to plain text mode");
    });

    it("switches to plain text mode when clicked", () => {
      const entries = [makeEntry({ text: "**bold** and *italic*" })];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const toggle = container.querySelector("[data-testid='agent-log-mode-toggle']") as HTMLButtonElement;

      // Click to switch to plain text mode
      fireEvent.click(toggle);

      // Button should update
      expect(toggle.textContent).toBe("Plain");
      expect(toggle.getAttribute("aria-pressed")).toBe("false");
      expect(toggle.getAttribute("aria-label")).toBe("Switch to markdown mode");

      // Text should now show raw markdown syntax literally
      const textSpans = container.querySelectorAll(".agent-log-text");
      expect(textSpans).toHaveLength(1);
      expect(textSpans[0].textContent).toBe("**bold** and *italic*");
      // No markdown elements should be present
      expect(textSpans[0].querySelector("strong")).toBeNull();
      expect(textSpans[0].querySelector("em")).toBeNull();
    });

    it("toggles back to markdown mode from plain text", () => {
      const entries = [makeEntry({ text: "**bold** text" })];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const toggle = container.querySelector("[data-testid='agent-log-mode-toggle']") as HTMLButtonElement;

      // Switch to plain text
      fireEvent.click(toggle);
      expect(toggle.textContent).toBe("Plain");

      // Switch back to markdown
      fireEvent.click(toggle);
      expect(toggle.textContent).toBe("Markdown");

      // Markdown elements should be present again
      const textSpans = container.querySelectorAll(".agent-log-text");
      const strong = textSpans[0].querySelector("strong");
      expect(strong).toBeTruthy();
      expect(strong!.textContent).toBe("bold");
    });

    it("shows raw markdown syntax literally in plain text mode for text entries", () => {
      const entries = [
        makeEntry({ text: "## Heading\n\n- item 1\n- item 2\n\n`code` and **bold**" }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const toggle = container.querySelector("[data-testid='agent-log-mode-toggle']") as HTMLButtonElement;

      // Switch to plain text
      fireEvent.click(toggle);

      const textSpans = container.querySelectorAll(".agent-log-text");
      expect(textSpans).toHaveLength(1);
      // Raw markdown syntax should appear literally
      expect(textSpans[0].textContent).toContain("## Heading");
      expect(textSpans[0].textContent).toContain("- item 1");
      expect(textSpans[0].textContent).toContain("`code`");
      expect(textSpans[0].textContent).toContain("**bold**");
      // No rendered markdown elements
      expect(textSpans[0].querySelector("h2")).toBeNull();
      expect(textSpans[0].querySelector("ul")).toBeNull();
      expect(textSpans[0].querySelector("code")).toBeNull();
      expect(textSpans[0].querySelector("strong")).toBeNull();
    });

    it("respects toggle for thinking entries", () => {
      const entries = [
        makeEntry({ text: "Thinking about **this**", type: "thinking" }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const toggle = container.querySelector("[data-testid='agent-log-mode-toggle']") as HTMLButtonElement;

      // In markdown mode, bold is rendered
      const thinkingSpans = container.querySelectorAll(".agent-log-thinking");
      expect(thinkingSpans[0].querySelector("strong")).toBeTruthy();

      // Switch to plain text
      fireEvent.click(toggle);

      const thinkingSpansUpdated = container.querySelectorAll(".agent-log-thinking");
      expect(thinkingSpansUpdated[0].textContent).toContain("Thinking about **this**");
      expect(thinkingSpansUpdated[0].querySelector("strong")).toBeNull();
    });

    it("does not affect tool entries in either mode", () => {
      const entries = [
        makeEntry({ text: "Read", type: "tool" }),
        makeEntry({ text: "done", type: "tool_result" }),
        makeEntry({ text: "fail", type: "tool_error" }),
      ];
      const { container, rerender } = render(<AgentLogViewer entries={entries} loading={false} />);
      const toggle = container.querySelector("[data-testid='agent-log-mode-toggle']") as HTMLButtonElement;

      // Tool entries in markdown mode
      expect(container.querySelector(".agent-log-tool")!.textContent).toContain("Read");
      expect(container.querySelector(".agent-log-tool-result")!.textContent).toContain("done");
      expect(container.querySelector(".agent-log-tool-error")!.textContent).toContain("fail");

      // Switch to plain text - tool entries should be unchanged
      fireEvent.click(toggle);

      expect(container.querySelector(".agent-log-tool")!.textContent).toContain("Read");
      expect(container.querySelector(".agent-log-tool-result")!.textContent).toContain("done");
      expect(container.querySelector(".agent-log-tool-error")!.textContent).toContain("fail");
    });

    it("safely renders HTML tags as text in plain text mode (no XSS)", () => {
      const entries = [
        makeEntry({ text: '<script>alert("xss")</script> and <b>bold</b>' }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      const toggle = container.querySelector("[data-testid='agent-log-mode-toggle']") as HTMLButtonElement;

      // Switch to plain text
      fireEvent.click(toggle);

      const textSpans = container.querySelectorAll(".agent-log-text");
      // The text content should contain the literal HTML tags
      expect(textSpans[0].textContent).toContain('<script>alert("xss")</script>');
      expect(textSpans[0].textContent).toContain("<b>bold</b>");
      // No actual script or bold HTML elements should be rendered
      expect(textSpans[0].querySelector("script")).toBeNull();
      expect(textSpans[0].querySelector("b")).toBeNull();
    });

    it("safely renders HTML in markdown mode via react-markdown sanitization", () => {
      const entries = [
        makeEntry({ text: "**safe** text here" }),
      ];
      const { container } = render(<AgentLogViewer entries={entries} loading={false} />);
      // In markdown mode, react-markdown sanitizes HTML (no script execution)
      const textSpans = container.querySelectorAll(".agent-log-text");
      // Markdown formatting should work
      const strong = textSpans[0].querySelector("strong");
      expect(strong).toBeTruthy();
      expect(strong!.textContent).toBe("safe");
      // No script elements are rendered for any HTML content in markdown
      expect(textSpans[0].querySelector("script")).toBeNull();
    });
  });
});