@allbluecn/web-app 0.12.0 → 0.12.2

package/src/components/harness/__tests__/gallery-behavior.test.tsx

@@ -0,0 +1,171 @@
+import { afterEach, beforeAll, beforeEach, describe, expect, it, vi } from "vitest"
+import { cleanup, fireEvent, render, screen, waitFor, within } from "@testing-library/react"
+import { QueryClient, QueryClientProvider } from "@tanstack/react-query"
+import { createMemoryHistory, createRootRoute, createRoute, createRouter, Outlet, RouterProvider } from "@tanstack/react-router"
+import { GalleryGrid } from "../gallery/gallery-grid"
+import { PreviewPanel } from "../code/preview-panel"
+import { Route as GalleryRoute } from "@/routes/_nav/harness/gallery/route.lazy"
+import { Route as HarnessRoute } from "@/routes/_nav/harness/route.lazy"
+import type { GalleryItem } from "../gallery/gallery-types"
+import { m } from "@/paraglide/messages.js"
+
+const { listGenerationRunsFn, createVariantFn, attachToStoryFn, attachToKbFn, attachPrototypeToPrdFn, listStoriesFn, listKnowledgeFn, listPrdFn } = vi.hoisted(() => ({
+  listGenerationRunsFn: vi.fn<() => Promise<{ runs: GalleryItem[]; nextCursor: string | null }>>(),
+  createVariantFn: vi.fn<() => Promise<{ runId: string; artifacts: unknown[]; images: unknown[] }>>(),
+  attachToStoryFn: vi.fn(),
+  attachToKbFn: vi.fn(),
+  attachPrototypeToPrdFn: vi.fn(),
+  listStoriesFn: vi.fn(),
+  listKnowledgeFn: vi.fn(),
+  listPrdFn: vi.fn(),
+}))
+vi.mock("@/components/billing/feature-gate", () => ({ FeatureGate: ({ children }: { children: React.ReactNode }) => children }))
+vi.mock("@/server/serverFns/harness/gallery", () => ({ listGenerationRunsFn, createVariantFn }))
+vi.mock("@/server/serverFns/harness/attach-to-story", () => ({ attachToStoryFn }))
+vi.mock("@/server/serverFns/harness/attach-to-kb", () => ({ attachToKbFn }))
+vi.mock("@/server/serverFns/harness/attach-prototype", () => ({ attachPrototypeToPrdFn }))
+vi.mock("@/server/serverFns/story/story-crud", () => ({ listStoriesFn }))
+vi.mock("@/server/serverFns/knowledge", () => ({ listKnowledgeFn }))
+vi.mock("@/plugins/core/prd/serverFns/prd", () => ({ listPrdFn }))
+vi.mock("../harness-workbench-page", () => ({ default: () => <div>Workbench</div> }))
+
+const item = (id: string, kind = "image", threadId = "thread-1"): GalleryItem => ({
+  id, kind, threadId, status: "completed", model: "test-model", provider: "test-provider",
+  startedAt: new Date("2026-09-17T00:00:00Z"), result: { prompt: `prompt-${id}` }, usage: null,
+  artifacts: [{ artifactId: id, runId: id, threadId, name: id, mimeType: kind === "speech" ? "audio/mpeg" : `${kind}/png`, size: 12, createdAt: "2026-09-17T00:00:00Z", url: `/media/${id}` }],
+})
+const baseItem = item("a")
+const clients: QueryClient[] = []
+function client() {
+  const value = new QueryClient({ defaultOptions: { queries: { retry: false }, mutations: { retry: false } } })
+  clients.push(value)
+  return value
+}
+function grid(props = {}) {
+  return render(<QueryClientProvider client={client()}><GalleryGrid {...props} /></QueryClientProvider>)
+}
+beforeAll(() => {
+  Element.prototype.scrollIntoView = () => {}
+})
+beforeEach(() => {
+  vi.mocked(listGenerationRunsFn).mockReset()
+  vi.mocked(createVariantFn).mockReset()
+  vi.mocked(attachToStoryFn).mockReset()
+  vi.mocked(attachToKbFn).mockReset()
+  vi.mocked(attachPrototypeToPrdFn).mockReset()
+  vi.mocked(listStoriesFn).mockReset()
+  vi.mocked(listKnowledgeFn).mockReset()
+  vi.mocked(listPrdFn).mockReset()
+})
+afterEach(() => { cleanup(); clients.splice(0).forEach((value) => value.clear()) })
+
+describe("gallery behavior", () => {
+  it("loads opaque cursors only on demand and keeps previous pages", async () => {
+    const cursor = '{"startedAt":"2026-09-17T00:00:00.000Z","id":"a"}'
+    vi.mocked(listGenerationRunsFn).mockResolvedValueOnce({ runs: [item("a")], nextCursor: cursor }).mockResolvedValue({ runs: [item("b")], nextCursor: null })
+    grid()
+    await screen.findByRole("button", { name: "预览 a" })
+    expect(listGenerationRunsFn).toHaveBeenCalledTimes(1)
+    expect(listGenerationRunsFn).toHaveBeenCalledWith({ data: { kind: undefined, threadId: undefined, cursor: undefined, limit: 24 } })
+    fireEvent.click(screen.getByRole("button", { name: "加载更多" }))
+    await screen.findByRole("button", { name: "预览 b" })
+    expect(listGenerationRunsFn).toHaveBeenLastCalledWith({ data: { kind: undefined, threadId: undefined, cursor, limit: 24 } })
+    expect(screen.getByRole("button", { name: "预览 a" })).toBeInTheDocument()
+    expect(screen.queryByRole("button", { name: "加载更多" })).not.toBeInTheDocument()
+  })
+
+  it("shows a localized retryable error instead of an empty gallery", async () => {
+    vi.mocked(listGenerationRunsFn).mockRejectedValueOnce(new Error("private backend error")).mockResolvedValue({ runs: [], nextCursor: null })
+    grid()
+    expect(await screen.findByRole("alert")).toHaveTextContent("加载媒体失败")
+    expect(screen.queryByText("暂无媒体记录")).not.toBeInTheDocument()
+    fireEvent.click(screen.getByRole("button", { name: "重试" }))
+    expect(await screen.findByText("暂无媒体记录")).toBeInTheDocument()
+  })
+
+  it("opens image parameters and navigates to real video and audio players", async () => {
+    vi.mocked(listGenerationRunsFn).mockResolvedValue({ runs: [item("a"), item("b", "video"), item("c", "speech")], nextCursor: null })
+    grid()
+    fireEvent.click(await screen.findByRole("button", { name: "预览 a" }))
+    const dialog = await screen.findByRole("dialog")
+    expect(within(dialog).getByRole("img")).toHaveAttribute("src", "/media/a")
+    expect(within(dialog).getByText("prompt-a")).toBeInTheDocument()
+    expect(within(dialog).getByText("test-model")).toBeInTheDocument()
+    fireEvent.click(within(dialog).getByRole("button", { name: "下一个" }))
+    expect(dialog.querySelector("video[controls]")).toHaveAttribute("src", "/media/b")
+    fireEvent.click(within(dialog).getByRole("button", { name: "下一个" }))
+    expect(dialog.querySelector("audio[controls]")).toHaveAttribute("src", "/media/c")
+    fireEvent.click(within(dialog).getByRole("button", { name: "上一个" }))
+    expect(dialog.querySelector("video[controls]")).toBeInTheDocument()
+    fireEvent.click(within(dialog).getByRole("button", { name: "关闭" }))
+    await waitFor(() => expect(screen.queryByRole("dialog")).not.toBeInTheDocument())
+  })
+
+  it("generates an image variant with its thread and refreshes after success", async () => {
+    vi.mocked(listGenerationRunsFn).mockResolvedValueOnce({ runs: [item("a")], nextCursor: null }).mockResolvedValue({ runs: [item("new"), item("a")], nextCursor: null })
+    vi.mocked(createVariantFn).mockResolvedValue({ runId: "new", artifacts: [], images: [] })
+    grid()
+    fireEvent.click(await screen.findByRole("button", { name: "生成变体" }))
+    await screen.findByRole("button", { name: "预览 new" })
+    expect(createVariantFn).toHaveBeenCalledWith({ data: { sourceRunId: "a", threadId: "thread-1" } })
+    expect(listGenerationRunsFn).toHaveBeenCalledTimes(2)
+  })
+
+  it("offers variants for videos and speech but not without a thread", async () => {
+    vi.mocked(listGenerationRunsFn).mockResolvedValue({ runs: [item("a", "image", ""), item("b", "video"), item("c", "speech")], nextCursor: null })
+    grid()
+    await screen.findByRole("button", { name: "预览 a" })
+    expect(screen.getAllByRole("button", { name: "生成变体" })).toHaveLength(2)
+  })
+
+  it("keeps the gallery and localizes variant failures", async () => {
+    vi.mocked(listGenerationRunsFn).mockResolvedValue({ runs: [item("a")], nextCursor: null })
+    vi.mocked(createVariantFn).mockRejectedValue(new Error("provider failure"))
+    grid()
+    fireEvent.click(await screen.findByRole("button", { name: "生成变体" }))
+    expect(await screen.findByRole("alert")).toHaveTextContent("生成变体失败")
+    expect(screen.getByRole("button", { name: "预览 a" })).toBeInTheDocument()
+    expect(listGenerationRunsFn).toHaveBeenCalledTimes(1)
+  })
+
+  it("reaches the full gallery through the real harness parent component", async () => {
+    vi.mocked(listGenerationRunsFn).mockResolvedValue({ runs: [item("full")], nextCursor: null })
+    const root = createRootRoute({ component: Outlet })
+    const parent = createRoute({ getParentRoute: () => root, path: "harness", component: HarnessRoute.options.component })
+    const gallery = createRoute({ getParentRoute: () => parent, path: "gallery", component: GalleryRoute.options.component })
+    const router = createRouter({ routeTree: root.addChildren([parent.addChildren([gallery])]), history: createMemoryHistory({ initialEntries: ["/harness/gallery"] }) })
+    render(<QueryClientProvider client={client()}><RouterProvider router={router} /></QueryClientProvider>)
+    expect(await screen.findByRole("button", { name: "预览 full" })).toBeInTheDocument()
+    expect(screen.queryByText("Workbench")).not.toBeInTheDocument()
+  })
+
+  it("shows a linked badge when linkedTarget is present", async () => {
+    const base = { ...baseItem, linkedTarget: { kind: "story" as const, targetId: "st-1", at: "2026-09-17T00:00:00Z" } }
+    vi.mocked(listGenerationRunsFn).mockResolvedValue({ runs: [base], nextCursor: null })
+    grid()
+    await screen.findByRole("button", { name: "预览 a" })
+    expect(screen.getByText(m.gallery_linked(), { exact: false })).toBeInTheDocument()
+  })
+
+  it("opens attach dialog and confirms story attachment", async () => {
+    vi.mocked(listGenerationRunsFn).mockResolvedValue({ runs: [baseItem], nextCursor: null })
+    vi.mocked(listStoriesFn).mockResolvedValue({ items: [{ id: "st-1", name: "Story One" }], total: 1, hasMore: false })
+    vi.mocked(attachToStoryFn).mockResolvedValue({ attachmentId: "att-1", storageUrl: "/api/attachments/att-1/content", linkedTarget: { kind: "story", targetId: "st-1", at: "2026-09-17T00:00:00Z" } })
+    grid()
+    fireEvent.click(await screen.findByRole("combobox", { name: m.gallery_attachTo() }))
+    fireEvent.click(await screen.findByText(m.gallery_attachToStory()))
+    fireEvent.click(await screen.findByText("Story One"))
+    fireEvent.click(screen.getByRole("button", { name: m.gallery_pickConfirm() }))
+    await waitFor(() => expect(attachToStoryFn).toHaveBeenCalled())
+    expect(attachToStoryFn).toHaveBeenCalledWith({ data: { generationRunId: "a", storyId: "st-1" } })
+  })
+})
+
+it("preview selection follows the parent route and reports changes", async () => {
+  const change = vi.fn()
+  const view = render(<PreviewPanel route="code" html="<p>code</p>" spec={{ type: "text", content: "spec output" }} onRouteChange={change} />)
+  view.rerender(<PreviewPanel route="spec" html="<p>code</p>" spec={{ type: "text", content: "spec output" }} onRouteChange={change} />)
+  expect(screen.getByText("spec output")).toBeVisible()
+  fireEvent.click(screen.getByRole("tab", { name: /代码/ }))
+  expect(change).toHaveBeenCalledWith("code")
+})

