test: 接入 jsdom 测试栈, 把 15 个 React hooks 纳入 80% 覆盖率门禁
新增 jsdom + @testing-library/react, 为 lib/** 全部 use*.ts hooks 写 renderHook 单测: 生成/CRUD(world/character/generator/outline/accept/foreshadow/rules)、 SSE流(draftStream/reviewStream)、定时轮询(autosave/jobPoll/kimiOauth)、 文风与注入(refine/styleLearn/injection)。覆盖率 mock api 客户端+Toast, 不打真实 LLM。 vitest.config.ts 去掉 use*.ts 排除; 前端覆盖率 63%→95%。CLAUDE.md 同步说明。
This commit is contained in:
Yaojia Wang
125
apps/web/lib/style/useRefine.test.ts
Normal file
125
apps/web/lib/style/useRefine.test.ts
Normal file
@@ -0,0 +1,125 @@
|
||||
// @vitest-environment jsdom
|
||||
import { act, renderHook } from "@testing-library/react";
|
||||
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
||||
|
||||
import { useRefine } from "./useRefine";
|
||||
|
||||
// 后端客户端与 Toast 是 hook 的外部副作用边界，单测一律 mock。
|
||||
const post = vi.fn();
|
||||
const toast = vi.fn();
|
||||
vi.mock("@/lib/api/client", () => ({ api: { POST: (...a: unknown[]) => post(...a) } }));
|
||||
vi.mock("@/components/Toast", () => ({ useToast: () => toast }));
|
||||
|
||||
describe("useRefine", () => {
|
||||
beforeEach(() => {
|
||||
post.mockReset();
|
||||
toast.mockReset();
|
||||
});
|
||||
afterEach(() => vi.clearAllMocks());
|
||||
|
||||
it("初始为 idle、无结果", () => {
|
||||
const { result } = renderHook(() => useRefine());
|
||||
expect(result.current.status).toBe("idle");
|
||||
expect(result.current.result).toBeNull();
|
||||
});
|
||||
|
||||
it("回炉成功：status=done 并返回新旧 diff", async () => {
|
||||
post.mockResolvedValue({
|
||||
data: { original: "旧文", refined: "新文" },
|
||||
error: null,
|
||||
});
|
||||
|
||||
const { result } = renderHook(() => useRefine());
|
||||
let outcome: unknown;
|
||||
await act(async () => {
|
||||
outcome = await result.current.refine("p1", 3, "旧文", "更紧凑");
|
||||
});
|
||||
|
||||
expect(outcome).toEqual({ original: "旧文", refined: "新文" });
|
||||
expect(result.current.status).toBe("done");
|
||||
expect(result.current.result).toEqual({ original: "旧文", refined: "新文" });
|
||||
expect(toast).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it("成功路径以 trim 后的段提交、可省略空指令", async () => {
|
||||
post.mockResolvedValue({
|
||||
data: { original: "段", refined: "改" },
|
||||
error: null,
|
||||
});
|
||||
|
||||
const { result } = renderHook(() => useRefine());
|
||||
await act(async () => {
|
||||
await result.current.refine("p1", 1, " 段 ");
|
||||
});
|
||||
|
||||
expect(post).toHaveBeenCalledWith(
|
||||
"/projects/{project_id}/chapters/{chapter_no}/refine",
|
||||
expect.objectContaining({ body: { segment: "段" } }),
|
||||
);
|
||||
});
|
||||
|
||||
it("LLM_UNAVAILABLE：status=error 且提示去设置页", async () => {
|
||||
post.mockResolvedValue({
|
||||
data: null,
|
||||
error: { error: { code: "LLM_UNAVAILABLE" } },
|
||||
});
|
||||
|
||||
const { result } = renderHook(() => useRefine());
|
||||
let outcome: unknown;
|
||||
await act(async () => {
|
||||
outcome = await result.current.refine("p1", 2, "段");
|
||||
});
|
||||
|
||||
expect(outcome).toBeNull();
|
||||
expect(result.current.status).toBe("error");
|
||||
expect(toast).toHaveBeenCalledWith(
|
||||
"未配置提供商，请先去设置页连一家。",
|
||||
"error",
|
||||
);
|
||||
});
|
||||
|
||||
it("其余后端错误：status=error 且弹通用回炉失败 toast", async () => {
|
||||
post.mockResolvedValue({
|
||||
data: null,
|
||||
error: { error: { code: "INTERNAL" } },
|
||||
});
|
||||
|
||||
const { result } = renderHook(() => useRefine());
|
||||
await act(async () => {
|
||||
await result.current.refine("p1", 2, "段");
|
||||
});
|
||||
|
||||
expect(result.current.status).toBe("error");
|
||||
expect(toast).toHaveBeenCalledWith("回炉失败，请稍后重试。", "error");
|
||||
});
|
||||
|
||||
it("请求抛异常：status=error 且弹网络异常 toast", async () => {
|
||||
post.mockRejectedValue(new Error("network down"));
|
||||
|
||||
const { result } = renderHook(() => useRefine());
|
||||
let outcome: unknown;
|
||||
await act(async () => {
|
||||
outcome = await result.current.refine("p1", 2, "段");
|
||||
});
|
||||
|
||||
expect(outcome).toBeNull();
|
||||
expect(result.current.status).toBe("error");
|
||||
expect(toast).toHaveBeenCalledWith("回炉请求异常，请检查网络。", "error");
|
||||
});
|
||||
|
||||
it("reset 清回 idle 与空结果", async () => {
|
||||
post.mockResolvedValue({
|
||||
data: { original: "a", refined: "b" },
|
||||
error: null,
|
||||
});
|
||||
|
||||
const { result } = renderHook(() => useRefine());
|
||||
await act(async () => {
|
||||
await result.current.refine("p1", 1, "a");
|
||||
});
|
||||
act(() => result.current.reset());
|
||||
|
||||
expect(result.current.status).toBe("idle");
|
||||
expect(result.current.result).toBeNull();
|
||||
});
|
||||
});
|
||||
@@ -1,4 +1,10 @@
|
||||
import { describe, expect, it } from "vitest";
|
||||
// @vitest-environment jsdom
|
||||
import { act, renderHook } from "@testing-library/react";
|
||||
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
||||
|
||||
import { learnSummary, useStyleLearn } from "./useStyleLearn";
|
||||
import type { JobView } from "@/lib/jobs/job";
|
||||
import type { UseJobPoll } from "@/lib/jobs/useJobPoll";
|
||||
|
||||
// P1-6 回归守卫：学文风轮询完成（done 边沿）时，effect 必须用「最近一次 learn 传入的
|
||||
// projectId」去拉指纹，而非闭包捕获的旧值。useStyleLearn 用 ref 追踪 projectId 来保证这点。
|
||||
@@ -39,3 +45,239 @@ describe("useStyleLearn projectId 追踪（P1-6 stale-closure 守卫）", () =>
|
||||
expect(t.onDone()).toBeNull();
|
||||
});
|
||||
});
|
||||
|
||||
// ── 真实 hook 覆盖：renderHook 驱动编排（POST 受理 → 轮询 → done 拉指纹）─────────
|
||||
const post = vi.fn();
|
||||
const get = vi.fn();
|
||||
const toast = vi.fn();
|
||||
vi.mock("@/lib/api/client", () => ({
|
||||
api: {
|
||||
POST: (...a: unknown[]) => post(...a),
|
||||
GET: (...a: unknown[]) => get(...a),
|
||||
},
|
||||
}));
|
||||
vi.mock("@/components/Toast", () => ({ useToast: () => toast }));
|
||||
|
||||
const STYLE_PATH = "/projects/{project_id}/style";
|
||||
const JOBS_PATH = "/jobs/{job_id}";
|
||||
|
||||
interface GetOpts {
|
||||
params: { path: { project_id?: string; job_id?: string } };
|
||||
}
|
||||
interface PostOpts {
|
||||
params: { path: { project_id: string } };
|
||||
}
|
||||
|
||||
// 微任务 + 0ms 定时器全部冲洗（轮询 tick→dispatch→effect→refetch 链需多跳）。
|
||||
async function flush(): Promise<void> {
|
||||
await act(async () => {
|
||||
await vi.advanceTimersByTimeAsync(0);
|
||||
});
|
||||
}
|
||||
|
||||
const FINGERPRINT_BODY = {
|
||||
dimensions: [{ name: "节奏", value: "快", evidence: ["短句"] }],
|
||||
version: 2,
|
||||
};
|
||||
|
||||
describe("useStyleLearn 真实 hook 编排", () => {
|
||||
beforeEach(() => {
|
||||
vi.useFakeTimers();
|
||||
post.mockReset();
|
||||
get.mockReset();
|
||||
toast.mockReset();
|
||||
});
|
||||
afterEach(() => {
|
||||
vi.clearAllMocks();
|
||||
vi.useRealTimers();
|
||||
});
|
||||
|
||||
it("初始：未提交则 busy=false、pollStatus=idle、指纹为传入初值", () => {
|
||||
const initial = { dimensions: [], version: 1 };
|
||||
const { result } = renderHook(() => useStyleLearn(initial));
|
||||
|
||||
expect(result.current.busy).toBe(false);
|
||||
expect(result.current.pollStatus).toBe("idle");
|
||||
expect(result.current.progress).toBe(0);
|
||||
expect(result.current.fingerprint).toBe(initial);
|
||||
});
|
||||
|
||||
it("受理失败（通用错误）：返回 false、弹通用 toast、busy 归位", async () => {
|
||||
post.mockResolvedValue({ data: null, error: { error: { code: "INTERNAL" } } });
|
||||
|
||||
const { result } = renderHook(() => useStyleLearn(null));
|
||||
let ok: boolean | undefined;
|
||||
await act(async () => {
|
||||
ok = await result.current.learn("p1", ["样本"], "create");
|
||||
});
|
||||
|
||||
expect(ok).toBe(false);
|
||||
expect(toast).toHaveBeenCalledWith("学文风受理失败，请稍后重试。", "error");
|
||||
expect(result.current.busy).toBe(false);
|
||||
expect(get).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it("受理失败（LLM_UNAVAILABLE）：提示去设置页连一家", async () => {
|
||||
post.mockResolvedValue({
|
||||
data: null,
|
||||
error: { error: { code: "LLM_UNAVAILABLE" } },
|
||||
});
|
||||
|
||||
const { result } = renderHook(() => useStyleLearn(null));
|
||||
let ok: boolean | undefined;
|
||||
await act(async () => {
|
||||
ok = await result.current.learn("p1", ["样本"], "create");
|
||||
});
|
||||
|
||||
expect(ok).toBe(false);
|
||||
expect(toast).toHaveBeenCalledWith(
|
||||
"未配置提供商，请先去设置页连一家。",
|
||||
"error",
|
||||
);
|
||||
});
|
||||
|
||||
it("受理成功 → 轮询 done → 拉最新指纹并弹成功 toast", async () => {
|
||||
post.mockResolvedValue({ data: { job_id: "job-1" }, error: null });
|
||||
get.mockImplementation((path: string, opts: GetOpts) => {
|
||||
if (path === JOBS_PATH) {
|
||||
return Promise.resolve({
|
||||
data: { id: "job-1", status: "done", progress: 100 },
|
||||
error: null,
|
||||
});
|
||||
}
|
||||
return Promise.resolve({ data: FINGERPRINT_BODY, error: null });
|
||||
});
|
||||
|
||||
const { result } = renderHook(() => useStyleLearn(null));
|
||||
let ok: boolean | undefined;
|
||||
await act(async () => {
|
||||
ok = await result.current.learn("p1", ["样本"], "create");
|
||||
});
|
||||
await flush();
|
||||
|
||||
expect(ok).toBe(true);
|
||||
expect(result.current.pollStatus).toBe("done");
|
||||
expect(result.current.fingerprint).toEqual({
|
||||
dimensions: [{ name: "节奏", value: "快", evidence: ["短句"] }],
|
||||
version: 2,
|
||||
});
|
||||
expect(toast).toHaveBeenCalledWith("文风指纹已更新。", "success");
|
||||
});
|
||||
|
||||
it("done 但拉指纹失败（GET /style error）：指纹不变、仍弹成功 toast", async () => {
|
||||
post.mockResolvedValue({ data: { job_id: "job-1" }, error: null });
|
||||
get.mockImplementation((path: string) => {
|
||||
if (path === JOBS_PATH) {
|
||||
return Promise.resolve({
|
||||
data: { id: "job-1", status: "done", progress: 100 },
|
||||
error: null,
|
||||
});
|
||||
}
|
||||
return Promise.resolve({ data: null, error: { detail: "404" } });
|
||||
});
|
||||
|
||||
const initial = { dimensions: [], version: 9 };
|
||||
const { result } = renderHook(() => useStyleLearn(initial));
|
||||
await act(async () => {
|
||||
await result.current.learn("p1", ["样本"], "create");
|
||||
});
|
||||
await flush();
|
||||
|
||||
expect(result.current.fingerprint).toBe(initial);
|
||||
expect(toast).toHaveBeenCalledWith("文风指纹已更新。", "success");
|
||||
});
|
||||
|
||||
it("轮询失败（job failed）：弹学文风失败 toast 带原因", async () => {
|
||||
post.mockResolvedValue({ data: { job_id: "job-1" }, error: null });
|
||||
get.mockImplementation((path: string) => {
|
||||
if (path === JOBS_PATH) {
|
||||
return Promise.resolve({
|
||||
data: { id: "job-1", status: "failed", progress: 40, error: "模型超时" },
|
||||
error: null,
|
||||
});
|
||||
}
|
||||
return Promise.resolve({ data: FINGERPRINT_BODY, error: null });
|
||||
});
|
||||
|
||||
const { result } = renderHook(() => useStyleLearn(null));
|
||||
await act(async () => {
|
||||
await result.current.learn("p1", ["样本"], "create");
|
||||
});
|
||||
await flush();
|
||||
|
||||
expect(result.current.pollStatus).toBe("error");
|
||||
expect(toast).toHaveBeenCalledWith("学文风失败：模型超时", "error");
|
||||
});
|
||||
|
||||
it("切项目：done 边沿用最近一次 learn 的 projectId 拉指纹（stale-closure 守卫）", async () => {
|
||||
const styleProjectIds: string[] = [];
|
||||
post.mockImplementation((_path: string, opts: PostOpts) => {
|
||||
const pid = opts.params.path.project_id;
|
||||
return Promise.resolve({
|
||||
data: { job_id: pid === "project-A" ? "job-A" : "job-B" },
|
||||
error: null,
|
||||
});
|
||||
});
|
||||
get.mockImplementation((path: string, opts: GetOpts) => {
|
||||
if (path === JOBS_PATH) {
|
||||
const jobId = opts.params.path.job_id;
|
||||
// A 仍在跑（排队 setTimeout 下一拍）；B 直接 done。
|
||||
const status = jobId === "job-A" ? "running" : "done";
|
||||
return Promise.resolve({
|
||||
data: { id: jobId, status, progress: status === "done" ? 100 : 30 },
|
||||
error: null,
|
||||
});
|
||||
}
|
||||
styleProjectIds.push(opts.params.path.project_id ?? "?");
|
||||
return Promise.resolve({ data: FINGERPRINT_BODY, error: null });
|
||||
});
|
||||
|
||||
const { result } = renderHook(() => useStyleLearn(null));
|
||||
await act(async () => {
|
||||
await result.current.learn("project-A", ["a"], "create");
|
||||
});
|
||||
await act(async () => {
|
||||
await result.current.learn("project-B", ["b"], "update");
|
||||
});
|
||||
await flush();
|
||||
|
||||
// done 边沿拉指纹用的应是后切的 project-B，而非陈旧的 project-A。
|
||||
expect(styleProjectIds).toContain("project-B");
|
||||
expect(styleProjectIds).not.toContain("project-A");
|
||||
expect(toast).toHaveBeenCalledWith("文风指纹已更新。", "success");
|
||||
});
|
||||
});
|
||||
|
||||
// learnSummary 是导出的纯函数：done 且有 job 时回显版本/维度数，否则 null。
|
||||
describe("learnSummary", () => {
|
||||
const baseJob: JobView = {
|
||||
id: "j",
|
||||
kind: "style_learn",
|
||||
status: "done",
|
||||
progress: 100,
|
||||
result: { version: 3, dims_count: 16 },
|
||||
error: null,
|
||||
};
|
||||
const pollOf = (over: Partial<UseJobPoll>): UseJobPoll =>
|
||||
({
|
||||
status: "done",
|
||||
progress: 100,
|
||||
job: baseJob,
|
||||
error: null,
|
||||
poll: vi.fn(),
|
||||
reset: vi.fn(),
|
||||
...over,
|
||||
}) as UseJobPoll;
|
||||
|
||||
it("done 且有 job：回显 version 与 dimsCount", () => {
|
||||
expect(learnSummary(pollOf({}))).toEqual({ version: 3, dimsCount: 16 });
|
||||
});
|
||||
|
||||
it("非 done：返回 null", () => {
|
||||
expect(learnSummary(pollOf({ status: "polling" }))).toBeNull();
|
||||
});
|
||||
|
||||
it("done 但无 job：返回 null", () => {
|
||||
expect(learnSummary(pollOf({ job: null }))).toBeNull();
|
||||
});
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user