64 lines
4.1 KiB
TypeScript
64 lines
4.1 KiB
TypeScript
import { describe, expect, test } from "bun:test";
|
|
import { readFileSync } from "node:fs";
|
|
import { resolve } from "node:path";
|
|
import { createPortraitTextureNode, isCanvasImageSourceNode } from "../src/lib/canvas/canvas-image-source";
|
|
import { canGenerateImageInPlace } from "../src/lib/canvas/canvas-generation-layout";
|
|
import { CanvasNodeType, type CanvasNodeData } from "../src/types/canvas";
|
|
|
|
const source: CanvasNodeData = {
|
|
id: "uploaded", type: CanvasNodeType.Image, title: "原图", position: { x: 10, y: 20 }, width: 300, height: 200,
|
|
metadata: { content: "original-image", storageKey: "original-key", status: "success" },
|
|
};
|
|
const read = (path: string) => readFileSync(resolve(import.meta.dir, "../src", path), "utf8");
|
|
|
|
describe("上传图片是输入素材", () => {
|
|
test("上传与导入图片没有生成框，空节点与生成结果仍可配置", () => {
|
|
expect(isCanvasImageSourceNode(source)).toBe(true);
|
|
expect(isCanvasImageSourceNode({ ...source, metadata: {} })).toBe(false);
|
|
expect(isCanvasImageSourceNode({ ...source, metadata: { ...source.metadata, generationType: "edit" } })).toBe(false);
|
|
expect(isCanvasImageSourceNode({ ...source, metadata: { ...source.metadata, taskId: "task" } })).toBe(false);
|
|
expect(isCanvasImageSourceNode({ ...source, type: CanvasNodeType.Video })).toBe(false);
|
|
expect(isCanvasImageSourceNode(null)).toBe(false);
|
|
});
|
|
test("显式复制/版本生成入口保持可用", () => {
|
|
expect(isCanvasImageSourceNode({ ...source, metadata: { ...source.metadata, generationResultPlacement: "replace-node" } })).toBe(false);
|
|
expect(isCanvasImageSourceNode({ ...source, metadata: { ...source.metadata, copiedFromNodeId: "original" } })).toBe(false);
|
|
});
|
|
test("人物质感创建独立编辑节点，不复制原图输出和任务状态", () => {
|
|
const before = structuredClone(source);
|
|
const child = createPortraitTextureNode(source, "texture");
|
|
expect(source).toEqual(before);
|
|
expect(child.id).toBe("texture");
|
|
expect(child.position).toEqual({ x: 406, y: 20 });
|
|
expect(child.metadata?.composerContent).toBe("@图片1");
|
|
expect(child.metadata?.generationType).toBe("edit");
|
|
expect(child.metadata?.content).toBeUndefined();
|
|
expect(child.metadata?.storageKey).toBeUndefined();
|
|
expect(child.metadata?.taskId).toBeUndefined();
|
|
expect(isCanvasImageSourceNode(child)).toBe(false);
|
|
expect(canGenerateImageInPlace(child)).toBe(true);
|
|
expect(canGenerateImageInPlace(source)).toBe(false);
|
|
});
|
|
test("页面和工具入口共享素材判定，人物质感接入原图连线", () => {
|
|
expect(read("pages/canvas/project.tsx")).toContain("dialogNode && !isCanvasImageSourceNode(dialogNode)");
|
|
expect(read("lib/canvas/tool-registry/definitions/node-hover-tools.tsx")).toContain("!isCanvasImageSourceNode(ctx.node)");
|
|
const mediaTools = read("pages/canvas/use-canvas-media-tools.ts");
|
|
const portrait = mediaTools.slice(mediaTools.indexOf("const openPortraitTextureEditor"), mediaTools.indexOf("const cropImageNode"));
|
|
expect(portrait).toContain("fromNodeId: node.id, toNodeId: child.id");
|
|
expect(portrait).toContain("setDialogNodeId(child.id)");
|
|
expect(portrait).not.toContain("current.map");
|
|
expect(mediaTools).toContain("payload.generationConfig");
|
|
expect(mediaTools).toContain("imageBatchExpanded: requestedCount > 1 ? true : undefined");
|
|
const dialog = read("components/canvas/canvas-node-mask-edit-dialog.tsx");
|
|
expect(dialog).toContain("ModelPicker");
|
|
expect(dialog).toContain("ImageSettingsPanel");
|
|
});
|
|
});
|
|
|
|
test("底部菜单使用画布浮层管理并响应打开、鼠标和键盘交互", () => {
|
|
const toolbar = read("components/canvas/canvas-toolbar.tsx");
|
|
expect(toolbar).toContain('useCanvasOverlayLayer("main-toolbar", "var(--z-toolbar)")');
|
|
expect(toolbar).toContain("if (addOpen || appearanceOpen) bringToFront()");
|
|
expect(toolbar).toContain("style={{ zIndex }} onPointerDownCapture={bringToFront} onFocusCapture={bringToFront}");
|
|
});
|