package/src/components/harness/__tests__/harness-workbench-mount.test.ts

@@ -0,0 +1,47 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+// AllBlue - 理想之海
+// Copyright (C) 2026 AllBlue Contributors
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <https://www.gnu.org/licenses/>.
+
+import { describe, expect, it } from "vitest"
+import { readFileSync } from "node:fs"
+import path from "node:path"
+import { resolveMountEffectPath } from "../harness-workbench-mount"
+
+describe("resolveMountEffectPath（mount effect 门控拆解：ctx 单条件 prefill + pending 单条件 autoSend）", () => {
+  it("ctx 非空 + pending 非空 → both（既有 prd/story 场景：既 prefill 又自动发消息）", () => {
+    expect(resolveMountEffectPath({ board: "prd", entityId: "1" }, "问题")).toBe("both")
+  })
+
+  it("ctx 非空 + pending 空 → prefill-only（4 板块「问 AI」新场景：仅注入 prefill，不自动发消息）", () => {
+    expect(resolveMountEffectPath({ board: "workspaces", entityId: "ws-1" }, null)).toBe("prefill-only")
+    expect(resolveMountEffectPath({ board: "agents", entityId: "ag-1" }, "  ")).toBe("prefill-only")
+  })
+
+  it("ctx 空 + pending 非空 → auto-send-only（既有 pending-only 场景：仅自动发消息，不 prefill）", () => {
+    expect(resolveMountEffectPath(null, "继续之前的对话")).toBe("auto-send-only")
+  })
+
+  it("ctx 空 + pending 空 → none（常规首访：什么也不做）", () => {
+    expect(resolveMountEffectPath(null, null)).toBe("none")
+    expect(resolveMountEffectPath(null, "")).toBe("none")
+  })
+
+  it("源码守护：harness-workbench-page 的 setHandoffContext 由 path（resolveMountEffectPath 结果）门控，而非 ctx&&pending 双非空", () => {
+    const src = readFileSync(path.resolve(__dirname, "../harness-workbench-page.tsx"), "utf-8")
+    expect(src).toContain('resolveMountEffectPath(ctx, pending)')
+    expect(src).not.toContain('if (ctx && pending')
+  })
+})

