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

test.describe('Load Workflow in Media', () => {
  ;['workflow.webp', 'edited_workflow.webp', 'no_workflow.webp'].forEach(
    async (fileName) => {
      test(`Load workflow in ${fileName}`, async ({ comfyPage }) => {
        await comfyPage.dragAndDropFile(fileName)
        await expect(comfyPage.canvas).toHaveScreenshot(`${fileName}.png`)
      })
    }
  )
})