nova28
126 lines
4.6 KiB
TypeScript
126 lines
4.6 KiB
TypeScript
import { cleanup, render, screen, waitFor } from "@testing-library/react";
|
|
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
|
|
|
const mockFetchTaskSession = vi.hoisted(() => vi.fn());
|
|
|
|
vi.mock("@/lib/api/domains/session-api", () => ({
|
|
fetchTaskSession: (...args: unknown[]) => mockFetchTaskSession(...args),
|
|
}));
|
|
|
|
// Prefetching executors and agent settings is orthogonal to session hydration,
|
|
// and it fires network calls this test has no opinion about.
|
|
vi.mock("@/hooks/domains/settings/use-settings-data", () => ({
|
|
useSettingsData: () => {},
|
|
}));
|
|
|
|
import { StateProvider } from "@/components/state-provider";
|
|
import { ToastProvider } from "@/components/toast-provider";
|
|
import { TooltipProvider } from "@kandev/ui/tooltip";
|
|
import { RunTranscript } from "./run-transcript";
|
|
|
|
const SESSION_ID = "session-1";
|
|
const TASK_ID = "task-1";
|
|
|
|
function transcript() {
|
|
return (
|
|
<StateProvider>
|
|
<ToastProvider>
|
|
<TooltipProvider>
|
|
<RunTranscript sessionId={SESSION_ID} taskId={TASK_ID} />
|
|
</TooltipProvider>
|
|
</ToastProvider>
|
|
</StateProvider>
|
|
);
|
|
}
|
|
|
|
function sessionInState(state: string) {
|
|
return { session: { id: SESSION_ID, task_id: TASK_ID, state } };
|
|
}
|
|
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
mockFetchTaskSession.mockResolvedValue({
|
|
session: { id: SESSION_ID, task_id: TASK_ID, state: "WAITING_FOR_INPUT" },
|
|
});
|
|
});
|
|
|
|
afterEach(() => {
|
|
// Each case renders the whole chat stack; without this the trees stack up and
|
|
// a by-test-id lookup starts matching the previous case's composer.
|
|
cleanup();
|
|
vi.restoreAllMocks();
|
|
});
|
|
|
|
describe("RunTranscript session hydration", () => {
|
|
// Automation tasks are hidden from the boot payload by their origin, so a
|
|
// direct load of /automations/<id> — a shared link, or a reload — arrives
|
|
// with no session row in the store. Without hydration `useSession` never
|
|
// subscribes and the composer rejects every reply as session-unavailable,
|
|
// which is the one thing this surface exists to allow.
|
|
//
|
|
// Mounted against a real store with `useChatPanelState` left unmocked, so it
|
|
// exercises the state a fresh navigation actually produces rather than a
|
|
// stand-in for it.
|
|
it("fetches the session the store was never given", async () => {
|
|
render(transcript());
|
|
|
|
await waitFor(() => expect(mockFetchTaskSession).toHaveBeenCalledWith(SESSION_ID));
|
|
});
|
|
|
|
it("asks only once, however many times the panel re-renders", async () => {
|
|
const { rerender } = render(transcript());
|
|
|
|
await waitFor(() => expect(mockFetchTaskSession).toHaveBeenCalledTimes(1));
|
|
rerender(transcript());
|
|
|
|
expect(mockFetchTaskSession).toHaveBeenCalledTimes(1);
|
|
});
|
|
});
|
|
|
|
describe("RunTranscript live-session controls", () => {
|
|
// Replying to a run starts the agent, it works the prompt, and it shuts down
|
|
// again. Controls that speak to a live ACP session — the model picker above
|
|
// all — must not outlive the turn: they still look operable, and changing a
|
|
// model on a process that no longer exists silently does nothing.
|
|
it("offers the model selector while a turn is running", async () => {
|
|
mockFetchTaskSession.mockResolvedValue(sessionInState("RUNNING"));
|
|
|
|
render(transcript());
|
|
|
|
await waitFor(() => expect(screen.getByTestId("toolbar-item-model")).toBeTruthy());
|
|
});
|
|
|
|
it("unmounts the model selector once the turn ends and the run parks", async () => {
|
|
// WAITING_FOR_INPUT is the state a finished automation run parks in — it is
|
|
// what keeps the run repliable, and it is exactly the state with no agent
|
|
// behind it.
|
|
mockFetchTaskSession.mockResolvedValue(sessionInState("WAITING_FOR_INPUT"));
|
|
|
|
render(transcript());
|
|
|
|
await waitFor(() => expect(screen.getByTestId("chat-input-area")).toBeTruthy());
|
|
expect(screen.queryByTestId("toolbar-item-model")).toBeNull();
|
|
});
|
|
|
|
it("keeps the composer itself, because the run is still repliable", async () => {
|
|
mockFetchTaskSession.mockResolvedValue(sessionInState("WAITING_FOR_INPUT"));
|
|
|
|
render(transcript());
|
|
|
|
await waitFor(() => expect(screen.getByTestId("chat-input-area")).toBeTruthy());
|
|
expect(screen.getByTestId("chat-input-toolbar")).toBeTruthy();
|
|
});
|
|
|
|
it("draws the composer on the page's own background, not the workbench plate", async () => {
|
|
// A lighter strip along the bottom of a `bg-background` page read as a
|
|
// panel that had been left behind.
|
|
mockFetchTaskSession.mockResolvedValue(sessionInState("WAITING_FOR_INPUT"));
|
|
|
|
render(transcript());
|
|
|
|
const area = await screen.findByTestId("chat-input-area");
|
|
expect(area.className).toContain("bg-background");
|
|
expect(area.className).not.toContain("bg-card");
|
|
});
|
|
});
|