package/src/components/harness/__tests__/media-route.test.tsx

@@ -0,0 +1,52 @@
+import { afterEach, expect, it, vi } from "vitest"
+import { cleanup, fireEvent, render, screen, waitFor } from "@testing-library/react"
+import { QueryClient, QueryClientProvider } from "@tanstack/react-query"
+import type { HarnessComposerProps } from "../composer/harness-composer"
+import { Button } from "@/components/ui/button"
+import HarnessWorkbenchPage from "../harness-workbench-page"
+import { dockRegistry } from "../dock/dock-registry"
+
+const { sendMessage } = vi.hoisted(() => ({ sendMessage: vi.fn() }))
+vi.mock("@tanstack/react-router", async (original) => ({ ...await original<typeof import("@tanstack/react-router")>(), useNavigate: () => vi.fn() }))
+vi.mock("@/lib/auth/auth-client", () => ({ useSession: () => ({ data: { user: { hasAiKey: true } } }) }))
+vi.mock("../chat-ui", () => ({ useHarnessChat: () => ({ sendMessage, messages: [], interrupts: [], isLoading: false, clear: vi.fn(), stop: vi.fn(), AppChat: () => null }) }))
+vi.mock("@/server/serverFns/chat/conversations", () => ({ listConversationsFn: vi.fn(async () => []), createConversationFn: vi.fn(), updateConversationFn: vi.fn(), deleteConversationFn: vi.fn(), forkConversationFn: vi.fn() }))
+vi.mock("@/server/serverFns/agents/agent-crud", () => ({ listAgentsFn: vi.fn(async () => []) }))
+vi.mock("@/server/serverFns/harness/cancel-run", () => ({ cancelRunFn: vi.fn() }))
+vi.mock("@/lib/ai/use-selected-model", () => ({ useSelectedModel: () => [undefined, vi.fn()] }))
+vi.mock("@/lib/ai/use-search-capability", () => ({ useSearchCapability: () => ({}) }))
+vi.mock("@/lib/ai/use-code-mode-readiness", () => ({ useCodeModeReadiness: () => ({}) }))
+vi.mock("../ai-chat-history", () => ({ default: () => null }))
+vi.mock("../agent-preset-picker", () => ({ AgentPresetPicker: () => null }))
+vi.mock("../code/workspace-context", () => ({ useWorkspace: () => ({}), WorkspaceProvider: ({ children }: { children: React.ReactNode }) => children }))
+vi.mock("../code/code-workspace-sidebar", () => ({ CodeWorkspaceSidebar: () => null }))
+vi.mock("../dock/dock-panel-container", () => ({ DockPanelContainer: () => null }))
+vi.mock("../workbench/workbench-shell", () => ({ WorkbenchShell: ({ center }: { center: React.ReactNode }) => center }))
+vi.mock("../composer/harness-composer", () => ({ HarnessComposer: (props: HarnessComposerProps) => <>
+  <output>{props.mediaRoute}</output>
+  <Button onClick={() => props.onMediaRouteChange?.("spec")}>select spec</Button>
+  <Button onClick={() => props.onMediaRouteChange?.("code")}>select code</Button>
+  <Button onClick={() => props.onSend("draw a page")}>send</Button>
+</> }))
+afterEach(cleanup)
+
+it("parent route selection allows the spec route (no longer blocks) and the existing code send path", async () => {
+  const client = new QueryClient({ defaultOptions: { queries: { retry: false } } })
+  render(<QueryClientProvider client={client}><HarnessWorkbenchPage initialThreadId="thread-1" /></QueryClientProvider>)
+  expect(screen.getByText("code")).toBeInTheDocument()
+  fireEvent.click(screen.getByRole("button", { name: "代码工作区" }))
+  expect(dockRegistry.getState().activeId).toBe("code-workspace")
+  fireEvent.click(screen.getByRole("button", { name: "select spec" }))
+  expect(screen.getByText("spec")).toBeInTheDocument()
+  const panel = dockRegistry.getPanel("code-workspace")
+  expect(panel).toBeDefined()
+  expect(panel?.render()).toMatchObject({ props: { mediaRoute: "spec", threadId: "thread-1" } })
+  fireEvent.click(screen.getByRole("button", { name: "send" }))
+  expect(screen.queryByRole("alert")).not.toBeInTheDocument()
+  await waitFor(() => expect(sendMessage).toHaveBeenCalledWith("draw a page", undefined))
+  fireEvent.click(screen.getByRole("button", { name: "select code" }))
+  sendMessage.mockClear()
+  fireEvent.click(screen.getByRole("button", { name: "send" }))
+  await waitFor(() => expect(sendMessage).toHaveBeenCalledWith("draw a page", undefined))
+  client.clear()
+})

