@allbluecn/web-app 0.4.22 → 0.5.1

package/src/components/story/inline-editors/story-select-dialog.tsx

@@ -19,7 +19,7 @@ import { useCallback, useEffect, useMemo, useRef, useState } from "react";
 import { useTranslation } from "react-i18next";
 import { useDebouncedCallback, SEARCH_DEBOUNCE_MS } from "@/lib/pacer";
 import { cn } from "@/lib/utils";
-import { Search, Sprout, ChevronsUpDown, Check, Trees, ListFilter, Flag } from "lucide-react";
+import { Search, Sprout, ChevronsUpDown, Check, Trees, ListFilter, Flag, Waypoints } from "lucide-react";
 import { Badge } from "@/components/ui/badge";
 import { Checkbox } from "@/components/ui/checkbox";
 import { ProjectIconDisplay } from "@/components/project/project-icon";
@@ -50,7 +50,9 @@ import {
   STATE_LABEL,
   STORY_STATES,
   STORY_PRIORITIES,
+  DEPENDENCY_TYPE_OPTIONS,
   type StoryState,
+  type DependencyTypeOption,
 } from "../constants";
 import { PriorityIcon } from "../priority-icon";
 
@@ -71,7 +73,7 @@ interface StorySelectDialogProps {
   initialProjectId?: string | null;
   defaultSelectedParentId?: string | null;
   defaultSelectedParentIds?: string[];
-  mode?: "parent" | "child";
+  mode?: "parent" | "child" | "dependency";
   onSelected: (parentId: string | string[]) => void;
   onActivities?: (activities: ActivityItem[]) => void;
   skipTransfer?: boolean;
@@ -79,9 +81,15 @@ interface StorySelectDialogProps {
   excludeProjectIds?: string[];
   hideAllProjects?: boolean;
   filteredTip?: string;
+  /** dependency 模式：额外排除的需求 id（已建立依赖的对方） */
+  excludeIds?: string[];
+  /** dependency 模式：当前选中的依赖类型（受控），由父组件持有以便提交时映射方向 */
+  dependencyType?: DependencyTypeOption["key"];
+  onDependencyTypeChange?: (key: DependencyTypeOption["key"]) => void;
 }
 
 const ALL_PROJECTS = "__all__";
+const EMPTY_IDS: string[] = [];
 const ALL_STATES = STORY_STATES.map((s) => s.value);
 const ALL_PRIORITIES = STORY_PRIORITIES.map((p) => p.value);
 
@@ -101,10 +109,13 @@ export function StorySelectDialog({
   excludeProjectIds = [],
   hideAllProjects = false,
   filteredTip,
+  excludeIds = EMPTY_IDS,
+  dependencyType,
+  onDependencyTypeChange,
 }: StorySelectDialogProps) {
   const { t } = useTranslation("story");
   const resolvedTitle = title ?? t("select.defaultTitle");
-  const isMulti = mode === "child";
+  const isMulti = mode === "child" || mode === "dependency";
   const [projectFilter, setProjectFilter] = useState<string>(ALL_PROJECTS);
   const [items, setItems] = useState<ParentStoryItem[]>([]);
   const [search, setSearch] = useState("");
@@ -117,6 +128,7 @@ export function StorySelectDialog({
   const [projectPopoverOpen, setProjectPopoverOpen] = useState(false);
   const [statePopoverOpen, setStatePopoverOpen] = useState(false);
   const [priorityPopoverOpen, setPriorityPopoverOpen] = useState(false);
+  const [depPopoverOpen, setDepPopoverOpen] = useState(false);
   const [selectedStates, setSelectedStates] = useState<Set<string>>(
     () => new Set(ALL_STATES),
   );
@@ -209,6 +221,7 @@ export function StorySelectDialog({
         projectId: projectFilter === ALL_PROJECTS ? undefined : projectFilter,
         search: debouncedSearch || undefined,
         excludeStoryId: storyId,
+        excludeIds: excludeIds.length > 0 ? excludeIds : undefined,
         mode,
       },
     })
@@ -226,7 +239,7 @@ export function StorySelectDialog({
         if (reqId !== requestIdRef.current) return;
         setLoading(false);
       });
-  }, [open, projectFilter, debouncedSearch, storyId, mode]);
+  }, [open, projectFilter, debouncedSearch, storyId, excludeIds, mode]);
 
   const loadMore = useCallback(async () => {
     if (loading || !hasMore || !nextCursor) return;
@@ -239,6 +252,7 @@ export function StorySelectDialog({
           search: debouncedSearch || undefined,
           cursor: nextCursor,
           excludeStoryId: storyId,
+          excludeIds: excludeIds.length > 0 ? excludeIds : undefined,
           mode,
         },
       });
@@ -253,11 +267,17 @@ export function StorySelectDialog({
       if (reqId !== requestIdRef.current) return;
       setLoading(false);
     }
-  }, [loading, hasMore, nextCursor, projectFilter, debouncedSearch, storyId, mode]);
+  }, [loading, hasMore, nextCursor, projectFilter, debouncedSearch, storyId, excludeIds, mode]);
 
   // 点击 Popover 外部时关闭（Dialog 嵌套下 Radix 事件被拦截）
   useEffect(() => {
-    if (!projectPopoverOpen && !statePopoverOpen && !priorityPopoverOpen) return;
+    if (
+      !projectPopoverOpen &&
+      !statePopoverOpen &&
+      !priorityPopoverOpen &&
+      !depPopoverOpen
+    )
+      return;
     const handlePointerDown = (e: PointerEvent) => {
       const target = e.target as Node;
       const content = document.querySelector('[data-slot="popover-content"]');
@@ -271,11 +291,12 @@ export function StorySelectDialog({
         setProjectPopoverOpen(false);
         setStatePopoverOpen(false);
         setPriorityPopoverOpen(false);
+        setDepPopoverOpen(false);
       }
     };
     document.addEventListener("pointerdown", handlePointerDown);
     return () => document.removeEventListener("pointerdown", handlePointerDown);
-  }, [projectPopoverOpen, statePopoverOpen, priorityPopoverOpen]);
+  }, [projectPopoverOpen, statePopoverOpen, priorityPopoverOpen, depPopoverOpen]);
 
   const handleScroll = (e: React.UIEvent<HTMLDivElement>) => {
     const el = e.currentTarget;
@@ -288,8 +309,8 @@ export function StorySelectDialog({
   const handleSubmit = async () => {
     if (selectedIds.length === 0) return;
     if (skipTransfer) {
+      // dependency 模式：关闭时机由回调方控制（部分失败时保留弹窗与已选项）
       const parentId = isMulti ? selectedIds : selectedIds[0];
-      onOpenChange(false);
       onSelected(parentId);
       return;
     }
@@ -384,7 +405,7 @@ export function StorySelectDialog({
                         )}
                       >
                         <Icon className={cn("size-3.5 shrink-0", STATE_ICON_COLOR[s.value])} />
-                        <span className="flex-1">{s.label}</span>
+                        <span className="flex-1">{t(s.label)}</span>
                         {checked && <Check className="size-3.5 shrink-0 text-primary" />}
                       </button>
                     );
@@ -432,7 +453,7 @@ export function StorySelectDialog({
                         )}
                       >
                         <PriorityIcon priority={p.value} />
-                        <span className="flex-1">{p.label}</span>
+                        <span className="flex-1">{t(p.label)}</span>
                         {checked && <Check className="size-3.5 shrink-0 text-primary" />}
                       </button>
                     );
@@ -541,8 +562,9 @@ export function StorySelectDialog({
                     const owner = projectById.get(story.projectId);
                     return (
                       <li key={story.id}>
-                        <button
-                          type="button"
+                        <div
+                          role="button"
+                          tabIndex={0}
                           onClick={() => {
                             if (isMulti) {
                               setSelectedIds((prev) =>
@@ -554,7 +576,21 @@ export function StorySelectDialog({
                               setSelectedIds([story.id]);
                             }
                           }}
-                          className={`flex w-full items-center gap-2.5 rounded-md px-3 py-2 text-left text-sm transition-colors ${
+                          onKeyDown={(e) => {
+                            if (e.key === "Enter" || e.key === " ") {
+                              e.preventDefault();
+                              if (isMulti) {
+                                setSelectedIds((prev) =>
+                                  prev.includes(story.id)
+                                    ? prev.filter((id) => id !== story.id)
+                                    : [...prev, story.id],
+                                );
+                              } else {
+                                setSelectedIds([story.id]);
+                              }
+                            }
+                          }}
+                          className={`flex w-full cursor-pointer items-center gap-2.5 rounded-md px-3 py-2 text-left text-sm transition-colors ${
                             isSelected
                               ? "bg-accent ring-1 ring-primary/30"
                               : "hover:bg-muted"
@@ -579,7 +615,7 @@ export function StorySelectDialog({
                               STATE_COLOR[state] ?? STATE_COLOR.backlog
                             }`}
                           >
-                            {STATE_LABEL[state] ?? state}
+                            {t(STATE_LABEL[state] ?? state)}
                           </span>
                           <PriorityIcon priority={priority} />
                           <span className="shrink-0 font-mono text-xs text-muted-foreground">
@@ -596,7 +632,7 @@ export function StorySelectDialog({
                           {!isMulti && isSelected && (
                             <Check className="size-4 shrink-0 text-primary" />
                           )}
-                        </button>
+                        </div>
                       </li>
                     );
                   })}
@@ -638,6 +674,57 @@ export function StorySelectDialog({
               <span className="text-xs text-muted-foreground">
                 {selectedCount > 0 ? t("select.selectedCount", { count: selectedCount }) : t("select.pleaseSelect")}
               </span>
+              {mode === "dependency" && (
+                <Popover open={depPopoverOpen} onOpenChange={setDepPopoverOpen}>
+                  <PopoverTrigger asChild>
+                    <Button
+                      variant="ghost"
+                      size="sm"
+                      className={cn(
+                        "h-9 gap-1.5",
+                        !dependencyType && "text-muted-foreground",
+                      )}
+                    >
+                      <Waypoints className="size-3.5" />
+                      {dependencyType
+                        ? t(
+                            DEPENDENCY_TYPE_OPTIONS.find((o) => o.key === dependencyType)!
+                              .optionLabelKey,
+                          )
+                        : t("dependency.typePlaceholder")}
+                    </Button>
+                  </PopoverTrigger>
+                  <PopoverContent align="start" className="w-80 p-1">
+                    {(["dependsOn", "dependents"] as const).map((group, gi) => (
+                      <div key={group} className={cn(gi > 0 && "mt-1 border-t pt-1")}>
+                        <p className="px-2 py-1 text-xs font-medium text-muted-foreground">
+                          {t(`dependency.optionGroups.${group}`)}
+                        </p>
+                        {DEPENDENCY_TYPE_OPTIONS.filter((o) => o.group === group).map((opt) => {
+                          const checked = dependencyType === opt.key;
+                          return (
+                            <button
+                              key={opt.key}
+                              type="button"
+                              onClick={() => {
+                                onDependencyTypeChange?.(opt.key);
+                                setDepPopoverOpen(false);
+                              }}
+                              className={cn(
+                                "flex w-full items-center gap-2 rounded px-2 py-1.5 text-left text-sm hover:bg-muted",
+                                checked && "font-medium",
+                              )}
+                            >
+                              <span className="flex-1">{t(opt.optionLabelKey)}</span>
+                              {checked && <Check className="size-3.5 shrink-0 text-primary" />}
+                            </button>
+                          );
+                        })}
+                      </div>
+                    ))}
+                  </PopoverContent>
+                </Popover>
+              )}
             </div>
           )}
           <div className="flex items-center gap-3">
@@ -650,7 +737,11 @@ export function StorySelectDialog({
             </Button>
             <Button
               onClick={handleSubmit}
-              disabled={(isMulti ? selectedCount === 0 : selectedIds.length === 0) || submitting}
+              disabled={
+                (isMulti ? selectedCount === 0 : selectedIds.length === 0) ||
+                submitting ||
+                (mode === "dependency" && !dependencyType)
+              }
             >
               {submitting ? t("select.loading") : isMulti ? t("select.confirmWithCount", { count: selectedCount }) : t("select.confirm")}
             </Button>

package/src/components/story/peek-panel.tsx

@@ -246,7 +246,14 @@ export function StoryPeekPanel({
       <>
         <div
           className="absolute inset-y-0 right-0 z-20 flex w-[min(100%,max(85vw,560px),1200px)] flex-col overflow-hidden border-l bg-popover rounded-r-2xl rounded-l-none shadow-xl ring-1 ring-foreground/10 @max-[1399px]:rounded-l-2xl"
-          onClick={(e) => e.stopPropagation()}
+          onClick={(e) => {
+            // 仅拦截本容器 DOM 内的点击（防止冒泡到列表背景误关 peek）。
+            // Radix 弹层 portal 到 body、不在本容器 DOM 内，其合成事件冒泡会经过这里，
+            // 若 stopPropagation 会同时终止原生 click 冒泡到 document，
+            // 导致 DismissableLayer 的 outside-dismiss 判定失效（弹窗内 popover 无法关闭）。
+            if (!e.currentTarget.contains(e.target as Node)) return;
+            e.stopPropagation();
+          }}
         >
           {header}
           {body}

package/src/components/story/relations/__tests__/attachment-manage-dialog.test.tsx

@@ -0,0 +1,171 @@
+// 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/>.
+
+// jsdom 缺少 ResizeObserver，Radix Dialog 依赖它
+const MockResizeObserver = class ResizeObserver {
+  observe() {}
+  unobserve() {}
+  disconnect() {}
+};
+if (typeof globalThis.ResizeObserver === "undefined") {
+  (globalThis as any).ResizeObserver = MockResizeObserver;
+}
+
+import { describe, it, expect, vi, beforeEach } from "vitest";
+import { render, screen, fireEvent, waitFor, type RenderResult } from "@testing-library/react";
+import { TooltipProvider } from "@/components/ui/tooltip";
+
+vi.mock("sonner", () => ({
+  toast: { success: vi.fn(), error: vi.fn(), warning: vi.fn(), info: vi.fn() },
+}));
+
+const fns = vi.hoisted(() => ({ uploadFetch: vi.fn() }));
+
+// 上传走 /api/attachments/upload（multipart fetch），mock 全局 fetch
+vi.stubGlobal("fetch", fns.uploadFetch);
+
+function uploadResponse(body: Record<string, unknown>) {
+  return {
+    ok: true,
+    json: () => Promise.resolve(body),
+  };
+}
+
+import { AttachmentManageDialog } from "../attachment-manage-dialog";
+import type { StoryAttachmentItem } from "../../types";
+
+const ATTS: StoryAttachmentItem[] = [
+  { id: "a1", fileName: "设计稿.png", fileSize: 2048, mimeType: "image/png", storageUrl: "/mock-storage/1" },
+  { id: "a2", fileName: "说明.pdf", fileSize: 1048576, mimeType: "application/pdf", storageUrl: "/mock-storage/2" },
+];
+
+function setup(): RenderResult & { onUpdate: ReturnType<typeof vi.fn>; onOpenChange: ReturnType<typeof vi.fn> } {
+  const onUpdate = vi.fn();
+  const onOpenChange = vi.fn();
+  const view = render(
+    <TooltipProvider>
+      <AttachmentManageDialog
+        open
+        onOpenChange={onOpenChange}
+        storyId="s1"
+        data={ATTS}
+        onUpdate={onUpdate}
+      />
+    </TooltipProvider>,
+  );
+  return { ...view, onUpdate, onOpenChange };
+}
+
+function getFileInput() {
+  return document.querySelector('input[type="file"]') as HTMLInputElement;
+}
+
+function stageFile(file: File) {
+  fireEvent.change(getFileInput(), { target: { files: [file] } });
+}
+
+beforeEach(() => {
+  vi.clearAllMocks();
+  fns.uploadFetch.mockResolvedValue(
+    uploadResponse({
+      id: "a3",
+      fileName: "新文件.txt",
+      fileSize: 5,
+      mimeType: "text/plain",
+      storageUrl: "/api/attachments/a3/content",
+    }),
+  );
+});
+
+describe("AttachmentManageDialog", () => {
+  it("不渲染已有附件列表，仅保留上传区", () => {
+    setup();
+    expect(screen.queryByText("设计稿.png")).not.toBeInTheDocument();
+    expect(screen.getByText("点击或拖拽上传")).toBeInTheDocument();
+  });
+
+  it("选择文件后先在弹窗内预览（暂存），点击确认上传才发起 POST", async () => {
+    setup();
+    const file = new File(["hello"], "新文件.txt", { type: "text/plain" });
+    stageFile(file);
+    // 暂存预览：文件名 + 移除按钮出现
+    expect(screen.getByRole("button", { name: "移除" })).toBeInTheDocument();
+    expect(screen.getByRole("button", { name: "移除" })).toBeInTheDocument();
+    // 未确认前不发起上传
+    expect(fns.uploadFetch).not.toHaveBeenCalled();
+    fireEvent.click(screen.getByRole("button", { name: "确认上传" }));
+    await waitFor(() => expect(fns.uploadFetch).toHaveBeenCalled());
+    const [url, init] = fns.uploadFetch.mock.calls[0];
+    expect(url).toBe("/api/attachments/upload");
+    expect(init.method).toBe("POST");
+    const formData = init.body as FormData;
+    expect(formData.get("storyId")).toBe("s1");
+    expect(formData.get("file")).toBeInstanceOf(File);
+  });
+
+  it("确认上传成功后回调 onUpdate（含内容路由 URL）并关闭弹窗", async () => {
+    const { onUpdate, onOpenChange } = setup();
+    const file = new File(["hello"], "新文件.txt", { type: "text/plain" });
+    stageFile(file);
+    fireEvent.click(screen.getByRole("button", { name: "确认上传" }));
+    await waitFor(() => expect(onUpdate).toHaveBeenCalled());
+    const added = onUpdate.mock.calls[0][0][0];
+    expect(added.storageUrl).toBe("/api/attachments/a3/content");
+    expect(onOpenChange).toHaveBeenCalledWith(false);
+  });
+
+  it("描述随确认上传提交到 FormData", async () => {
+    setup();
+    const file = new File(["hello"], "新文件.txt", { type: "text/plain" });
+    stageFile(file);
+    fireEvent.change(screen.getByPlaceholderText("补充说明…"), { target: { value: "测试描述" } });
+    fireEvent.click(screen.getByRole("button", { name: "确认上传" }));
+    await waitFor(() => expect(fns.uploadFetch).toHaveBeenCalled());
+    const formData = fns.uploadFetch.mock.calls[0][1].body as FormData;
+    expect(formData.get("description")).toBe("测试描述");
+  });
+
+  it("上传接口报错时 toast 提示且不回调 onUpdate", async () => {
+    const { toast } = await import("sonner");
+    fns.uploadFetch.mockResolvedValue({ ok: false, json: () => Promise.resolve({ error: "文件大小需在 10MB 以内" }) });
+    const { onUpdate } = setup();
+    const file = new File(["hello"], "新文件.txt", { type: "text/plain" });
+    stageFile(file);
+    fireEvent.click(screen.getByRole("button", { name: "确认上传" }));
+    await waitFor(() => expect(toast.error).toHaveBeenCalled());
+    expect(onUpdate).not.toHaveBeenCalled();
+  });
+
+  it("超过 10MB 的文件被拒绝且不入暂存列表", async () => {
+    const { toast } = await import("sonner");
+    setup();
+    const big = new File([new ArrayBuffer(11 * 1024 * 1024)], "大文件.zip", { type: "application/zip" });
+    stageFile(big);
+    await waitFor(() => expect(toast.error).toHaveBeenCalled());
+    expect(fns.uploadFetch).not.toHaveBeenCalled();
+    expect(screen.queryByText("大文件.zip")).not.toBeInTheDocument();
+  });
+
+  it("取消按钮关闭弹窗", () => {
+    const { onOpenChange } = setup();
+    const file = new File(["hello"], "新文件.txt", { type: "text/plain" });
+    stageFile(file);
+    expect(screen.getByRole("button", { name: "移除" })).toBeInTheDocument();
+    fireEvent.click(screen.getByRole("button", { name: "取消" }));
+    expect(onOpenChange).toHaveBeenCalledWith(false);
+  });
+});

package/src/components/story/relations/__tests__/knowledge-select-dialog.test.tsx

@@ -0,0 +1,174 @@
+// 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, it, expect, vi, beforeEach } from "vitest";
+import { render, screen, fireEvent, waitFor } from "@testing-library/react";
+
+vi.mock("@tanstack/react-start", () => ({
+  useServerFn: (fn: any) => vi.fn().mockImplementation((args: any) => fn(args)),
+}));
+
+const fns = vi.hoisted(() => ({
+  listKbs: vi.fn(),
+  search: vi.fn(),
+  link: vi.fn(),
+  unlink: vi.fn(),
+}));
+
+vi.mock("@/server/serverFns/story", () => ({
+  listStoryAvailableKbsFn: fns.listKbs,
+  searchKbDocsFn: fns.search,
+  linkDocsFn: fns.link,
+  unlinkDocFn: fns.unlink,
+}));
+
+vi.mock("sonner", () => ({
+  toast: { success: vi.fn(), error: vi.fn(), warning: vi.fn() },
+}));
+
+import { KnowledgeSelectDialog } from "../knowledge-select-dialog";
+import type { DocBrief } from "../../types";
+
+const doc = (id: string, title: string, parentId: string | null = null): DocBrief => ({
+  id,
+  title,
+  parentId,
+  kbId: "kb1",
+  kbName: "团队库",
+  kbIcon: "notebook",
+  docIcon: null,
+});
+
+const LINKED: DocBrief[] = [doc("d1", "已关联页")];
+
+function setup() {
+  const onUpdate = vi.fn();
+  render(
+    <KnowledgeSelectDialog
+      open
+      onOpenChange={() => {}}
+      storyId="s1"
+      linkedDocs={LINKED}
+      onUpdate={onUpdate}
+    />,
+  );
+  return { onUpdate };
+}
+
+beforeEach(() => {
+  vi.clearAllMocks();
+  fns.listKbs.mockResolvedValue([
+    { id: "kb1", name: "团队库", icon: "notebook", visibility: "team" },
+    { id: "kb2", name: "通用库", icon: "book", visibility: "general" },
+  ]);
+  fns.search.mockResolvedValue({
+    items: [doc("d1", "已关联页"), doc("d2", "子页", "d1"), doc("d3", "普通页")],
+    nextCursor: null,
+  });
+});
+
+describe("KnowledgeSelectDialog", () => {
+  it("加载可用 KB 与页面列表，已关联项默认勾选且带 badge", async () => {
+    setup();
+    // KB 名称出现在 Popover 触发器和文档列表中，用含角色断言避免歧义
+    await waitFor(() => expect(screen.getByText("已关联页")).toBeInTheDocument());
+    expect(screen.getByText("已关联")).toBeInTheDocument();
+  });
+
+  it("切换勾选后确认：新增走 linkDocsFn、取消走 unlinkDocFn", async () => {
+    fns.link.mockResolvedValue({ linked: ["d2", "d3"], skipped: [], failed: [] });
+    fns.unlink.mockResolvedValue({ ok: true });
+    const { onUpdate } = setup();
+    await waitFor(() => screen.getByText("普通页"));
+    // 勾选 d3（新增）
+    const checkboxes = screen.getAllByRole("checkbox");
+    // d3 是第三个（已关联页、子页、普通页）
+    fireEvent.click(checkboxes[2]);
+    // 取消 d1
+    fireEvent.click(checkboxes[0]);
+    fireEvent.click(screen.getByRole("button", { name: "确认" }));
+    await waitFor(() => expect(fns.link).toHaveBeenCalledWith({
+      data: { storyId: "s1", docIds: expect.arrayContaining(["d3"]) },
+    }));
+    await waitFor(() => expect(fns.unlink).toHaveBeenCalledWith({
+      data: { storyId: "s1", docId: "d1" },
+    }));
+    await waitFor(() => expect(onUpdate).toHaveBeenCalled());
+  });
+
+  it("linkDocsFn 返回 failed 时 toast warning 且不关弹窗", async () => {
+    fns.link.mockResolvedValue({ linked: [], skipped: [], failed: [{ docId: "d9", reason: "outOfScope" }] });
+    const { toast } = await import("sonner");
+    setup();
+    await waitFor(() => screen.getByText("普通页"));
+    const checkboxes = screen.getAllByRole("checkbox");
+    fireEvent.click(checkboxes[2]);
+    fireEvent.click(screen.getByRole("button", { name: "确认" }));
+    await waitFor(() => expect(toast.warning).toHaveBeenCalled());
+  });
+
+  it("KB 选择弹层含我的知识库与通用知识库两个分组区（五个通用组）", async () => {
+    setup();
+    await waitFor(() => screen.getByText("普通页"));
+    fireEvent.click(screen.getByRole("button", { name: /全部知识库/ }));
+    expect(screen.getByText("我的知识库")).toBeInTheDocument();
+    expect(screen.getByText("通用知识库")).toBeInTheDocument();
+    for (const label of ["法律法规", "行业合规", "词典词条", "网站收藏", "周报月报"]) {
+      expect(screen.getByText(label)).toBeInTheDocument();
+    }
+  });
+
+  it("选择通用知识库分组后搜索带 generalGroup 参数", async () => {
+    setup();
+    await waitFor(() => screen.getByText("普通页"));
+    fireEvent.click(screen.getByRole("button", { name: /全部知识库/ }));
+    fireEvent.click(screen.getByText("法律法规"));
+    await waitFor(() =>
+      expect(fns.search).toHaveBeenCalledWith(
+        expect.objectContaining({ data: expect.objectContaining({ generalGroup: "legal" }) }),
+      ),
+    );
+  });
+
+  it("底部全选复选框支持全选与部分选中（indeterminate）状态", async () => {
+    setup();
+    await waitFor(() => screen.getByText("普通页"));
+    const footerCheckbox = screen.getByRole("checkbox", { name: "全选" });
+    // 初始：仅 d1 已勾选（部分选中）→ indeterminate
+    expect(footerCheckbox.getAttribute("aria-checked")).toBe("mixed");
+    // 全选
+    fireEvent.click(footerCheckbox);
+    const checkboxes = screen.getAllByRole("checkbox");
+    for (const cb of checkboxes) expect(cb.getAttribute("aria-checked")).toBe("true");
+    // 再次点击取消全选
+    fireEvent.click(footerCheckbox);
+    expect(footerCheckbox.getAttribute("aria-checked")).toBe("false");
+  });
+
+  it("点击行任意位置切换选中（勾选框点击不重复触发）", async () => {
+    setup();
+    await waitFor(() => screen.getByText("普通页"));
+    // 点击行文本区域 → 选中
+    fireEvent.click(screen.getByText("普通页"));
+    const checkboxes = screen.getAllByRole("checkbox");
+    // d3（普通页）对应第三个行勾选框
+    expect(checkboxes[2].getAttribute("aria-checked")).toBe("true");
+    // 再次点击行 → 取消选中
+    fireEvent.click(screen.getByText("普通页"));
+    expect(checkboxes[2].getAttribute("aria-checked")).toBe("false");
+  });
+});