ComfyUI_frontend/browser_tests/tests/litegraphEvent.spec.ts

import { expect } from '@playwright/test'

import { comfyPageFixture as test } from '../fixtures/ComfyPage'

test.beforeEach(async ({ comfyPage }) => {
  await comfyPage.settings.setSetting('Comfy.UseNewMenu', 'Disabled')
})

function listenForEvent(): Promise<Event> {
  return new Promise<Event>((resolve) => {
    document.addEventListener('litegraph:canvas', (e) => resolve(e), {
      once: true
    })
  })
}

test.describe('Canvas Event', { tag: '@canvas' }, () => {
  test('Emit litegraph:canvas empty-release', async ({ comfyPage }) => {
    const eventPromise = comfyPage.page.evaluate(listenForEvent)
    const disconnectPromise = comfyPage.canvasOps.disconnectEdge()
    const event = await eventPromise
    await disconnectPromise

    expect(event).not.toBeNull()
    // No further check on event content as the content is dropped by
    // playwright for some reason.
    // See https://github.com/microsoft/playwright/issues/31580
  })

  test('Emit litegraph:canvas empty-double-click', async ({ comfyPage }) => {
    const eventPromise = comfyPage.page.evaluate(listenForEvent)
    const doubleClickPromise = comfyPage.canvasOps.doubleClick()
    const event = await eventPromise
    await doubleClickPromise

    expect(event).not.toBeNull()
    // No further check on event content as the content is dropped by
    // playwright for some reason.
    // See https://github.com/microsoft/playwright/issues/31580
  })
})