35 lines
1.0 KiB
TypeScript
35 lines
1.0 KiB
TypeScript
|
import { describe, expect, it } from "vitest";
|
||
|
|
import { collectSubtree } from "../../convex/_utils/documentTree";
|
||
|
|
|
||
|
|
describe("collectSubtree", () => {
|
||
|
|
it("会收集根节点及其所有后代", () => {
|
||
|
|
const rows = [
|
||
|
|
{ id: "A", parent_id: null },
|
||
|
|
{ id: "B", parent_id: "A" },
|
||
|
|
{ id: "C", parent_id: "B" },
|
||
|
|
{ id: "D", parent_id: "A" },
|
||
|
|
{ id: "E", parent_id: null },
|
||
|
|
] as const;
|
||
|
|
|
||
|
|
const subtree = collectSubtree(rows, "A");
|
||
|
|
expect(new Set(subtree.map((r) => r.id))).toEqual(new Set(["A", "B", "C", "D"]));
|
||
|
|
});
|
||
|
|
|
||
|
|
it("根节点不存在时返回空数组", () => {
|
||
|
|
const rows = [{ id: "A", parent_id: null }] as const;
|
||
|
|
expect(collectSubtree(rows, "missing")).toEqual([]);
|
||
|
|
});
|
||
|
|
|
||
|
|
it("存在环时不会死循环", () => {
|
||
|
|
const rows = [
|
||
|
|
{ id: "A", parent_id: "B" },
|
||
|
|
{ id: "B", parent_id: "A" },
|
||
|
|
{ id: "C", parent_id: null },
|
||
|
|
] as const;
|
||
|
|
|
||
|
|
const subtree = collectSubtree(rows, "A");
|
||
|
|
expect(new Set(subtree.map((r) => r.id))).toEqual(new Set(["A", "B"]));
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|