José Almeida
107 lines
3.5 KiB
TypeScript
107 lines
3.5 KiB
TypeScript
import { cleanup, render, waitFor } from "@testing-library/react";
|
|
import { afterEach, describe, expect, it, vi } from "vitest";
|
|
|
|
import { DiffViewer } from "./diff-viewer";
|
|
import type { FileDiffData } from "@/lib/diff/types";
|
|
|
|
const expansionState = vi.hoisted(() => ({
|
|
isContentLoaded: true,
|
|
isLoading: false,
|
|
error: null as string | null,
|
|
loadContent: vi.fn(),
|
|
canExpand: true,
|
|
}));
|
|
|
|
vi.mock("./use-expandable-diff", () => ({
|
|
useExpandableDiff: (options: { fileDiffMetadata: unknown }) => ({
|
|
...expansionState,
|
|
metadata: options.fileDiffMetadata,
|
|
}),
|
|
}));
|
|
|
|
const fileDiffProps: Array<{
|
|
options?: { overflow?: string; hunkSeparators?: string };
|
|
}> = [];
|
|
|
|
vi.mock("@/components/theme/app-theme", () => ({
|
|
useTheme: () => ({ resolvedTheme: "dark" }),
|
|
}));
|
|
|
|
vi.mock("@pierre/diffs/react", () => ({
|
|
FileDiff: (props: { options?: { overflow?: string; hunkSeparators?: string } }) => {
|
|
fileDiffProps.push(props);
|
|
return <div data-testid="file-diff" />;
|
|
},
|
|
}));
|
|
|
|
const data: FileDiffData = {
|
|
filePath: "src/example.ts",
|
|
diff: "diff --git a/src/example.ts b/src/example.ts\n--- a/src/example.ts\n+++ b/src/example.ts\n@@ -1 +1 @@\n-old\n+new\n",
|
|
oldContent: "old\n",
|
|
newContent: "new\n",
|
|
additions: 1,
|
|
deletions: 1,
|
|
};
|
|
|
|
afterEach(() => {
|
|
cleanup();
|
|
fileDiffProps.length = 0;
|
|
expansionState.canExpand = true;
|
|
expansionState.loadContent.mockReset();
|
|
});
|
|
|
|
describe("DiffViewer", () => {
|
|
it("wraps diff lines by default", async () => {
|
|
render(<DiffViewer data={data} />);
|
|
|
|
await waitFor(() => expect(fileDiffProps.length).toBeGreaterThan(0));
|
|
expect(fileDiffProps.at(-1)?.options?.overflow).toBe("wrap");
|
|
});
|
|
|
|
it("rerenders when the controlled delete handler changes", async () => {
|
|
const firstDelete = vi.fn();
|
|
const secondDelete = vi.fn();
|
|
const { rerender } = render(<DiffViewer data={data} onCommentDelete={firstDelete} />);
|
|
|
|
await waitFor(() => expect(fileDiffProps.length).toBeGreaterThan(0));
|
|
const renderCount = fileDiffProps.length;
|
|
|
|
rerender(<DiffViewer data={data} onCommentDelete={secondDelete} />);
|
|
|
|
await waitFor(() => expect(fileDiffProps.length).toBeGreaterThan(renderCount));
|
|
});
|
|
|
|
it("rerenders when the controlled update handler changes", async () => {
|
|
const firstUpdate = vi.fn();
|
|
const secondUpdate = vi.fn();
|
|
const { rerender } = render(<DiffViewer data={data} onCommentUpdate={firstUpdate} />);
|
|
|
|
await waitFor(() => expect(fileDiffProps.length).toBeGreaterThan(0));
|
|
const renderCount = fileDiffProps.length;
|
|
|
|
rerender(<DiffViewer data={data} onCommentUpdate={secondUpdate} />);
|
|
|
|
await waitFor(() => expect(fileDiffProps.length).toBeGreaterThan(renderCount));
|
|
});
|
|
|
|
it("rerenders when walkthrough annotations are disabled", async () => {
|
|
const { rerender } = render(<DiffViewer data={data} enableWalkthroughAnnotations />);
|
|
|
|
await waitFor(() => expect(fileDiffProps.length).toBeGreaterThan(0));
|
|
const renderCount = fileDiffProps.length;
|
|
|
|
rerender(<DiffViewer data={data} enableWalkthroughAnnotations={false} />);
|
|
|
|
await waitFor(() => expect(fileDiffProps.length).toBeGreaterThan(renderCount));
|
|
});
|
|
|
|
it("keeps simple hunk separators when full-content reparse rejects expansion", async () => {
|
|
expansionState.canExpand = false;
|
|
|
|
render(<DiffViewer data={data} sessionId="session-1" enableExpansion />);
|
|
|
|
await waitFor(() => expect(fileDiffProps.length).toBeGreaterThan(0));
|
|
expect(fileDiffProps.at(-1)?.options?.hunkSeparators).toBe("simple");
|
|
});
|
|
});
|