Convert i/o slot to class in LGraphNode.configure (#506)

This commit is contained in:
Chenlei Hu
2025-02-09 15:23:38 -05:00
committed by GitHub
parent 997d673bd9
commit c75157e86e
2 changed files with 59 additions and 27 deletions

View File

@@ -1,5 +1,6 @@
import { describe, expect } from "vitest"
import { LGraphNode } from "@/litegraph"
import { NodeInputSlot, NodeOutputSlot } from "@/NodeSlot"
import { lgTest } from "./lgTest"
describe("LGraphNode", () => {
@@ -13,4 +14,39 @@ describe("LGraphNode", () => {
expect(node.size).toEqual(new Float32Array([100, 100]))
expect(node.serialize().size).toEqual([100, 100])
})
lgTest("should configure inputs correctly", () => {
const node = new LGraphNode("TestNode")
node.configure({
id: 0,
inputs: [{ name: "TestInput", type: "number", link: null }],
})
expect(node.inputs.length).toEqual(1)
expect(node.inputs[0].name).toEqual("TestInput")
expect(node.inputs[0].link).toEqual(null)
expect(node.inputs[0]).instanceOf(NodeInputSlot)
// Should not override existing inputs
node.configure({ id: 1 })
expect(node.id).toEqual(1)
expect(node.inputs.length).toEqual(1)
})
lgTest("should configure outputs correctly", () => {
const node = new LGraphNode("TestNode")
node.configure({
id: 0,
outputs: [{ name: "TestOutput", type: "number", links: [] }],
})
expect(node.outputs.length).toEqual(1)
expect(node.outputs[0].name).toEqual("TestOutput")
expect(node.outputs[0].type).toEqual("number")
expect(node.outputs[0].links).toEqual([])
expect(node.outputs[0]).instanceOf(NodeOutputSlot)
// Should not override existing outputs
node.configure({ id: 1 })
expect(node.id).toEqual(1)
expect(node.outputs.length).toEqual(1)
})
})