mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-03-07 22:20:03 +00:00
bae1081a08
## Summary Fixes flakey test to only assert that the upload request is made with the correct URL ## Changes - **What** - Replace waitForResponse with waitForRequest for the no_workflow.webp upload test to only assert the request is initiated with the correct URL - Move request listener setup before the drag-drop action to avoid race conditions - Remove screenshot assertion for the upload case since the upload may not complete before the screenshot is taken ┆Issue is synchronized with this [Notion page](https://www.notion.so/PR-9488-fix-update-loadWorkflowInMedia-test-to-only-assert-upload-request-URL-31b6d73d365081f69a9aeb1095da7d60) by [Unito](https://www.unito.io)
72 lines
2.2 KiB
TypeScript
72 lines
2.2 KiB
TypeScript
import { expect } from '@playwright/test'
|
|
|
|
import { comfyPageFixture as test } from '../fixtures/ComfyPage'
|
|
|
|
test.beforeEach(async ({ comfyPage }) => {
|
|
await comfyPage.settings.setSetting('Comfy.UseNewMenu', 'Disabled')
|
|
})
|
|
|
|
test.describe(
|
|
'Load Workflow in Media',
|
|
{ tag: ['@screenshot', '@workflow'] },
|
|
() => {
|
|
const fileNames = [
|
|
'workflow.webp',
|
|
'edited_workflow.webp',
|
|
'no_workflow.webp',
|
|
'large_workflow.webp',
|
|
'workflow_prompt_parameters.png',
|
|
'workflow_itxt.png',
|
|
'workflow.webm',
|
|
// Skipped due to 3d widget unstable visual result.
|
|
// 3d widget shows grid after fully loaded.
|
|
// 'workflow.glb',
|
|
'workflow.mp4',
|
|
'workflow.mov',
|
|
'workflow.m4v',
|
|
'workflow.svg'
|
|
// TODO: Re-enable after fixing test asset to use core nodes only
|
|
// Currently opens missing nodes dialog which is outside scope of AVIF loading functionality
|
|
// 'workflow.avif'
|
|
]
|
|
const filesWithUpload = new Set(['no_workflow.webp'])
|
|
|
|
fileNames.forEach(async (fileName) => {
|
|
test(`Load workflow in ${fileName} (drop from filesystem)`, async ({
|
|
comfyPage
|
|
}) => {
|
|
const shouldUpload = filesWithUpload.has(fileName)
|
|
const uploadRequestPromise = shouldUpload
|
|
? comfyPage.page.waitForRequest((req) =>
|
|
req.url().includes('/upload/')
|
|
)
|
|
: null
|
|
|
|
await comfyPage.dragDrop.dragAndDropFile(`workflowInMedia/${fileName}`)
|
|
|
|
if (uploadRequestPromise) {
|
|
const request = await uploadRequestPromise
|
|
expect(request.url()).toContain('/upload/')
|
|
} else {
|
|
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.dragDrop.dragAndDropURL(url)
|
|
const readableName = url.split('/').pop()
|
|
await expect(comfyPage.canvas).toHaveScreenshot(
|
|
`dropped_workflow_url_${readableName}.png`
|
|
)
|
|
})
|
|
})
|
|
}
|
|
)
|