mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-11 14:02:37 +08:00
103 lines
3.2 KiB
JavaScript
103 lines
3.2 KiB
JavaScript
/// <reference path="../node_modules/@types/jest/index.d.ts" />
|
|
// @ts-check
|
|
|
|
const { start } = require("../utils");
|
|
const lg = require("../utils/litegraph");
|
|
|
|
beforeEach(() => {
|
|
lg.setup(global);
|
|
});
|
|
|
|
afterEach(() => {
|
|
lg.teardown(global);
|
|
jest.resetModules();
|
|
});
|
|
|
|
test("converted widget works after reload", async () => {
|
|
const { graph, ez } = await start();
|
|
let { $: n } = ez.CheckpointLoaderSimple();
|
|
|
|
const inputCount = n.inputs.length;
|
|
|
|
// Convert ckpt name to an input
|
|
n.widgets.ckpt_name.convertToInput();
|
|
expect(n.widgets.ckpt_name.isConvertedToInput).toBeTruthy();
|
|
expect(n.inputs.ckpt_name).toBeTruthy();
|
|
expect(n.inputs.length).toEqual(inputCount + 1);
|
|
|
|
// Convert back to widget and ensure input is removed
|
|
n.widgets.ckpt_name.convertToWidget();
|
|
expect(n.widgets.ckpt_name.isConvertedToInput).toBeFalsy();
|
|
expect(n.inputs.ckpt_name).toBeFalsy();
|
|
expect(n.inputs.length).toEqual(inputCount);
|
|
|
|
// Convert again and reload the graph to ensure it maintains state
|
|
n.widgets.ckpt_name.convertToInput();
|
|
expect(n.inputs.length).toEqual(inputCount + 1);
|
|
|
|
let { $: primitive } = ez.PrimitiveNode();
|
|
primitive.outputs[0].connectTo(n.inputs.ckpt_name);
|
|
|
|
await graph.reload();
|
|
|
|
// Find the reloaded nodes in the graph
|
|
n = graph.find(n);
|
|
primitive = graph.find(primitive);
|
|
|
|
// Ensure widget is converted
|
|
expect(n.widgets.ckpt_name.isConvertedToInput).toBeTruthy();
|
|
expect(n.inputs.ckpt_name).toBeTruthy();
|
|
expect(n.inputs.length).toEqual(inputCount + 1);
|
|
|
|
// Ensure primitive is connected
|
|
let { connections } = primitive.outputs[0];
|
|
expect(connections).toHaveLength(1);
|
|
expect(connections[0].targetNode.id).toBe(n.node.id);
|
|
|
|
// Disconnect & reconnect
|
|
connections[0].disconnect();
|
|
({ connections } = primitive.outputs[0]);
|
|
expect(connections).toHaveLength(0);
|
|
primitive.outputs[0].connectTo(n.inputs.ckpt_name);
|
|
|
|
({ connections } = primitive.outputs[0]);
|
|
expect(connections).toHaveLength(1);
|
|
expect(connections[0].targetNode.id).toBe(n.node.id);
|
|
|
|
// Convert back to widget and ensure input is removed
|
|
n.widgets.ckpt_name.convertToWidget();
|
|
expect(n.widgets.ckpt_name.isConvertedToInput).toBeFalsy();
|
|
expect(n.inputs.ckpt_name).toBeFalsy();
|
|
expect(n.inputs.length).toEqual(inputCount);
|
|
});
|
|
|
|
test("converted widget works on clone", async () => {
|
|
const { graph, ez } = await start();
|
|
let { $: n } = ez.CheckpointLoaderSimple();
|
|
|
|
// Convert the widget to an input
|
|
n.widgets.ckpt_name.convertToInput();
|
|
expect(n.widgets.ckpt_name.isConvertedToInput).toBeTruthy();
|
|
|
|
// Clone the node
|
|
n.menu["Clone"].call();
|
|
expect(graph.nodes).toHaveLength(2);
|
|
const clone = graph.nodes[1];
|
|
expect(clone.id).not.toEqual(n.id);
|
|
|
|
// Ensure the clone has an input
|
|
expect(clone.widgets.ckpt_name.isConvertedToInput).toBeTruthy();
|
|
expect(clone.inputs.ckpt_name).toBeTruthy();
|
|
|
|
// Ensure primitive connects to both nodes
|
|
let { $: primitive } = ez.PrimitiveNode();
|
|
primitive.outputs[0].connectTo(n.inputs.ckpt_name);
|
|
primitive.outputs[0].connectTo(clone.inputs.ckpt_name);
|
|
expect(primitive.outputs[0].connections).toHaveLength(2);
|
|
|
|
// Convert back to widget and ensure input is removed
|
|
clone.widgets.ckpt_name.convertToWidget();
|
|
expect(clone.widgets.ckpt_name.isConvertedToInput).toBeFalsy();
|
|
expect(clone.inputs.ckpt_name).toBeFalsy();
|
|
});
|