ComfyUI_frontend/browser_tests/tests/loadWorkflowInMedia.spec.ts

import { expect } from '@playwright/test'

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

test.describe('Load Workflow in Media', () => {
  const fileNames = [
    'workflow.webp',
    'edited_workflow.webp',
    'no_workflow.webp',
    'large_workflow.webp',
    'workflow.webm',
    'workflow.glb',
    'workflow.mp4',
    'workflow.mov',
    'workflow.m4v',
    'workflow.svg'
  ]
  fileNames.forEach(async (fileName) => {
    test(`Load workflow in ${fileName} (drop from filesystem)`, async ({
      comfyPage
    }) => {
      await comfyPage.dragAndDropFile(fileName)
      await expect(comfyPage.canvas).toHaveScreenshot(`${fileName}.png`)
    })
  })

  const urls = [
    'https://comfyanonymous.github.io/ComfyUI_examples/hidream/hidream_dev_example.png'
  ]
  urls.forEach(async (url) => {
    test(`Load workflow from URL${url} (drop from different browser tabs)`, async ({
      comfyPage
    }) => {
      await comfyPage.dragAndDropURL(url)
      const readableName = url.split('/').pop()
      await expect(comfyPage.canvas).toHaveScreenshot(
        `dropped_workflow_url_${readableName}.png`
      )
    })
  })
})