90d3fab483
Add ~330 tests across server (Jest), client (Vitest), editor-ext (Vitest)
and packages/mcp (node:test) for the gitmost features added since
053a9c0d: AI chat, AI agent roles, public-share assistant, MCP per-user
auth, HTML embed, page templates/embed, realtime tree, tree
expand/collapse, and the AI-settings UI.
Test-tooling fixes (prerequisite, were silently hiding coverage):
- Repair 3 page-template specs broken by the 11-arg TransclusionService
constructor; they never compiled, so template access-control / content
-leak / unsync-strip coverage was fictitious.
- Build @docmost/editor-ext before server tests via a `pretest` hook;
the stale dist omitted the new HtmlEmbed/PageEmbed exports (TS2305).
- Let jest resolve the .tsx email templates: add `tsx` to
moduleFileExtensions and widen the ts-jest transform to (t|j)sx?.
Behaviour-preserving "extract pure core" refactors that the tests drive:
- server: resolveShareAssistantRequest + uiMessageTextLength
(public-share controller), decideBasicGate + mapAuthResultToResponse
(mcp), buildErrorAssistantRecord (ai-chat), jsonbObject export (roles).
- client: render-raw-html + shouldExecute/canEdit, decide-embed-state,
page-embed picker utils, tree-socket reducers, open/close branch maps,
isEndpointConfigured/resolveKeyField; buildTreeWithChildren now treats
a permission-trimmed orphan as a root instead of crashing.
Deferred (need a test DB or HTTP harness, documented in the specs):
repo-level Postgres integration tests and the public-share XFF E2E.
Pre-existing DI/lib0-ESM suite failures are untouched and out of scope.
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
101 lines
2.9 KiB
TypeScript
101 lines
2.9 KiB
TypeScript
import { describe, it, expect } from "vitest";
|
|
import {
|
|
toolCitations,
|
|
toolRunState,
|
|
type ToolUiPart,
|
|
} from "./tool-parts";
|
|
|
|
describe("toolCitations", () => {
|
|
it("emits one citation per searchPages item with a /p/{id} href", () => {
|
|
const part: ToolUiPart = {
|
|
type: "tool-searchPages",
|
|
state: "output-available",
|
|
output: [
|
|
{ id: "p1", title: "First" },
|
|
{ id: "p2", title: "Second" },
|
|
],
|
|
};
|
|
expect(toolCitations(part)).toEqual([
|
|
{ pageId: "p1", title: "First", href: "/p/p1" },
|
|
{ pageId: "p2", title: "Second", href: "/p/p2" },
|
|
]);
|
|
});
|
|
|
|
it("drops searchPages items missing an id", () => {
|
|
const part: ToolUiPart = {
|
|
type: "tool-searchPages",
|
|
state: "output-available",
|
|
output: [{ title: "No id here" }, { id: "p2", title: "Kept" }],
|
|
};
|
|
expect(toolCitations(part)).toEqual([
|
|
{ pageId: "p2", title: "Kept", href: "/p/p2" },
|
|
]);
|
|
});
|
|
|
|
it("falls back to input.pageId / input.title for a page-op with only pageId", () => {
|
|
// The mutating tools echo `pageId` (no `id`); title is taken from the input.
|
|
const part: ToolUiPart = {
|
|
type: "tool-updatePageContent",
|
|
state: "output-available",
|
|
input: { pageId: "host-1", title: "From input" },
|
|
output: { pageId: "host-1" },
|
|
};
|
|
expect(toolCitations(part)).toEqual([
|
|
{ pageId: "host-1", title: "From input", href: "/p/host-1" },
|
|
]);
|
|
});
|
|
|
|
it("prefers output.id over input.pageId when both exist", () => {
|
|
const part: ToolUiPart = {
|
|
type: "tool-getPage",
|
|
state: "output-available",
|
|
input: { pageId: "input-id", title: "Input title" },
|
|
output: { id: "output-id", title: "Output title" },
|
|
};
|
|
expect(toolCitations(part)).toEqual([
|
|
{ pageId: "output-id", title: "Output title", href: "/p/output-id" },
|
|
]);
|
|
});
|
|
|
|
it("returns [] when the state is not output-available", () => {
|
|
const part: ToolUiPart = {
|
|
type: "tool-getPage",
|
|
state: "input-available",
|
|
output: { id: "p1", title: "Pending" },
|
|
};
|
|
expect(toolCitations(part)).toEqual([]);
|
|
});
|
|
|
|
it("returns [] for a page-op output with no resolvable id", () => {
|
|
const part: ToolUiPart = {
|
|
type: "tool-getPage",
|
|
state: "output-available",
|
|
input: {},
|
|
output: { title: "Only a title" },
|
|
};
|
|
expect(toolCitations(part)).toEqual([]);
|
|
});
|
|
});
|
|
|
|
describe("toolRunState", () => {
|
|
it('maps "output-error" to error', () => {
|
|
expect(toolRunState("output-error")).toBe("error");
|
|
});
|
|
|
|
it('maps "output-denied" to error', () => {
|
|
expect(toolRunState("output-denied")).toBe("error");
|
|
});
|
|
|
|
it('maps "output-available" to done', () => {
|
|
expect(toolRunState("output-available")).toBe("done");
|
|
});
|
|
|
|
it('maps "input-available" to running', () => {
|
|
expect(toolRunState("input-available")).toBe("running");
|
|
});
|
|
|
|
it("maps undefined to running", () => {
|
|
expect(toolRunState(undefined)).toBe("running");
|
|
});
|
|
});
|