package/src/components/harness/__tests__/mention-resolvers.test.ts

@@ -0,0 +1,41 @@
+import { describe, expect, it } from "vitest"
+import { groupMentionCandidates, extractKbIds } from "../composer/mention/mention-resolvers"
+
+const candidates = [
+  { kind: "kb-doc" as const, label: "架构文档", id: "d1", kbId: "kb1" },
+  { kind: "file" as const, label: "src/index.ts", id: "f1" },
+  { kind: "prd" as const, label: "登录 PRD", id: "p1" },
+  { kind: "story" as const, label: "注册故事", id: "s1" },
+  { kind: "plugin-tool" as const, label: "web_search", id: "t1" },
+  { kind: "thread" as const, label: "昨天的会话", id: "th1" },
+] as const
+
+describe("groupMentionCandidates（文件在前，业务实体次之，工具/会话在后）", () => {
+  it("按组序输出", () => {
+    const groups = groupMentionCandidates("", candidates)
+    expect(groups.map((g) => g.kind)).toEqual([
+      "file",
+      "kb-doc",
+      "prd",
+      "story",
+      "plugin-tool",
+      "thread",
+    ])
+  })
+  it("子串过滤各组", () => {
+    const groups = groupMentionCandidates("登录", candidates)
+    expect(groups.flatMap((g) => g.items.map((i) => i.label))).toEqual(["登录 PRD"])
+  })
+})
+
+describe("extractKbIds", () => {
+  it("从 chip 集合提取去重 kbId（供请求注入）", () => {
+    expect(
+      extractKbIds([
+        { kind: "kb-doc" as const, kbId: "kb1" },
+        { kind: "kb-doc" as const, kbId: "kb1" },
+        { kind: "prd" as const },
+      ]),
+    ).toEqual(["kb1"])
+  })
+})

