import { describe, expect, it } from "bun:test";
import { renderToStaticMarkup } from "react-dom/server";
import type { ThemedToken } from "shiki";
import { markMatches } from "./code-search-results";
const token = (content: string, offset: number, color: string) =>
({
content,
offset,
htmlStyle: { "--shiki-light": color },
}) as unknown as ThemedToken;
// const findMe = 1;
const tokens = [
token("const", 0, "red"),
token(" ", 5, "grey"),
token("findMe", 6, "blue"),
token(" = 1;", 12, "grey"),
];
const render = (ranges: { start: number; end: number }[]) =>
renderToStaticMarkup(<>{markMatches(tokens, ranges)});
describe("markMatches", () => {
it("keeps every token as it is when nothing matched", () => {
expect(render([])).toBe(
'const findMe = 1;',
);
});
it("cuts a token where a match starts and ends inside it", () => {
expect(render([{ start: 8, end: 10 }])).toContain(
'findMe',
);
});
it("marks a match spanning several tokens as one mark, each piece in its own colour", () => {
expect(render([{ start: 3, end: 9 }])).toBe(
'const findMe = 1;',
);
});
it("keeps adjacent matches apart", () => {
expect(
render([
{ start: 0, end: 5 },
{ start: 5, end: 6 },
]),
).toBe(
'const findMe = 1;',
);
});
});