ComfyUI_frontend/browser_tests/nodeSearchBox.spec.ts

import { expect } from "@playwright/test";
import { comfyPageFixture as test } from "./ComfyPage";

test.describe("Node search box", () => {
  test("Can trigger on empty canvas double click", async ({ comfyPage }) => {
    await comfyPage.doubleClickCanvas();
    await expect(comfyPage.searchBox.input).toHaveCount(1);
  });

  test("Can trigger on link release", async ({ comfyPage }) => {
    await comfyPage.page.keyboard.down("Shift");
    await comfyPage.disconnectEdge();
    await expect(comfyPage.searchBox.input).toHaveCount(1);
  });

  test("Does not trigger on link release (no shift)", async ({ comfyPage }) => {
    await comfyPage.disconnectEdge();
    await expect(comfyPage.searchBox.input).toHaveCount(0);
  });

  test("Can add node", async ({ comfyPage }) => {
    await comfyPage.doubleClickCanvas();
    await expect(comfyPage.searchBox.input).toHaveCount(1);
    await comfyPage.searchBox.fillAndSelectFirstNode("KSampler");
    await expect(comfyPage.canvas).toHaveScreenshot("added-node.png");
  });

  test("Can auto link node", async ({ comfyPage }) => {
    await comfyPage.page.keyboard.down("Shift");
    await comfyPage.disconnectEdge();
    await comfyPage.page.keyboard.up("Shift");
    await comfyPage.searchBox.fillAndSelectFirstNode("CLIPTextEncode");
    await expect(comfyPage.canvas).toHaveScreenshot("auto-linked-node.png");
  });
});