ComfyUI_frontend/tests-ui/tests/nodeSearchService.test.ts

import { NodeSearchService } from "@/services/nodeSearchService";
import { ComfyNodeDef } from "@/types/apiTypes";

const EXAMPLE_NODE_DEFS: ComfyNodeDef[] = [
  {
    input: {
      required: {
        ckpt_name: [["model1.safetensors", "model2.ckpt"]],
      },
    },
    output: ["MODEL", "CLIP", "VAE"],
    output_is_list: [false, false, false],
    output_name: ["MODEL", "CLIP", "VAE"],
    name: "CheckpointLoaderSimple",
    display_name: "Load Checkpoint",
    description: "",
    python_module: "nodes",
    category: "loaders",
    output_node: false,
  },
  {
    input: {
      required: {
        samples: ["LATENT"],
        batch_index: [
          "INT",
          {
            default: 0,
            min: 0,
            max: 63,
          },
        ],
        length: [
          "INT",
          {
            default: 1,
            min: 1,
            max: 64,
          },
        ],
      },
    },
    output: ["LATENT"],
    output_is_list: [false],
    output_name: ["LATENT"],
    name: "LatentFromBatch",
    display_name: "Latent From Batch",
    description: "",
    python_module: "nodes",
    category: "latent/batch",
    output_node: false,
  },
];

describe("nodeSearchService", () => {
  it("searches with input filter", () => {
    const service = new NodeSearchService(EXAMPLE_NODE_DEFS);
    const inputFilter = service.getFilterById("input");
    expect(service.searchNode("L", [[inputFilter, "LATENT"]])).toHaveLength(1);
    expect(service.searchNode("L")).toHaveLength(2);
  });
});