// Production CSS is part of the behavior under test because row height depends on it. import "../index.css"; import { ORCHESTRATION_WS_METHODS, type MessageId, type OrchestrationReadModel, type ProjectId, type ServerConfig, type ThreadId, type WsWelcomePayload, WS_CHANNELS, WS_METHODS, OrchestrationSessionStatus, } from "@t3tools/contracts"; import { RouterProvider, createMemoryHistory } from "@tanstack/react-router"; import { HttpResponse, http, ws } from "msw"; import { setupWorker } from "msw/browser"; import { page } from "vitest/browser"; import { afterAll, afterEach, beforeAll, beforeEach, describe, expect, it, vi } from "vitest"; import { render } from "vitest-browser-react"; import { useComposerDraftStore } from "../composerDraftStore"; import { INLINE_TERMINAL_CONTEXT_PLACEHOLDER, type TerminalContextDraft, } from "../lib/terminalContext"; import { isMacPlatform } from "../lib/utils"; import { getRouter } from "../router"; import { useStore } from "../store"; import { estimateTimelineMessageHeight } from "./timelineHeight"; const THREAD_ID = "thread-browser-test" as ThreadId; const UUID_ROUTE_RE = /^\/[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/; const PROJECT_ID = "project-1" as ProjectId; const NOW_ISO = "2026-03-04T12:00:00.000Z"; const BASE_TIME_MS = Date.parse(NOW_ISO); const ATTACHMENT_SVG = ""; interface WsRequestEnvelope { id: string; body: { _tag: string; [key: string]: unknown; }; } interface TestFixture { snapshot: OrchestrationReadModel; serverConfig: ServerConfig; welcome: WsWelcomePayload; } let fixture: TestFixture; const wsRequests: WsRequestEnvelope["body"][] = []; const wsLink = ws.link(/ws(s)?:\/\/.*/); interface ViewportSpec { name: string; width: number; height: number; textTolerancePx: number; attachmentTolerancePx: number; } const DEFAULT_VIEWPORT: ViewportSpec = { name: "desktop", width: 960, height: 1_100, textTolerancePx: 44, attachmentTolerancePx: 56, }; const TEXT_VIEWPORT_MATRIX = [ DEFAULT_VIEWPORT, { name: "tablet", width: 720, height: 1_024, textTolerancePx: 44, attachmentTolerancePx: 56 }, { name: "mobile", width: 430, height: 932, textTolerancePx: 56, attachmentTolerancePx: 56 }, { name: "narrow", width: 320, height: 700, textTolerancePx: 84, attachmentTolerancePx: 56 }, ] as const satisfies readonly ViewportSpec[]; const ATTACHMENT_VIEWPORT_MATRIX = [ DEFAULT_VIEWPORT, { name: "mobile", width: 430, height: 932, textTolerancePx: 56, attachmentTolerancePx: 56 }, { name: "narrow", width: 320, height: 700, textTolerancePx: 84, attachmentTolerancePx: 56 }, ] as const satisfies readonly ViewportSpec[]; interface UserRowMeasurement { measuredRowHeightPx: number; timelineWidthMeasuredPx: number; renderedInVirtualizedRegion: boolean; } interface MountedChatView { cleanup: () => Promise; measureUserRow: (targetMessageId: MessageId) => Promise; setViewport: (viewport: ViewportSpec) => Promise; router: ReturnType; } function isoAt(offsetSeconds: number): string { return new Date(BASE_TIME_MS + offsetSeconds * 1_000).toISOString(); } function createBaseServerConfig(): ServerConfig { return { cwd: "/repo/project", keybindingsConfigPath: "/repo/project/.t3code-keybindings.json", keybindings: [], issues: [], providers: [ { provider: "codex", status: "ready", available: true, authStatus: "authenticated", checkedAt: NOW_ISO, }, ], availableEditors: [], }; } function createUserMessage(options: { id: MessageId; text: string; offsetSeconds: number; attachments?: Array<{ type: "image"; id: string; name: string; mimeType: string; sizeBytes: number; }>; }) { return { id: options.id, role: "user" as const, text: options.text, ...(options.attachments ? { attachments: options.attachments } : {}), turnId: null, streaming: false, createdAt: isoAt(options.offsetSeconds), updatedAt: isoAt(options.offsetSeconds + 1), }; } function createAssistantMessage(options: { id: MessageId; text: string; offsetSeconds: number }) { return { id: options.id, role: "assistant" as const, text: options.text, turnId: null, streaming: false, createdAt: isoAt(options.offsetSeconds), updatedAt: isoAt(options.offsetSeconds + 1), }; } function createTerminalContext(input: { id: string; terminalLabel: string; lineStart: number; lineEnd: number; text: string; }): TerminalContextDraft { return { id: input.id, threadId: THREAD_ID, terminalId: `terminal-${input.id}`, terminalLabel: input.terminalLabel, lineStart: input.lineStart, lineEnd: input.lineEnd, text: input.text, createdAt: NOW_ISO, }; } function createSnapshotForTargetUser(options: { targetMessageId: MessageId; targetText: string; targetAttachmentCount?: number; sessionStatus?: OrchestrationSessionStatus; }): OrchestrationReadModel { const messages: Array = []; for (let index = 0; index < 22; index += 1) { const isTarget = index === 3; const userId = `msg-user-${index}` as MessageId; const assistantId = `msg-assistant-${index}` as MessageId; const attachments = isTarget && (options.targetAttachmentCount ?? 0) > 0 ? Array.from({ length: options.targetAttachmentCount ?? 0 }, (_, attachmentIndex) => ({ type: "image" as const, id: `attachment-${attachmentIndex + 1}`, name: `attachment-${attachmentIndex + 1}.png`, mimeType: "image/png", sizeBytes: 128, })) : undefined; messages.push( createUserMessage({ id: isTarget ? options.targetMessageId : userId, text: isTarget ? options.targetText : `filler user message ${index}`, offsetSeconds: messages.length * 3, ...(attachments ? { attachments } : {}), }), ); messages.push( createAssistantMessage({ id: assistantId, text: `assistant filler ${index}`, offsetSeconds: messages.length * 3, }), ); } return { snapshotSequence: 1, projects: [ { id: PROJECT_ID, title: "Project", workspaceRoot: "/repo/project", defaultModel: "gpt-5", scripts: [], createdAt: NOW_ISO, updatedAt: NOW_ISO, deletedAt: null, }, ], threads: [ { id: THREAD_ID, projectId: PROJECT_ID, title: "Browser test thread", model: "gpt-5", interactionMode: "default", runtimeMode: "full-access", branch: "main", worktreePath: null, latestTurn: null, createdAt: NOW_ISO, updatedAt: NOW_ISO, deletedAt: null, messages, activities: [], proposedPlans: [], checkpoints: [], session: { threadId: THREAD_ID, status: options.sessionStatus ?? "ready", providerName: "codex", runtimeMode: "full-access", activeTurnId: null, lastError: null, updatedAt: NOW_ISO, }, }, ], updatedAt: NOW_ISO, }; } function buildFixture(snapshot: OrchestrationReadModel): TestFixture { return { snapshot, serverConfig: createBaseServerConfig(), welcome: { cwd: "/repo/project", projectName: "Project", bootstrapProjectId: PROJECT_ID, bootstrapThreadId: THREAD_ID, }, }; } function addThreadToSnapshot( snapshot: OrchestrationReadModel, threadId: ThreadId, ): OrchestrationReadModel { return { ...snapshot, snapshotSequence: snapshot.snapshotSequence + 1, threads: [ ...snapshot.threads, { id: threadId, projectId: PROJECT_ID, title: "New thread", model: "gpt-5", interactionMode: "default", runtimeMode: "full-access", branch: "main", worktreePath: null, latestTurn: null, createdAt: NOW_ISO, updatedAt: NOW_ISO, deletedAt: null, messages: [], activities: [], proposedPlans: [], checkpoints: [], session: { threadId, status: "ready", providerName: "codex", runtimeMode: "full-access", activeTurnId: null, lastError: null, updatedAt: NOW_ISO, }, }, ], }; } function createDraftOnlySnapshot(): OrchestrationReadModel { const snapshot = createSnapshotForTargetUser({ targetMessageId: "msg-user-draft-target" as MessageId, targetText: "draft thread", }); return { ...snapshot, threads: [], }; } function createSnapshotWithLongProposedPlan(): OrchestrationReadModel { const snapshot = createSnapshotForTargetUser({ targetMessageId: "msg-user-plan-target" as MessageId, targetText: "plan thread", }); const planMarkdown = [ "# Ship plan mode follow-up", "", "- Step 1: capture the thread-open trace", "- Step 2: identify the main-thread bottleneck", "- Step 3: keep collapsed cards cheap", "- Step 4: render the full markdown only on demand", "- Step 5: preserve export and save actions", "- Step 6: add regression coverage", "- Step 7: verify route transitions stay responsive", "- Step 8: confirm no server-side work changed", "- Step 9: confirm short plans still render normally", "- Step 10: confirm long plans stay collapsed by default", "- Step 11: confirm preview text is still useful", "- Step 12: confirm plan follow-up flow still works", "- Step 13: confirm timeline virtualization still behaves", "- Step 14: confirm theme styling still looks correct", "- Step 15: confirm save dialog behavior is unchanged", "- Step 16: confirm download behavior is unchanged", "- Step 17: confirm code fences do not parse until expand", "- Step 18: confirm preview truncation ends cleanly", "- Step 19: confirm markdown links still open in editor after expand", "- Step 20: confirm deep hidden detail only appears after expand", "", "```ts", "export const hiddenPlanImplementationDetail = 'deep hidden detail only after expand';", "```", ].join("\n"); return { ...snapshot, threads: snapshot.threads.map((thread) => thread.id === THREAD_ID ? Object.assign({}, thread, { proposedPlans: [ { id: "plan-browser-test", turnId: null, planMarkdown, implementedAt: null, implementationThreadId: null, createdAt: isoAt(1_000), updatedAt: isoAt(1_001), }, ], updatedAt: isoAt(1_001), }) : thread, ), }; } function resolveWsRpc(body: WsRequestEnvelope["body"]): unknown { const tag = body._tag; if (tag === ORCHESTRATION_WS_METHODS.getSnapshot) { return fixture.snapshot; } if (tag === WS_METHODS.serverGetConfig) { return fixture.serverConfig; } if (tag === WS_METHODS.gitListBranches) { return { isRepo: true, hasOriginRemote: true, branches: [ { name: "main", current: true, isDefault: true, worktreePath: null, }, ], }; } if (tag === WS_METHODS.gitStatus) { return { branch: "main", hasWorkingTreeChanges: false, workingTree: { files: [], insertions: 0, deletions: 0, }, hasUpstream: true, aheadCount: 0, behindCount: 0, pr: null, }; } if (tag === WS_METHODS.projectsSearchEntries) { return { entries: [], truncated: false, }; } if (tag === WS_METHODS.terminalOpen) { return { threadId: typeof body.threadId === "string" ? body.threadId : THREAD_ID, terminalId: typeof body.terminalId === "string" ? body.terminalId : "default", cwd: typeof body.cwd === "string" ? body.cwd : "/repo/project", status: "running", pid: 123, history: "", exitCode: null, exitSignal: null, updatedAt: NOW_ISO, }; } return {}; } const worker = setupWorker( wsLink.addEventListener("connection", ({ client }) => { client.send( JSON.stringify({ type: "push", sequence: 1, channel: WS_CHANNELS.serverWelcome, data: fixture.welcome, }), ); client.addEventListener("message", (event) => { const rawData = event.data; if (typeof rawData !== "string") return; let request: WsRequestEnvelope; try { request = JSON.parse(rawData) as WsRequestEnvelope; } catch { return; } const method = request.body?._tag; if (typeof method !== "string") return; wsRequests.push(request.body); client.send( JSON.stringify({ id: request.id, result: resolveWsRpc(request.body), }), ); }); }), http.get("*/attachments/:attachmentId", () => HttpResponse.text(ATTACHMENT_SVG, { headers: { "Content-Type": "image/svg+xml", }, }), ), http.get("*/api/project-favicon", () => new HttpResponse(null, { status: 204 })), ); async function nextFrame(): Promise { await new Promise((resolve) => { window.requestAnimationFrame(() => resolve()); }); } async function waitForLayout(): Promise { await nextFrame(); await nextFrame(); await nextFrame(); } async function setViewport(viewport: ViewportSpec): Promise { await page.viewport(viewport.width, viewport.height); await waitForLayout(); } async function waitForProductionStyles(): Promise { await vi.waitFor( () => { expect( getComputedStyle(document.documentElement).getPropertyValue("--background").trim(), ).not.toBe(""); expect(getComputedStyle(document.body).marginTop).toBe("0px"); }, { timeout: 4_000, interval: 16, }, ); } async function waitForElement( query: () => T | null, errorMessage: string, ): Promise { let element: T | null = null; await vi.waitFor( () => { element = query(); expect(element, errorMessage).toBeTruthy(); }, { timeout: 8_000, interval: 16, }, ); if (!element) { throw new Error(errorMessage); } return element; } async function waitForURL( router: ReturnType, predicate: (pathname: string) => boolean, errorMessage: string, ): Promise { let pathname = ""; await vi.waitFor( () => { pathname = router.state.location.pathname; expect(predicate(pathname), errorMessage).toBe(true); }, { timeout: 8_000, interval: 16 }, ); return pathname; } async function waitForComposerEditor(): Promise { return waitForElement( () => document.querySelector('[contenteditable="true"]'), "Unable to find composer editor.", ); } async function waitForSendButton(): Promise { return waitForElement( () => document.querySelector('button[aria-label="Send message"]'), "Unable to find send button.", ); } async function waitForInteractionModeButton( expectedLabel: "Chat" | "Plan", ): Promise { return waitForElement( () => Array.from(document.querySelectorAll("button")).find( (button) => button.textContent?.trim() === expectedLabel, ) as HTMLButtonElement | null, `Unable to find ${expectedLabel} interaction mode button.`, ); } async function waitForImagesToLoad(scope: ParentNode): Promise { const images = Array.from(scope.querySelectorAll("img")); if (images.length === 0) { return; } await Promise.all( images.map( (image) => new Promise((resolve) => { if (image.complete) { resolve(); return; } image.addEventListener("load", () => resolve(), { once: true }); image.addEventListener("error", () => resolve(), { once: true }); }), ), ); await waitForLayout(); } async function measureUserRow(options: { host: HTMLElement; targetMessageId: MessageId; }): Promise { const { host, targetMessageId } = options; const rowSelector = `[data-message-id="${targetMessageId}"][data-message-role="user"]`; const scrollContainer = await waitForElement( () => host.querySelector("div.overflow-y-auto.overscroll-y-contain"), "Unable to find ChatView message scroll container.", ); let row: HTMLElement | null = null; await vi.waitFor( async () => { scrollContainer.scrollTop = 0; scrollContainer.dispatchEvent(new Event("scroll")); await waitForLayout(); row = host.querySelector(rowSelector); expect(row, "Unable to locate targeted user message row.").toBeTruthy(); }, { timeout: 8_000, interval: 16, }, ); await waitForImagesToLoad(row!); scrollContainer.scrollTop = 0; scrollContainer.dispatchEvent(new Event("scroll")); await nextFrame(); const timelineRoot = row!.closest('[data-timeline-root="true"]') ?? host.querySelector('[data-timeline-root="true"]'); if (!(timelineRoot instanceof HTMLElement)) { throw new Error("Unable to locate timeline root container."); } let timelineWidthMeasuredPx = 0; let measuredRowHeightPx = 0; let renderedInVirtualizedRegion = false; await vi.waitFor( async () => { scrollContainer.scrollTop = 0; scrollContainer.dispatchEvent(new Event("scroll")); await nextFrame(); const measuredRow = host.querySelector(rowSelector); expect(measuredRow, "Unable to measure targeted user row height.").toBeTruthy(); timelineWidthMeasuredPx = timelineRoot.getBoundingClientRect().width; measuredRowHeightPx = measuredRow!.getBoundingClientRect().height; renderedInVirtualizedRegion = measuredRow!.closest("[data-index]") instanceof HTMLElement; expect(timelineWidthMeasuredPx, "Unable to measure timeline width.").toBeGreaterThan(0); expect(measuredRowHeightPx, "Unable to measure targeted user row height.").toBeGreaterThan(0); }, { timeout: 4_000, interval: 16, }, ); return { measuredRowHeightPx, timelineWidthMeasuredPx, renderedInVirtualizedRegion }; } async function mountChatView(options: { viewport: ViewportSpec; snapshot: OrchestrationReadModel; configureFixture?: (fixture: TestFixture) => void; }): Promise { fixture = buildFixture(options.snapshot); options.configureFixture?.(fixture); await setViewport(options.viewport); await waitForProductionStyles(); const host = document.createElement("div"); host.style.position = "fixed"; host.style.inset = "0"; host.style.width = "100vw"; host.style.height = "100vh"; host.style.display = "grid"; host.style.overflow = "hidden"; document.body.append(host); const router = getRouter( createMemoryHistory({ initialEntries: [`/${THREAD_ID}`], }), ); const screen = await render(, { container: host, }); await waitForLayout(); return { cleanup: async () => { await screen.unmount(); host.remove(); }, measureUserRow: async (targetMessageId: MessageId) => measureUserRow({ host, targetMessageId }), setViewport: async (viewport: ViewportSpec) => { await setViewport(viewport); await waitForProductionStyles(); }, router, }; } async function measureUserRowAtViewport(options: { snapshot: OrchestrationReadModel; targetMessageId: MessageId; viewport: ViewportSpec; }): Promise { const mounted = await mountChatView({ viewport: options.viewport, snapshot: options.snapshot, }); try { return await mounted.measureUserRow(options.targetMessageId); } finally { await mounted.cleanup(); } } describe("ChatView timeline estimator parity (full app)", () => { beforeAll(async () => { fixture = buildFixture( createSnapshotForTargetUser({ targetMessageId: "msg-user-bootstrap" as MessageId, targetText: "bootstrap", }), ); await worker.start({ onUnhandledRequest: "bypass", quiet: true, serviceWorker: { url: "/mockServiceWorker.js", }, }); }); afterAll(async () => { await worker.stop(); }); beforeEach(async () => { await setViewport(DEFAULT_VIEWPORT); localStorage.clear(); document.body.innerHTML = ""; wsRequests.length = 0; useComposerDraftStore.setState({ draftsByThreadId: {}, draftThreadsByThreadId: {}, projectDraftThreadIdByProjectId: {}, }); useStore.setState({ projects: [], threads: [], threadsHydrated: false, }); }); afterEach(() => { document.body.innerHTML = ""; }); it.each(TEXT_VIEWPORT_MATRIX)( "keeps long user message estimate close at the $name viewport", async (viewport) => { const userText = "x".repeat(3_200); const targetMessageId = `msg-user-target-long-${viewport.name}` as MessageId; const mounted = await mountChatView({ viewport, snapshot: createSnapshotForTargetUser({ targetMessageId, targetText: userText, }), }); try { const { measuredRowHeightPx, timelineWidthMeasuredPx, renderedInVirtualizedRegion } = await mounted.measureUserRow(targetMessageId); expect(renderedInVirtualizedRegion).toBe(true); const estimatedHeightPx = estimateTimelineMessageHeight( { role: "user", text: userText, attachments: [] }, { timelineWidthPx: timelineWidthMeasuredPx }, ); expect(Math.abs(measuredRowHeightPx - estimatedHeightPx)).toBeLessThanOrEqual( viewport.textTolerancePx, ); } finally { await mounted.cleanup(); } }, ); it("tracks wrapping parity while resizing an existing ChatView across the viewport matrix", async () => { const userText = "x".repeat(3_200); const targetMessageId = "msg-user-target-resize" as MessageId; const mounted = await mountChatView({ viewport: TEXT_VIEWPORT_MATRIX[0], snapshot: createSnapshotForTargetUser({ targetMessageId, targetText: userText, }), }); try { const measurements: Array< UserRowMeasurement & { viewport: ViewportSpec; estimatedHeightPx: number } > = []; for (const viewport of TEXT_VIEWPORT_MATRIX) { await mounted.setViewport(viewport); const measurement = await mounted.measureUserRow(targetMessageId); const estimatedHeightPx = estimateTimelineMessageHeight( { role: "user", text: userText, attachments: [] }, { timelineWidthPx: measurement.timelineWidthMeasuredPx }, ); expect(measurement.renderedInVirtualizedRegion).toBe(true); expect(Math.abs(measurement.measuredRowHeightPx - estimatedHeightPx)).toBeLessThanOrEqual( viewport.textTolerancePx, ); measurements.push({ ...measurement, viewport, estimatedHeightPx }); } expect( new Set(measurements.map((measurement) => Math.round(measurement.timelineWidthMeasuredPx))) .size, ).toBeGreaterThanOrEqual(3); const byMeasuredWidth = measurements.toSorted( (left, right) => left.timelineWidthMeasuredPx - right.timelineWidthMeasuredPx, ); const narrowest = byMeasuredWidth[0]!; const widest = byMeasuredWidth.at(-1)!; expect(narrowest.timelineWidthMeasuredPx).toBeLessThan(widest.timelineWidthMeasuredPx); expect(narrowest.measuredRowHeightPx).toBeGreaterThan(widest.measuredRowHeightPx); expect(narrowest.estimatedHeightPx).toBeGreaterThan(widest.estimatedHeightPx); } finally { await mounted.cleanup(); } }); it("tracks additional rendered wrapping when ChatView width narrows between desktop and mobile viewports", async () => { const userText = "x".repeat(2_400); const targetMessageId = "msg-user-target-wrap" as MessageId; const snapshot = createSnapshotForTargetUser({ targetMessageId, targetText: userText, }); const desktopMeasurement = await measureUserRowAtViewport({ viewport: TEXT_VIEWPORT_MATRIX[0], snapshot, targetMessageId, }); const mobileMeasurement = await measureUserRowAtViewport({ viewport: TEXT_VIEWPORT_MATRIX[2], snapshot, targetMessageId, }); const estimatedDesktopPx = estimateTimelineMessageHeight( { role: "user", text: userText, attachments: [] }, { timelineWidthPx: desktopMeasurement.timelineWidthMeasuredPx }, ); const estimatedMobilePx = estimateTimelineMessageHeight( { role: "user", text: userText, attachments: [] }, { timelineWidthPx: mobileMeasurement.timelineWidthMeasuredPx }, ); const measuredDeltaPx = mobileMeasurement.measuredRowHeightPx - desktopMeasurement.measuredRowHeightPx; const estimatedDeltaPx = estimatedMobilePx - estimatedDesktopPx; expect(measuredDeltaPx).toBeGreaterThan(0); expect(estimatedDeltaPx).toBeGreaterThan(0); const ratio = estimatedDeltaPx / measuredDeltaPx; expect(ratio).toBeGreaterThan(0.65); expect(ratio).toBeLessThan(1.35); }); it.each(ATTACHMENT_VIEWPORT_MATRIX)( "keeps user attachment estimate close at the $name viewport", async (viewport) => { const targetMessageId = `msg-user-target-attachments-${viewport.name}` as MessageId; const userText = "message with image attachments"; const mounted = await mountChatView({ viewport, snapshot: createSnapshotForTargetUser({ targetMessageId, targetText: userText, targetAttachmentCount: 3, }), }); try { const { measuredRowHeightPx, timelineWidthMeasuredPx, renderedInVirtualizedRegion } = await mounted.measureUserRow(targetMessageId); expect(renderedInVirtualizedRegion).toBe(true); const estimatedHeightPx = estimateTimelineMessageHeight( { role: "user", text: userText, attachments: [{ id: "attachment-1" }, { id: "attachment-2" }, { id: "attachment-3" }], }, { timelineWidthPx: timelineWidthMeasuredPx }, ); expect(Math.abs(measuredRowHeightPx - estimatedHeightPx)).toBeLessThanOrEqual( viewport.attachmentTolerancePx, ); } finally { await mounted.cleanup(); } }, ); it("opens the project cwd for draft threads without a worktree path", async () => { useComposerDraftStore.setState({ draftThreadsByThreadId: { [THREAD_ID]: { projectId: PROJECT_ID, createdAt: NOW_ISO, runtimeMode: "full-access", interactionMode: "default", branch: null, worktreePath: null, envMode: "local", }, }, projectDraftThreadIdByProjectId: { [PROJECT_ID]: THREAD_ID, }, }); const mounted = await mountChatView({ viewport: DEFAULT_VIEWPORT, snapshot: createDraftOnlySnapshot(), configureFixture: (nextFixture) => { nextFixture.serverConfig = { ...nextFixture.serverConfig, availableEditors: ["vscode"], }; }, }); try { const openButton = await waitForElement( () => Array.from(document.querySelectorAll("button")).find( (button) => button.textContent?.trim() === "Open", ) as HTMLButtonElement | null, "Unable to find Open button.", ); openButton.click(); await vi.waitFor( () => { const openRequest = wsRequests.find( (request) => request._tag === WS_METHODS.shellOpenInEditor, ); expect(openRequest).toMatchObject({ _tag: WS_METHODS.shellOpenInEditor, cwd: "/repo/project", editor: "vscode", }); }, { timeout: 8_000, interval: 16 }, ); } finally { await mounted.cleanup(); } }); it("toggles plan mode with Shift+Tab only while the composer is focused", async () => { const mounted = await mountChatView({ viewport: DEFAULT_VIEWPORT, snapshot: createSnapshotForTargetUser({ targetMessageId: "msg-user-target-hotkey" as MessageId, targetText: "hotkey target", }), }); try { const initialModeButton = await waitForInteractionModeButton("Chat"); expect(initialModeButton.title).toContain("enter plan mode"); window.dispatchEvent( new KeyboardEvent("keydown", { key: "Tab", shiftKey: true, bubbles: true, cancelable: true, }), ); await waitForLayout(); expect((await waitForInteractionModeButton("Chat")).title).toContain("enter plan mode"); const composerEditor = await waitForComposerEditor(); composerEditor.focus(); composerEditor.dispatchEvent( new KeyboardEvent("keydown", { key: "Tab", shiftKey: true, bubbles: true, cancelable: true, }), ); await vi.waitFor( async () => { expect((await waitForInteractionModeButton("Plan")).title).toContain( "return to normal chat mode", ); }, { timeout: 8_000, interval: 16 }, ); composerEditor.dispatchEvent( new KeyboardEvent("keydown", { key: "Tab", shiftKey: true, bubbles: true, cancelable: true, }), ); await vi.waitFor( async () => { expect((await waitForInteractionModeButton("Chat")).title).toContain("enter plan mode"); }, { timeout: 8_000, interval: 16 }, ); } finally { await mounted.cleanup(); } }); it("keeps backspaced terminal context pills removed when a new one is added", async () => { const removedLabel = "Terminal 1 lines 1-2"; const addedLabel = "Terminal 2 lines 9-10"; useComposerDraftStore.getState().addTerminalContext( THREAD_ID, createTerminalContext({ id: "ctx-removed", terminalLabel: "Terminal 1", lineStart: 1, lineEnd: 2, text: "bun i\nno changes", }), ); const mounted = await mountChatView({ viewport: DEFAULT_VIEWPORT, snapshot: createSnapshotForTargetUser({ targetMessageId: "msg-user-terminal-pill-backspace" as MessageId, targetText: "terminal pill backspace target", }), }); try { await vi.waitFor( () => { expect(document.body.textContent).toContain(removedLabel); }, { timeout: 8_000, interval: 16 }, ); const composerEditor = await waitForComposerEditor(); composerEditor.focus(); composerEditor.dispatchEvent( new KeyboardEvent("keydown", { key: "Backspace", bubbles: true, cancelable: true, }), ); await vi.waitFor( () => { expect(useComposerDraftStore.getState().draftsByThreadId[THREAD_ID]).toBeUndefined(); expect(document.body.textContent).not.toContain(removedLabel); }, { timeout: 8_000, interval: 16 }, ); useComposerDraftStore.getState().addTerminalContext( THREAD_ID, createTerminalContext({ id: "ctx-added", terminalLabel: "Terminal 2", lineStart: 9, lineEnd: 10, text: "git status\nOn branch main", }), ); await vi.waitFor( () => { const draft = useComposerDraftStore.getState().draftsByThreadId[THREAD_ID]; expect(draft?.terminalContexts.map((context) => context.id)).toEqual(["ctx-added"]); expect(document.body.textContent).toContain(addedLabel); expect(document.body.textContent).not.toContain(removedLabel); }, { timeout: 8_000, interval: 16 }, ); } finally { await mounted.cleanup(); } }); it("disables send when the composer only contains an expired terminal pill", async () => { const expiredLabel = "Terminal 1 line 4"; useComposerDraftStore.getState().addTerminalContext( THREAD_ID, createTerminalContext({ id: "ctx-expired-only", terminalLabel: "Terminal 1", lineStart: 4, lineEnd: 4, text: "", }), ); const mounted = await mountChatView({ viewport: DEFAULT_VIEWPORT, snapshot: createSnapshotForTargetUser({ targetMessageId: "msg-user-expired-pill-disabled" as MessageId, targetText: "expired pill disabled target", }), }); try { await vi.waitFor( () => { expect(document.body.textContent).toContain(expiredLabel); }, { timeout: 8_000, interval: 16 }, ); const sendButton = await waitForSendButton(); expect(sendButton.disabled).toBe(true); } finally { await mounted.cleanup(); } }); it("warns when sending text while omitting expired terminal pills", async () => { const expiredLabel = "Terminal 1 line 4"; useComposerDraftStore.getState().addTerminalContext( THREAD_ID, createTerminalContext({ id: "ctx-expired-send-warning", terminalLabel: "Terminal 1", lineStart: 4, lineEnd: 4, text: "", }), ); useComposerDraftStore .getState() .setPrompt(THREAD_ID, `yoo${INLINE_TERMINAL_CONTEXT_PLACEHOLDER}waddup`); const mounted = await mountChatView({ viewport: DEFAULT_VIEWPORT, snapshot: createSnapshotForTargetUser({ targetMessageId: "msg-user-expired-pill-warning" as MessageId, targetText: "expired pill warning target", }), }); try { await vi.waitFor( () => { expect(document.body.textContent).toContain(expiredLabel); }, { timeout: 8_000, interval: 16 }, ); const sendButton = await waitForSendButton(); expect(sendButton.disabled).toBe(false); sendButton.click(); await vi.waitFor( () => { expect(document.body.textContent).toContain( "Expired terminal context omitted from message", ); expect(document.body.textContent).not.toContain(expiredLabel); expect(document.body.textContent).toContain("yoowaddup"); }, { timeout: 8_000, interval: 16 }, ); } finally { await mounted.cleanup(); } }); it("shows a pointer cursor for the running stop button", async () => { const mounted = await mountChatView({ viewport: DEFAULT_VIEWPORT, snapshot: createSnapshotForTargetUser({ targetMessageId: "msg-user-stop-button-cursor" as MessageId, targetText: "stop button cursor target", sessionStatus: "running", }), }); try { const stopButton = await waitForElement( () => document.querySelector('button[aria-label="Stop generation"]'), "Unable to find stop generation button.", ); expect(getComputedStyle(stopButton).cursor).toBe("pointer"); } finally { await mounted.cleanup(); } }); it("keeps the new thread selected after clicking the new-thread button", async () => { const mounted = await mountChatView({ viewport: DEFAULT_VIEWPORT, snapshot: createSnapshotForTargetUser({ targetMessageId: "msg-user-new-thread-test" as MessageId, targetText: "new thread selection test", }), }); try { // Wait for the sidebar to render with the project. const newThreadButton = page.getByTestId("new-thread-button"); await expect.element(newThreadButton).toBeInTheDocument(); await newThreadButton.click(); // The route should change to a new draft thread ID. const newThreadPath = await waitForURL( mounted.router, (path) => UUID_ROUTE_RE.test(path), "Route should have changed to a new draft thread UUID.", ); const newThreadId = newThreadPath.slice(1) as ThreadId; // The composer editor should be present for the new draft thread. await waitForComposerEditor(); // Simulate the snapshot sync arriving from the server after the draft // thread has been promoted to a server thread (thread.create + turn.start // succeeded). The snapshot now includes the new thread, and the sync // should clear the draft without disrupting the route. const { syncServerReadModel } = useStore.getState(); syncServerReadModel(addThreadToSnapshot(fixture.snapshot, newThreadId)); // Clear the draft now that the server thread exists (mirrors EventRouter behavior). useComposerDraftStore.getState().clearDraftThread(newThreadId); // The route should still be on the new thread — not redirected away. await waitForURL( mounted.router, (path) => path === newThreadPath, "New thread should remain selected after snapshot sync clears the draft.", ); // The empty thread view and composer should still be visible. await expect .element(page.getByText("Send a message to start the conversation.")) .toBeInTheDocument(); await expect.element(page.getByTestId("composer-editor")).toBeInTheDocument(); } finally { await mounted.cleanup(); } }); it("creates a new thread from the global chat.new shortcut", async () => { const mounted = await mountChatView({ viewport: DEFAULT_VIEWPORT, snapshot: createSnapshotForTargetUser({ targetMessageId: "msg-user-chat-shortcut-test" as MessageId, targetText: "chat shortcut test", }), configureFixture: (nextFixture) => { nextFixture.serverConfig = { ...nextFixture.serverConfig, keybindings: [ { command: "chat.new", shortcut: { key: "o", metaKey: false, ctrlKey: false, shiftKey: true, altKey: false, modKey: true, }, whenAst: { type: "not", node: { type: "identifier", name: "terminalFocus" }, }, }, ], }; }, }); try { const useMetaForMod = isMacPlatform(navigator.platform); window.dispatchEvent( new KeyboardEvent("keydown", { key: "o", shiftKey: true, metaKey: useMetaForMod, ctrlKey: !useMetaForMod, bubbles: true, cancelable: true, }), ); await waitForURL( mounted.router, (path) => UUID_ROUTE_RE.test(path), "Route should have changed to a new draft thread UUID from the shortcut.", ); } finally { await mounted.cleanup(); } }); it("creates a fresh draft after the previous draft thread is promoted", async () => { const mounted = await mountChatView({ viewport: DEFAULT_VIEWPORT, snapshot: createSnapshotForTargetUser({ targetMessageId: "msg-user-promoted-draft-shortcut-test" as MessageId, targetText: "promoted draft shortcut test", }), configureFixture: (nextFixture) => { nextFixture.serverConfig = { ...nextFixture.serverConfig, keybindings: [ { command: "chat.new", shortcut: { key: "o", metaKey: false, ctrlKey: false, shiftKey: true, altKey: false, modKey: true, }, whenAst: { type: "not", node: { type: "identifier", name: "terminalFocus" }, }, }, ], }; }, }); try { const newThreadButton = page.getByTestId("new-thread-button"); await expect.element(newThreadButton).toBeInTheDocument(); await newThreadButton.click(); const promotedThreadPath = await waitForURL( mounted.router, (path) => UUID_ROUTE_RE.test(path), "Route should have changed to a promoted draft thread UUID.", ); const promotedThreadId = promotedThreadPath.slice(1) as ThreadId; const { syncServerReadModel } = useStore.getState(); syncServerReadModel(addThreadToSnapshot(fixture.snapshot, promotedThreadId)); useComposerDraftStore.getState().clearDraftThread(promotedThreadId); const useMetaForMod = isMacPlatform(navigator.platform); window.dispatchEvent( new KeyboardEvent("keydown", { key: "o", shiftKey: true, metaKey: useMetaForMod, ctrlKey: !useMetaForMod, bubbles: true, cancelable: true, }), ); const freshThreadPath = await waitForURL( mounted.router, (path) => UUID_ROUTE_RE.test(path) && path !== promotedThreadPath, "Shortcut should create a fresh draft instead of reusing the promoted thread.", ); expect(freshThreadPath).not.toBe(promotedThreadPath); } finally { await mounted.cleanup(); } }); it("keeps long proposed plans lightweight until the user expands them", async () => { const mounted = await mountChatView({ viewport: DEFAULT_VIEWPORT, snapshot: createSnapshotWithLongProposedPlan(), }); try { await waitForElement( () => Array.from(document.querySelectorAll("button")).find( (button) => button.textContent?.trim() === "Expand plan", ) as HTMLButtonElement | null, "Unable to find Expand plan button.", ); expect(document.body.textContent).not.toContain("deep hidden detail only after expand"); const expandButton = await waitForElement( () => Array.from(document.querySelectorAll("button")).find( (button) => button.textContent?.trim() === "Expand plan", ) as HTMLButtonElement | null, "Unable to find Expand plan button.", ); expandButton.click(); await vi.waitFor( () => { expect(document.body.textContent).toContain("deep hidden detail only after expand"); }, { timeout: 8_000, interval: 16 }, ); } finally { await mounted.cleanup(); } }); });