package/src/components/harness/__tests__/message-actions.test.tsx

@@ -0,0 +1,199 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+// AllBlue - 理想之海
+// Copyright (C) 2026 AllBlue Contributors
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <https://www.gnu.org/licenses/>.
+
+import { describe, expect, it, beforeEach, vi } from 'vitest'
+import { render, screen, fireEvent } from '@testing-library/react'
+import userEvent from '@testing-library/user-event'
+import type { ComponentType } from 'react'
+import { HotkeysProvider } from '@tanstack/react-hotkeys'
+import type { UIMessage } from '@tanstack/ai-client'
+
+const { mockChat } = vi.hoisted(() => ({
+  mockChat: {
+    messages: [] as UIMessage[],
+    isLoading: false,
+    reload: vi.fn(async () => undefined),
+  },
+}))
+
+vi.mock('../harness-chat-contexts', () => ({
+  harnessChatContexts: {
+    useChatContext: () => mockChat,
+  },
+}))
+
+vi.mock('../ai-citations', () => ({
+  default: () => null,
+}))
+
+import { HarnessMessage } from '../harness-message'
+import { MessageActions } from '../message-actions'
+
+function makeMessage(
+  id: string,
+  role: 'user' | 'assistant',
+  text: string,
+): UIMessage {
+  return { id, role, parts: [{ type: 'text', content: text }] }
+}
+
+const StubParts: ComponentType = () => <span>消息正文</span>
+
+function renderAssistant(overrides: Partial<typeof mockChat> = {}) {
+  mockChat.messages = [makeMessage('a1', 'assistant', '你好，世界')]
+  mockChat.isLoading = false
+  Object.assign(mockChat, overrides)
+  return render(
+    <HotkeysProvider>
+      <HarnessMessage
+        message={makeMessage('a1', 'assistant', '你好，世界')}
+        Parts={StubParts}
+      />
+    </HotkeysProvider>,
+  )
+}
+
+async function openMenu() {
+  const user = userEvent.setup()
+  fireEvent.click(screen.getByRole('button', { name: '消息操作菜单' }))
+  return user
+}
+
+describe('MessageActions', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+  })
+
+  it('打开菜单并提供复制项，点击后回调 onCopy', async () => {
+    const onCopy = vi.fn()
+
+    render(
+      <HotkeysProvider>
+        <MessageActions copied={false} onCopy={onCopy} />
+      </HotkeysProvider>,
+    )
+
+    const user = await openMenu()
+    await user.click(screen.getByRole('menuitem', { name: /复制/ }))
+
+    expect(onCopy).toHaveBeenCalledTimes(1)
+  })
+
+  it('onRegenerate 存在时提供重新生成项，点击后回调', async () => {
+    const onRegenerate = vi.fn()
+
+    render(
+      <HotkeysProvider>
+        <MessageActions
+          copied={false}
+          onCopy={vi.fn()}
+          onRegenerate={onRegenerate}
+        />
+      </HotkeysProvider>,
+    )
+
+    const user = await openMenu()
+    await user.click(screen.getByRole('menuitem', { name: '重新生成' }))
+
+    expect(onRegenerate).toHaveBeenCalledTimes(1)
+  })
+
+  it('未传 onRegenerate 时不渲染重新生成项', async () => {
+    render(
+      <HotkeysProvider>
+        <MessageActions copied={false} onCopy={vi.fn()} />
+      </HotkeysProvider>,
+    )
+
+    await openMenu()
+
+    expect(screen.queryByRole('menuitem', { name: '重新生成' })).toBeNull()
+  })
+
+  it('copied 为真时复制项文案切换为「已复制」', async () => {
+    render(
+      <HotkeysProvider>
+        <MessageActions copied onCopy={vi.fn()} />
+      </HotkeysProvider>,
+    )
+
+    await openMenu()
+
+    expect(screen.getByRole('menuitem', { name: /已复制/ })).toBeTruthy()
+  })
+})
+
+describe('HarnessMessage 操作按钮挂载', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+  })
+
+  it('assistant 消息渲染操作菜单，点击复制写入正文文本', async () => {
+    renderAssistant()
+
+    expect(screen.getByRole('button', { name: '消息操作菜单' })).toBeTruthy()
+
+    const user = await openMenu()
+    await user.click(screen.getByRole('menuitem', { name: /复制/ }))
+    await new Promise((r) => setTimeout(r, 50))
+
+    expect(await navigator.clipboard.readText()).toBe('你好，世界')
+  })
+
+  it('最后一条 assistant 消息提供重新生成，点击调用 chat.reload', async () => {
+    renderAssistant()
+
+    const user = await openMenu()
+    await user.click(screen.getByRole('menuitem', { name: '重新生成' }))
+
+    expect(mockChat.reload).toHaveBeenCalledTimes(1)
+  })
+
+  it('非最后一条消息不提供重新生成', async () => {
+    renderAssistant({
+      messages: [
+        makeMessage('a1', 'assistant', '上一轮'),
+        makeMessage('u2', 'user', '再问一次'),
+      ],
+    })
+
+    await openMenu()
+
+    expect(screen.queryByRole('menuitem', { name: '重新生成' })).toBeNull()
+  })
+
+  it('流式进行中隐藏操作菜单', () => {
+    renderAssistant({ isLoading: true })
+
+    expect(screen.queryByRole('button', { name: '消息操作菜单' })).toBeNull()
+  })
+
+  it('user 消息不渲染操作菜单', () => {
+    mockChat.messages = [makeMessage('u1', 'user', '提问')]
+    mockChat.isLoading = false
+    render(
+      <HotkeysProvider>
+        <HarnessMessage
+          message={makeMessage('u1', 'user', '提问')}
+          Parts={StubParts}
+        />
+      </HotkeysProvider>,
+    )
+
+    expect(screen.queryByRole('button', { name: '消息操作菜单' })).toBeNull()
+  })
+})

