Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
73 changes: 73 additions & 0 deletions packages/app/e2e/session/session-composer-dock.spec.ts
Original file line number Diff line number Diff line change
Expand Up @@ -738,6 +738,79 @@ test("blocked question flow supports escape dismiss", async ({ page, llm, projec
)
})

test("blocked question dock disables controls while response is pending", async ({ page, llm, project }) => {
await project.open()
await withDockSession(
project.sdk,
"e2e composer dock question pending disabled",
async (session) => {
await withDockSeed(project.sdk, session.id, async () => {
await project.gotoSession(session.id)

let responseCalls = 0
let releaseResponse: (() => void) | undefined
const responseReleased = new Promise<void>((resolve) => {
releaseResponse = resolve
})
const respondRoute = async (route: any) => {
responseCalls += 1
await responseReleased
await route
.fulfill({ status: 200, contentType: "application/json", body: JSON.stringify(true) })
.catch(() => undefined)
}

try {
const questions = defaultQuestions
await llm.toolMatch(inputMatch({ questions }), "question", { questions })
await seedSessionQuestion(project.sdk, {
sessionID: session.id,
questions,
})

const dock = page.locator(questionDockSelector)
await expectQuestionBlocked(page)

await page.route("**/session/*/tool/respond", respondRoute)

const customOption = dock.locator('[data-slot="question-option"][data-custom="true"]')
await customOption.click()

const customInput = dock.locator('[data-slot="question-custom-input"]')
const firstOption = dock.locator('[data-slot="question-option"]').first()
const submit = dock.getByRole("button", { name: /submit/i })
await expect(customInput).toBeVisible()

await dock.evaluate((el) => {
el.dispatchEvent(new KeyboardEvent("keydown", { key: "Escape", bubbles: true, cancelable: true }))
})
await expect.poll(() => responseCalls, { timeout: 10_000 }).toBe(1)

await expect(customInput).toBeDisabled()
await expect(firstOption).toBeDisabled()
await expect(submit).toBeDisabled()

const duplicateResponse = page
.waitForRequest((request) => request.url().includes("/tool/respond"), { timeout: 1_000 })
.then(() => true)
.catch(() => false)
await dock.evaluate((el) => {
el.dispatchEvent(new KeyboardEvent("keydown", { key: "Escape", bubbles: true, cancelable: true }))
})
await submit.evaluate((button: HTMLButtonElement) => button.click())
await firstOption.evaluate((button: HTMLButtonElement) => button.click())
expect(await duplicateResponse).toBe(false)
expect(responseCalls).toBe(1)
} finally {
releaseResponse?.()
await page.unroute("**/session/*/tool/respond", respondRoute)
}
})
},
{ trackSession: project.trackSession },
)
})

