@allbluecn/web-app 0.5.0 → 0.5.1

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");
+  });
+});

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

@@ -0,0 +1,93 @@
+// 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(() => ({
+  addLink: vi.fn(),
+  updateLink: vi.fn(),
+  removeLink: vi.fn(),
+}));
+
+vi.mock("@/server/serverFns/story", () => ({
+  addLinkFn: fns.addLink,
+  updateLinkFn: fns.updateLink,
+  removeLinkFn: fns.removeLink,
+}));
+
+vi.mock("sonner", () => ({
+  toast: { success: vi.fn(), error: vi.fn(), warning: vi.fn() },
+}));
+
+import { LinkManageDialog } from "../link-manage-dialog";
+import type { StoryLinkItem } from "../../types";
+
+const LINKS: StoryLinkItem[] = [
+  { id: "l1", url: "https://a.com", title: "站点A", description: null },
+  { id: "l2", url: "https://b.com", title: null, description: "备注" },
+];
+
+function setup(props?: Partial<Parameters<typeof LinkManageDialog>[0]>) {
+  const onUpdate = vi.fn();
+  render(
+    <LinkManageDialog
+      open
+      onOpenChange={() => {}}
+      storyId="s1"
+      data={LINKS}
+      onUpdate={onUpdate}
+      initialEdit={null}
+      {...props}
+    />,
+  );
+  return { onUpdate };
+}
+
+beforeEach(() => vi.clearAllMocks());
+
+describe("LinkManageDialog", () => {
+  it("不渲染已有链接列表，仅保留添加表单", () => {
+    setup();
+    expect(screen.queryByText("站点A")).not.toBeInTheDocument();
+    expect(screen.getByLabelText(/URL/i)).toBeInTheDocument();
+  });
+
+  it("填写 URL 提交调用 addLinkFn 并回调 onUpdate", async () => {
+    fns.addLink.mockResolvedValue({ id: "l3", url: "https://c.com", title: "C", description: null });
+    const { onUpdate } = setup();
+    fireEvent.change(screen.getByLabelText(/URL/i), { target: { value: "https://c.com" } });
+    fireEvent.click(screen.getByRole("button", { name: "添加" }));
+    await waitFor(() => expect(onUpdate).toHaveBeenCalled());
+    const call = fns.addLink.mock.calls[0][0];
+    expect(call.data.storyId).toBe("s1");
+    expect(call.data.url).toBe("https://c.com");
+  });
+
+  it("initialEdit 进入编辑态，保存调 updateLinkFn", async () => {
+    fns.updateLink.mockResolvedValue({});
+    setup({ initialEdit: LINKS[0] });
+    expect(screen.getByRole("button", { name: "保存" })).toBeInTheDocument();
+    fireEvent.click(screen.getByRole("button", { name: "保存" }));
+    await waitFor(() => expect(fns.updateLink).toHaveBeenCalled());
+    expect(fns.updateLink.mock.calls[0][0].data.linkId).toBe("l1");
+  });
+});