package/src/components/harness/__tests__/resource-uri.test.ts

@@ -0,0 +1,53 @@
+import { describe, expect, it } from "vitest"
+import { parseResourceUri } from "../dock/resource-uri"
+
+describe("parseResourceUri", () => {
+  it("file URI 解析 workspaceId + relPath", () => {
+    expect(parseResourceUri("allblue-resource://file/ws1/src/index.ts")).toEqual({
+      kind: "file",
+      workspaceId: "ws1",
+      relPath: "src/index.ts",
+    })
+  })
+
+  it("file URI 多级路径", () => {
+    expect(parseResourceUri("allblue-resource://file/ws2/deep/nested/file.ts")).toEqual({
+      kind: "file",
+      workspaceId: "ws2",
+      relPath: "deep/nested/file.ts",
+    })
+  })
+
+  it("kb URI 解析 docId", () => {
+    expect(parseResourceUri("allblue-resource://kb/doc1")).toEqual({
+      kind: "kb",
+      docId: "doc1",
+    })
+  })
+
+  it("prd URI 解析 id", () => {
+    expect(parseResourceUri("allblue-resource://prd/p1")).toEqual({
+      kind: "prd",
+      id: "p1",
+    })
+  })
+
+  it("story URI 解析 id", () => {
+    expect(parseResourceUri("allblue-resource://story/s1")).toEqual({
+      kind: "story",
+      id: "s1",
+    })
+  })
+
+  it("非法 scheme → null", () => {
+    expect(parseResourceUri("https://example.com")).toBeNull()
+  })
+
+  it("无 scheme 纯路径 → null", () => {
+    expect(parseResourceUri("some/other/path")).toBeNull()
+  })
+
+  it("未知 kind → null", () => {
+    expect(parseResourceUri("allblue-resource://unknown/foo")).toBeNull()
+  })
+})

