ComfyUI_frontend/test/LGraph.test.ts

import { LGraph, LGraphGroup, LGraphNode, LiteGraph } from "../src/litegraph"
import { LiteGraphGlobal } from "../src/LiteGraphGlobal"

function makeGraph() {
  const LiteGraph = new LiteGraphGlobal()
  LiteGraph.registerNodeType("TestNode", LGraphNode)
  LiteGraph.registerNodeType("OtherNode", LGraphNode)
  LiteGraph.registerNodeType("", LGraphNode)
  return new LGraph()
}

describe("LGraph", () => {
  it("can be instantiated", () => {
    // @ts-ignore TODO: Remove once relative imports fix goes in.
    const graph = new LGraph({ extra: "TestGraph" })
    expect(graph).toBeInstanceOf(LGraph)
    expect(graph.extra).toBe("TestGraph")
  })
})

describe("Legacy LGraph Compatibility Layer", () => {
  it("can be extended via prototype", () => {
    const graph = new LGraph()
    // @ts-expect-error Should always be an error.
    LGraph.prototype.newMethod = function () {
      return "New method added via prototype"
    }
    // @ts-expect-error Should always be an error.
    expect(graph.newMethod()).toBe("New method added via prototype")
  })

  it("is correctly assigned to LiteGraph", () => {
    expect(LiteGraph.LGraph).toBe(LGraph)
  })
})

describe("LGraph Serialisation", () => {
  it("should serialise", () => {
    const graph = new LGraph()
    graph.add(new LGraphNode("Test Node"))
    graph.add(new LGraphGroup("Test Group"))
    expect(graph.nodes.length).toBe(1)
    expect(graph.groups.length).toBe(1)
  })
})