3c9901b2cf
- Add planning mode checkbox to NewTaskModal with onPlanGenerated callback - Wire up planning mode flow in App.tsx to open modal after task creation - Add initialPlan prop to PlanningModeModal with auto-start logic - Add comprehensive tests for planning mode checkbox flow in both modals - Include changeset for the planning mode checkbox fix
377 lines
12 KiB
TypeScript
377 lines
12 KiB
TypeScript
import { describe, it, expect, vi, beforeEach } from "vitest";
|
||
import { render, screen, fireEvent, waitFor } from "@testing-library/react";
|
||
import { NewTaskModal } from "../NewTaskModal";
|
||
import type { Task, Column } from "@kb/core";
|
||
|
||
// Mock the api module
|
||
vi.mock("../../api", () => ({
|
||
uploadAttachment: vi.fn().mockResolvedValue({}),
|
||
fetchModels: vi.fn().mockResolvedValue([
|
||
{ provider: "anthropic", id: "claude-sonnet-4-5", name: "Claude Sonnet 4.5", reasoning: true, contextWindow: 200000 },
|
||
{ provider: "openai", id: "gpt-4o", name: "GPT-4o", reasoning: false, contextWindow: 128000 },
|
||
]),
|
||
updateTask: vi.fn().mockResolvedValue({}),
|
||
}));
|
||
|
||
function makeTask(id: string): Task {
|
||
return {
|
||
id,
|
||
title: `Task ${id}`,
|
||
description: `Description for ${id}`,
|
||
column: "todo" as Column,
|
||
status: undefined as any,
|
||
steps: [],
|
||
currentStep: 0,
|
||
dependencies: [],
|
||
log: [],
|
||
createdAt: "2026-01-01T00:00:00Z",
|
||
updatedAt: "2026-01-01T00:00:00Z",
|
||
};
|
||
}
|
||
|
||
function renderNewTaskModal(props = {}) {
|
||
const defaultProps = {
|
||
isOpen: true,
|
||
onClose: vi.fn(),
|
||
tasks: [] as Task[],
|
||
onCreateTask: vi.fn().mockResolvedValue({ id: "KB-001" }),
|
||
addToast: vi.fn(),
|
||
};
|
||
const mergedProps = { ...defaultProps, ...props };
|
||
const result = render(<NewTaskModal {...mergedProps} />);
|
||
return { ...result, props: mergedProps };
|
||
}
|
||
|
||
describe("NewTaskModal", () => {
|
||
beforeEach(() => {
|
||
vi.clearAllMocks();
|
||
});
|
||
|
||
it("does not render when isOpen is false", () => {
|
||
renderNewTaskModal({ isOpen: false });
|
||
expect(screen.queryByText("New Task")).toBeNull();
|
||
});
|
||
|
||
it("renders all form fields when open", () => {
|
||
renderNewTaskModal();
|
||
|
||
expect(screen.getByText("New Task")).toBeTruthy();
|
||
expect(screen.getByLabelText(/Title/i)).toBeTruthy();
|
||
expect(screen.getByLabelText(/Description/i)).toBeTruthy();
|
||
expect(screen.getByRole("button", { name: "Add dependencies" })).toBeTruthy();
|
||
expect(screen.getByText(/Model Configuration/i)).toBeTruthy();
|
||
expect(screen.getByLabelText(/Enable planning mode/i)).toBeTruthy();
|
||
expect(screen.getByText(/Attachments/i)).toBeTruthy();
|
||
expect(screen.getByRole("button", { name: "Create Task" })).toBeTruthy();
|
||
expect(screen.getByRole("button", { name: "Cancel" })).toBeTruthy();
|
||
});
|
||
|
||
it("creates task with all provided data on submit", async () => {
|
||
const { props } = renderNewTaskModal();
|
||
|
||
const titleInput = screen.getByLabelText(/Title/i);
|
||
const descTextarea = screen.getByLabelText(/Description/i);
|
||
|
||
fireEvent.change(titleInput, { target: { value: "My Task Title" } });
|
||
fireEvent.change(descTextarea, { target: { value: "My task description" } });
|
||
|
||
fireEvent.click(screen.getByRole("button", { name: "Create Task" }));
|
||
|
||
await waitFor(() => {
|
||
expect(props.onCreateTask).toHaveBeenCalledWith(
|
||
expect.objectContaining({
|
||
title: "My Task Title",
|
||
description: "My task description",
|
||
column: "triage",
|
||
}),
|
||
);
|
||
});
|
||
});
|
||
|
||
it("calls onClose after successful creation", async () => {
|
||
const { props } = renderNewTaskModal();
|
||
|
||
const descTextarea = screen.getByLabelText(/Description/i);
|
||
fireEvent.change(descTextarea, { target: { value: "Test description" } });
|
||
|
||
fireEvent.click(screen.getByRole("button", { name: "Create Task" }));
|
||
|
||
await waitFor(() => {
|
||
expect(props.onClose).toHaveBeenCalled();
|
||
});
|
||
});
|
||
|
||
it("shows error toast on creation failure", async () => {
|
||
const { props } = renderNewTaskModal({
|
||
onCreateTask: vi.fn().mockRejectedValue(new Error("Creation failed")),
|
||
});
|
||
|
||
const descTextarea = screen.getByLabelText(/Description/i);
|
||
fireEvent.change(descTextarea, { target: { value: "Test description" } });
|
||
|
||
fireEvent.click(screen.getByRole("button", { name: "Create Task" }));
|
||
|
||
await waitFor(() => {
|
||
expect(props.addToast).toHaveBeenCalledWith("Creation failed", "error");
|
||
});
|
||
});
|
||
|
||
it("disables create button when description is empty", () => {
|
||
renderNewTaskModal();
|
||
|
||
const createButton = screen.getByRole("button", { name: "Create Task" });
|
||
expect(createButton).toBeDisabled();
|
||
});
|
||
|
||
it("enables create button when description is not empty", () => {
|
||
renderNewTaskModal();
|
||
|
||
const descTextarea = screen.getByLabelText(/Description/i);
|
||
fireEvent.change(descTextarea, { target: { value: "Some description" } });
|
||
|
||
const createButton = screen.getByRole("button", { name: "Create Task" });
|
||
expect(createButton).not.toBeDisabled();
|
||
});
|
||
|
||
it("closes modal on cancel button click", () => {
|
||
const { props } = renderNewTaskModal();
|
||
|
||
fireEvent.click(screen.getByRole("button", { name: "Cancel" }));
|
||
|
||
expect(props.onClose).toHaveBeenCalled();
|
||
});
|
||
|
||
it("closes modal on X button click", () => {
|
||
const { props } = renderNewTaskModal();
|
||
|
||
fireEvent.click(screen.getByText("×"));
|
||
|
||
expect(props.onClose).toHaveBeenCalled();
|
||
});
|
||
|
||
it("adds dependencies via dropdown", async () => {
|
||
const tasks = [makeTask("KB-010"), makeTask("KB-020")];
|
||
const { props } = renderNewTaskModal({ tasks });
|
||
|
||
// Open dependencies dropdown
|
||
fireEvent.click(screen.getByRole("button", { name: "Add dependencies" }));
|
||
|
||
// Wait for dropdown to appear
|
||
await waitFor(() => {
|
||
expect(document.querySelector(".dep-dropdown")).toBeTruthy();
|
||
});
|
||
|
||
// Click on a task to add it as dependency
|
||
const items = document.querySelectorAll(".dep-dropdown-item");
|
||
expect(items.length).toBeGreaterThan(0);
|
||
fireEvent.click(items[0]!);
|
||
|
||
// Verify dependency was added (it should show as selected)
|
||
const descTextarea = screen.getByLabelText(/Description/i);
|
||
fireEvent.change(descTextarea, { target: { value: "Task with deps" } });
|
||
|
||
fireEvent.click(screen.getByRole("button", { name: "Create Task" }));
|
||
|
||
await waitFor(() => {
|
||
expect(props.onCreateTask).toHaveBeenCalledWith(
|
||
expect.objectContaining({
|
||
dependencies: expect.any(Array),
|
||
}),
|
||
);
|
||
});
|
||
});
|
||
|
||
it("renders model selectors when models are loaded", async () => {
|
||
renderNewTaskModal();
|
||
|
||
await waitFor(() => {
|
||
expect(screen.getByText("Executor")).toBeTruthy();
|
||
expect(screen.getByText("Validator")).toBeTruthy();
|
||
});
|
||
});
|
||
|
||
it("toggles planning mode checkbox", () => {
|
||
renderNewTaskModal();
|
||
|
||
const checkbox = screen.getByLabelText(/Enable planning mode/i) as HTMLInputElement;
|
||
expect(checkbox.checked).toBe(false);
|
||
|
||
fireEvent.click(checkbox);
|
||
expect(checkbox.checked).toBe(true);
|
||
|
||
fireEvent.click(checkbox);
|
||
expect(checkbox.checked).toBe(false);
|
||
});
|
||
|
||
it("shows success toast after creation", async () => {
|
||
const { props } = renderNewTaskModal({
|
||
onCreateTask: vi.fn().mockResolvedValue({ id: "KB-042" }),
|
||
});
|
||
|
||
const descTextarea = screen.getByLabelText(/Description/i);
|
||
fireEvent.change(descTextarea, { target: { value: "Test description" } });
|
||
|
||
fireEvent.click(screen.getByRole("button", { name: "Create Task" }));
|
||
|
||
await waitFor(() => {
|
||
expect(props.addToast).toHaveBeenCalledWith("Created KB-042", "success");
|
||
});
|
||
});
|
||
|
||
it("confirms before closing with dirty state", () => {
|
||
const { props } = renderNewTaskModal();
|
||
|
||
// Add some content to make it dirty
|
||
const descTextarea = screen.getByLabelText(/Description/i);
|
||
fireEvent.change(descTextarea, { target: { value: "Some text" } });
|
||
|
||
// Mock confirm to return false (cancel)
|
||
const originalConfirm = window.confirm;
|
||
window.confirm = vi.fn().mockReturnValue(false);
|
||
|
||
fireEvent.click(screen.getByRole("button", { name: "Cancel" }));
|
||
|
||
expect(window.confirm).toHaveBeenCalledWith("You have unsaved changes. Discard them?");
|
||
expect(props.onClose).not.toHaveBeenCalled();
|
||
|
||
window.confirm = originalConfirm;
|
||
});
|
||
|
||
it("closes without confirm when state is not dirty", () => {
|
||
const { props } = renderNewTaskModal();
|
||
|
||
fireEvent.click(screen.getByRole("button", { name: "Cancel" }));
|
||
|
||
expect(props.onClose).toHaveBeenCalled();
|
||
});
|
||
|
||
it("creates task without title when title is empty", async () => {
|
||
const { props } = renderNewTaskModal();
|
||
|
||
const descTextarea = screen.getByLabelText(/Description/i);
|
||
fireEvent.change(descTextarea, { target: { value: "Only description" } });
|
||
|
||
fireEvent.click(screen.getByRole("button", { name: "Create Task" }));
|
||
|
||
await waitFor(() => {
|
||
expect(props.onCreateTask).toHaveBeenCalledWith(
|
||
expect.objectContaining({
|
||
title: undefined,
|
||
description: "Only description",
|
||
}),
|
||
);
|
||
});
|
||
});
|
||
|
||
// Planning mode tests
|
||
it("calls onPlanningMode when planning mode is checked and form is submitted", async () => {
|
||
const onPlanningMode = vi.fn();
|
||
const { props } = renderNewTaskModal({ onPlanningMode });
|
||
|
||
const titleInput = screen.getByLabelText(/Title/i);
|
||
const descTextarea = screen.getByLabelText(/Description/i);
|
||
const checkbox = screen.getByLabelText(/Enable planning mode/i);
|
||
|
||
fireEvent.change(titleInput, { target: { value: "My Task" } });
|
||
fireEvent.change(descTextarea, { target: { value: "Build a login system" } });
|
||
fireEvent.click(checkbox);
|
||
|
||
fireEvent.click(screen.getByRole("button", { name: "Create Task" }));
|
||
|
||
await waitFor(() => {
|
||
expect(onPlanningMode).toHaveBeenCalledWith("Build a login system");
|
||
});
|
||
});
|
||
|
||
it("does NOT call onCreateTask when planning mode is checked", async () => {
|
||
const onPlanningMode = vi.fn();
|
||
const { props } = renderNewTaskModal({ onPlanningMode });
|
||
|
||
const descTextarea = screen.getByLabelText(/Description/i);
|
||
const checkbox = screen.getByLabelText(/Enable planning mode/i);
|
||
|
||
fireEvent.change(descTextarea, { target: { value: "Build a login system" } });
|
||
fireEvent.click(checkbox);
|
||
|
||
fireEvent.click(screen.getByRole("button", { name: "Create Task" }));
|
||
|
||
await waitFor(() => {
|
||
expect(onPlanningMode).toHaveBeenCalled();
|
||
});
|
||
|
||
expect(props.onCreateTask).not.toHaveBeenCalled();
|
||
});
|
||
|
||
it("calls onCreateTask normally when planning mode is unchecked", async () => {
|
||
const onPlanningMode = vi.fn();
|
||
const { props } = renderNewTaskModal({ onPlanningMode });
|
||
|
||
const descTextarea = screen.getByLabelText(/Description/i);
|
||
fireEvent.change(descTextarea, { target: { value: "Normal task" } });
|
||
|
||
// Ensure planning mode is unchecked
|
||
const checkbox = screen.getByLabelText(/Enable planning mode/i) as HTMLInputElement;
|
||
expect(checkbox.checked).toBe(false);
|
||
|
||
fireEvent.click(screen.getByRole("button", { name: "Create Task" }));
|
||
|
||
await waitFor(() => {
|
||
expect(props.onCreateTask).toHaveBeenCalledWith(
|
||
expect.objectContaining({
|
||
description: "Normal task",
|
||
}),
|
||
);
|
||
});
|
||
|
||
expect(onPlanningMode).not.toHaveBeenCalled();
|
||
});
|
||
|
||
it("closes modal after triggering planning mode", async () => {
|
||
const onPlanningMode = vi.fn();
|
||
const { props } = renderNewTaskModal({ onPlanningMode });
|
||
|
||
const descTextarea = screen.getByLabelText(/Description/i);
|
||
const checkbox = screen.getByLabelText(/Enable planning mode/i);
|
||
|
||
fireEvent.change(descTextarea, { target: { value: "Build a login system" } });
|
||
fireEvent.click(checkbox);
|
||
|
||
fireEvent.click(screen.getByRole("button", { name: "Create Task" }));
|
||
|
||
await waitFor(() => {
|
||
expect(props.onClose).toHaveBeenCalled();
|
||
});
|
||
});
|
||
|
||
it("clears form state after triggering planning mode", async () => {
|
||
const onPlanningMode = vi.fn();
|
||
renderNewTaskModal({ onPlanningMode });
|
||
|
||
const titleInput = screen.getByLabelText(/Title/i);
|
||
const descTextarea = screen.getByLabelText(/Description/i);
|
||
const checkbox = screen.getByLabelText(/Enable planning mode/i);
|
||
|
||
fireEvent.change(titleInput, { target: { value: "My Task" } });
|
||
fireEvent.change(descTextarea, { target: { value: "Build a login system" } });
|
||
fireEvent.click(checkbox);
|
||
|
||
fireEvent.click(screen.getByRole("button", { name: "Create Task" }));
|
||
|
||
await waitFor(() => {
|
||
expect(onPlanningMode).toHaveBeenCalled();
|
||
});
|
||
|
||
// Re-open the modal and check that state is cleared
|
||
renderNewTaskModal({
|
||
isOpen: true,
|
||
onPlanningMode,
|
||
onClose: vi.fn(),
|
||
});
|
||
|
||
await waitFor(() => {
|
||
const newDescTextarea = screen.getAllByLabelText(/Description/i)[0];
|
||
expect(newDescTextarea).toHaveValue("");
|
||
});
|
||
});
|
||
});
|