Carlos Florêncio
101 lines
3.1 KiB
TypeScript
101 lines
3.1 KiB
TypeScript
import { createRef } from "react";
|
|
import { cleanup, render, screen } from "@testing-library/react";
|
|
import { TooltipProvider } from "@kandev/ui/tooltip";
|
|
import { afterEach, describe, expect, it, vi } from "vitest";
|
|
import { ChatInputBody, type ChatInputBodyProps } from "./chat-input-body";
|
|
|
|
vi.mock("./tiptap-input", () => ({
|
|
TipTapInput: () => <div data-testid="mock-tiptap-input" />,
|
|
}));
|
|
|
|
vi.mock("./chat-input-toolbar", () => ({
|
|
ChatInputToolbar: () => <div data-testid="mock-chat-input-toolbar" />,
|
|
}));
|
|
|
|
vi.mock("./context-items/context-zone", () => ({
|
|
ContextZone: () => <div data-testid="mock-context-zone" />,
|
|
}));
|
|
|
|
afterEach(() => {
|
|
cleanup();
|
|
});
|
|
|
|
function props(overrides: Partial<ChatInputBodyProps> = {}): ChatInputBodyProps {
|
|
return {
|
|
containerRef: createRef<HTMLDivElement>(),
|
|
height: 120,
|
|
resizeHandleProps: { onMouseDown: vi.fn(), onDoubleClick: vi.fn() },
|
|
isStarting: false,
|
|
isAgentBusy: false,
|
|
hasClarification: false,
|
|
showRequestChangesTooltip: false,
|
|
hasPendingComments: false,
|
|
planModeEnabled: false,
|
|
showFocusHint: false,
|
|
needsRecovery: false,
|
|
addFiles: vi.fn().mockResolvedValue(undefined),
|
|
contextAreaProps: {
|
|
hasContextZone: false,
|
|
allItems: [],
|
|
sessionId: "session-1",
|
|
},
|
|
editorAreaProps: {
|
|
inputRef: createRef(),
|
|
value: "",
|
|
handleChange: vi.fn(),
|
|
handleSubmitWithReset: vi.fn(),
|
|
inputPlaceholder: "Ask to make changes",
|
|
isDisabled: false,
|
|
submitDisabled: false,
|
|
hasClarification: false,
|
|
planModeEnabled: false,
|
|
planModeAvailable: true,
|
|
mcpServers: [],
|
|
submitKey: "cmd_enter",
|
|
setIsInputFocused: vi.fn(),
|
|
sessionId: "session-1",
|
|
taskId: "task-1",
|
|
planContextEnabled: false,
|
|
handleAgentCommand: vi.fn(),
|
|
addFiles: vi.fn().mockResolvedValue(undefined),
|
|
fileInputRef: createRef(),
|
|
showRequestChangesTooltip: false,
|
|
isAgentBusy: false,
|
|
onPlanModeChange: vi.fn(),
|
|
taskDescription: "",
|
|
isSending: false,
|
|
onCancel: vi.fn(),
|
|
contextCount: 0,
|
|
contextPopoverOpen: false,
|
|
setContextPopoverOpen: vi.fn(),
|
|
contextFiles: [],
|
|
},
|
|
...overrides,
|
|
};
|
|
}
|
|
|
|
describe("ChatInputBody", () => {
|
|
it("reserves right-side editable space while the focus hint is visible", () => {
|
|
render(
|
|
<TooltipProvider>
|
|
<ChatInputBody {...props({ showFocusHint: true })} />
|
|
</TooltipProvider>,
|
|
);
|
|
|
|
expect(screen.getByText("to focus")).toBeTruthy();
|
|
expect(screen.getByTestId("chat-input-editor-shell").className).not.toContain("pr-28");
|
|
expect(screen.getByTestId("mock-tiptap-input").parentElement?.className).toContain("pr-28");
|
|
});
|
|
|
|
it("does not reserve focus-hint space when the hint is hidden", () => {
|
|
render(
|
|
<TooltipProvider>
|
|
<ChatInputBody {...props({ showFocusHint: false })} />
|
|
</TooltipProvider>,
|
|
);
|
|
|
|
expect(screen.getByTestId("chat-input-editor-shell").className).not.toContain("pr-28");
|
|
expect(screen.getByTestId("mock-tiptap-input").parentElement?.className).not.toContain("pr-28");
|
|
});
|
|
});
|