package/src/components/harness/__tests__/submit-machine.test.ts

@@ -0,0 +1,61 @@
+import { describe, expect, it } from "vitest"
+import { decideSubmit, type SubmitIntent } from "../composer/submit-machine"
+
+const base: SubmitIntent = {
+  text: "你好",
+  hasUploadingAttachments: false,
+  isRunning: false,
+  busyPreference: "queue",
+  modifierKey: false,
+  composerLocked: false,
+}
+
+describe("decideSubmit", () => {
+  it("空闲 + 有文本 → 立即提交并清空输入", () => {
+    expect(decideSubmit(base)).toEqual({
+      action: "submit", clearInput: true,
+    })
+  })
+
+  it("空文本 → 阻塞（空）", () => {
+    expect(decideSubmit({ ...base, text: "  " })).toEqual({
+      action: "blocked", reason: "empty", clearInput: false,
+    })
+  })
+
+  it("附件上传中 → 阻塞（上传中），绝不静默丢弃", () => {
+    expect(decideSubmit({ ...base, hasUploadingAttachments: true })).toEqual({
+      action: "blocked", reason: "uploading", clearInput: false,
+    })
+  })
+
+  it("运行中 + 偏好 queue → 排队", () => {
+    expect(decideSubmit({ ...base, isRunning: true })).toEqual({
+      action: "queue", clearInput: true,
+    })
+  })
+
+  it("运行中 + 偏好 steer → 插队", () => {
+    expect(
+      decideSubmit({ ...base, isRunning: true, busyPreference: "steer" }),
+    ).toEqual({ action: "steer", clearInput: true })
+  })
+
+  it("运行中 + Cmd 取反偏好（queue→steer）", () => {
+    expect(
+      decideSubmit({ ...base, isRunning: true, modifierKey: true }),
+    ).toEqual({ action: "steer", clearInput: true })
+  })
+
+  it("运行中 + 空草稿 + Cmd → steer 整个待发队列", () => {
+    expect(
+      decideSubmit({ ...base, text: "", isRunning: true, modifierKey: true }),
+    ).toEqual({ action: "steerQueue", clearInput: false })
+  })
+
+  it("composer 被 claim 锁定（斜杠命令占位）→ 阻塞", () => {
+    expect(decideSubmit({ ...base, composerLocked: true })).toEqual({
+      action: "blocked", reason: "composer-locked", clearInput: false,
+    })
+  })
+})