test("blocked permission flow supports allow once", async ({ page, project }) => {
await project.open()
await withDockSession(
Expand Down
1 change: 1 addition & 0 deletions packages/app/src/i18n/en.ts
Original file line number Diff line number Diff line change
Expand Up @@ -645,6 +645,7 @@ export const dict = {
"session.question.error.staleSession": "Session restarted; resend your last message to try again.",
"session.question.error.alreadyAnswered": "This question was already answered elsewhere.",
"session.question.error.invalidPayload": "Your answer did not match the question; please try again.",
"session.question.error.unknown": "Could not submit your answer. Please try again.",
"session.followupDock.summary.one": "{{count}} queued message",
"session.followupDock.summary.other": "{{count}} queued messages",
"session.followupDock.sendNow": "Send now",
Expand Down
1 change: 1 addition & 0 deletions packages/app/src/i18n/zh.ts
Original file line number Diff line number Diff line change
Expand Up @@ -1049,6 +1049,7 @@ export const dict = {
"session.question.error.staleSession": "会话已重启,请重新发送上一条消息",
"session.question.error.alreadyAnswered": "已在另一处回答这个问题",
"session.question.error.invalidPayload": "回答不符合问题要求,请重试",
"session.question.error.unknown": "未能提交回答,请重试",
"session.header.open.finder": "访达",
"session.header.open.fileExplorer": "文件资源管理器",
"session.header.open.fileManager": "文件管理器",
Expand Down
138 changes: 132 additions & 6 deletions packages/app/src/pages/session/composer/session-question-dock.test.ts
Original file line number Diff line number Diff line change
@@ -1,39 +1,165 @@
import { describe, expect, test } from "bun:test"
import { resolveSkipAction } from "./session-question-dock"
import { runBrowserCheck } from "@/testing/browser-subprocess"
import {
createQuestionResponseGuard,
isSameQuestionRequest,
normalizeToolRespondError,
resolveSkipAction,
} from "./session-question-dock"

describe("resolveSkipAction", () => {
test("navigates to next unsettled question when one exists after current", () => {
// 3 questions: Q0 settled, Q1 unsettled, Q2 (current) just skipped → now settled
const isSettled = (i: number) => i !== 1
const result = resolveSkipAction(2, isSettled, 3)
expect(result).toEqual({ type: "navigate", tab: 1 })
})

test("navigates to first unsettled overall when nothing after current", () => {
// 3 questions: Q0 unsettled, Q1 settled, Q2 (current) just skipped → settled
const isSettled = (i: number) => i !== 0
const result = resolveSkipAction(2, isSettled, 3)
expect(result).toEqual({ type: "navigate", tab: 0 })
})

test("submits when there is only one question and it was just skipped", () => {
// Single question: Q0 just skipped → settled
const isSettled = () => true
const result = resolveSkipAction(0, isSettled, 1)
expect(result).toEqual({ type: "submit" })
})

test("submits when all questions are settled after skipping the last one", () => {
// 3 questions: all settled (Q0 and Q1 answered, Q2 just skipped)
const isSettled = () => true
const result = resolveSkipAction(2, isSettled, 3)
expect(result).toEqual({ type: "submit" })
})

test("navigates to next unsettled before current when current is not the last", () => {
// 3 questions: Q0 settled, Q1 (current) just skipped → settled, Q2 unsettled
const isSettled = (i: number) => i !== 2
const result = resolveSkipAction(1, isSettled, 3)
expect(result).toEqual({ type: "navigate", tab: 2 })
})
})

describe("normalizeToolRespondError", () => {
test("normalizes plain already_resolved objects without exposing [object Object]", () => {
const result = normalizeToolRespondError({ error: "already_resolved" })

expect(result).toEqual({ type: "already_resolved", requestID: undefined })
expect(JSON.stringify(result)).not.toContain("[object Object]")
})

test("keeps answer_count_mismatch details readable for 422 responses", () => {
const result = normalizeToolRespondError({
response: { status: 422 },
error: "answer_count_mismatch",
details: { expected: 2, received: 1 },
})

expect(result).toEqual({
type: "invalid_payload",
detail: 'answer_count_mismatch {"expected":2,"received":1}',
})
expect(JSON.stringify(result)).not.toContain("[object Object]")
})

test("maps plain no_pending_tool_call body errors to stale session", () => {
expect(normalizeToolRespondError({ error: "no_pending_tool_call" })).toEqual({ type: "stale_session" })
})

test("keeps plain answer_count_mismatch body details readable without a status", () => {
const result = normalizeToolRespondError({
error: "answer_count_mismatch",
details: { expected: 2, received: 1 },
})

expect(result).toEqual({
type: "invalid_payload",
detail: 'answer_count_mismatch {"expected":2,"received":1}',
})
expect(JSON.stringify(result)).not.toContain("[object Object]")
})

test("supports common error shapes without stringifying unknown objects", () => {
expect(normalizeToolRespondError(new Error("network failed"))).toEqual({
type: "unknown",
detail: "network failed",
})
expect(normalizeToolRespondError("offline")).toEqual({ type: "unknown", detail: "offline" })
expect(normalizeToolRespondError({ status: 404 })).toEqual({ type: "stale_session" })
expect(normalizeToolRespondError({ statusCode: 409, request: { id: "req_1" } })).toEqual({
type: "already_resolved",
requestID: "req_1",
})
expect(normalizeToolRespondError({ nested: true })).toEqual({ type: "unknown" })
})
})

describe("question response local completion guard", () => {
const request = { id: "req_1", sessionID: "ses_1", messageID: "msg_1", callID: "call_1" }

test("does not treat already_resolved as completion without a same-request local submit", () => {
expect(isSameQuestionRequest(undefined, request, "req_1")).toBe(false)
expect(isSameQuestionRequest({ ...request, id: "req_other" }, request, "req_1")).toBe(false)
expect(isSameQuestionRequest({ ...request, callID: "call_other" }, request, "req_1")).toBe(false)
})

test("treats already_resolved as idempotent only for the same local request", () => {
expect(isSameQuestionRequest(request, request, "req_1")).toBe(true)
expect(isSameQuestionRequest(request, request)).toBe(true)
})
})

describe("question response duplicate submission guard", () => {
test("blocks repeated response attempts while pending and while waiting for sync close", () => {
const guard = createQuestionResponseGuard("req_1")

expect(guard.begin("req_1")).toBe(true)
expect(guard.begin("req_1")).toBe(false)

guard.confirm("req_1")

expect(guard.canInteract("req_1")).toBe(false)
expect(guard.begin("req_1")).toBe(false)
})

test("restores interaction after failed submit or a new request", () => {
const guard = createQuestionResponseGuard("req_1")

expect(guard.begin("req_1")).toBe(true)
guard.fail("req_1")
expect(guard.begin("req_1")).toBe(true)

guard.confirm("req_1")

expect(guard.canInteract("req_1")).toBe(false)
expect(guard.canInteract("req_2")).toBe(true)
expect(guard.begin("req_2")).toBe(true)
})

test("updates reactive disabled state immediately after begin and confirm", () => {
runBrowserCheck(String.raw`
import { createMemo, createRoot } from "solid-js"
import { createQuestionResponseGuard } from "./src/pages/session/composer/session-question-dock.tsx"

const assert = (condition, message) => {
if (!condition) throw new Error(message)
}

createRoot((dispose) => {
const guard = createQuestionResponseGuard("req_1")
const disabled = createMemo(() => !guard.canInteract("req_1"))

assert(disabled() === false, "initial request should be interactive")
assert(guard.begin("req_1") === true, "first begin should submit")
assert(disabled() === true, "begin should immediately disable reactive UI")

guard.fail("req_1")
assert(disabled() === false, "failure should immediately restore interaction")
assert(guard.begin("req_1") === true, "retry should submit after failure")
guard.confirm("req_1")
assert(disabled() === true, "confirmed response should stay disabled while waiting for sync close")

dispose()
})
`)
})
})
Loading