import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";

const cliHighlightMocks = vi.hoisted(() => ({

highlight: vi.fn((code: string) => code),

supportsLanguage: vi.fn((_lang: string) => true),

}));

vi.mock("cli-highlight", () => cliHighlightMocks);

import { afterEach, describe, expect, it, vi } from "vitest";

const { markdownTheme, searchableSelectListTheme, selectListTheme, theme } =

await import("./theme.js");

describe("markdownTheme", () => {

describe("highlightCode", () => {

beforeEach(() => {

cliHighlightMocks.highlight.mockClear();

cliHighlightMocks.supportsLanguage.mockClear();

cliHighlightMocks.highlight.mockImplementation((code: string) => code);

cliHighlightMocks.supportsLanguage.mockReturnValue(true);

});

it("passes supported language through to the highlighter", () => {

markdownTheme.highlightCode!("const x = 42;", "javascript");

expect(cliHighlightMocks.supportsLanguage).toHaveBeenCalledWith("javascript");

expect(cliHighlightMocks.highlight).toHaveBeenCalledWith(

"const x = 42;",

expect.objectContaining({ language: "javascript" }),

);

});

it("falls back to auto-detect for unknown language and preserves lines", () => {

cliHighlightMocks.supportsLanguage.mockReturnValue(false);

cliHighlightMocks.highlight.mockImplementation((code: string) => `${code}\nline-2`);

it("renders code blocks with the theme code color and preserves lines", () => {

const result = markdownTheme.highlightCode!(`echo "hello"`, "not-a-real-language");

expect(cliHighlightMocks.highlight).toHaveBeenCalledWith(

`echo "hello"`,

expect.objectContaining({ language: undefined }),

);

expect(stripAnsi(result[0] ?? "")).toContain("echo");

expect(stripAnsi(result[1] ?? "")).toBe("line-2");

});

it("returns plain highlighted lines when highlighting throws", () => {

cliHighlightMocks.highlight.mockImplementation(() => {

throw new Error("boom");

});

const result = markdownTheme.highlightCode!("echo hello", "javascript");

expect(result).toHaveLength(1);

expect(stripAnsi(result[0] ?? "")).toBe("echo hello");

it("preserves multi-line code blocks", () => {

const result = markdownTheme.highlightCode!("line-1\nline-2", "javascript");

expect(result.map((line) => stripAnsi(line))).toEqual(["line-1", "line-2"]);

});

});

});