mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-04-19 22:09:37 +00:00
897cf9cb8f
CC @christian-byrne ┆Issue is synchronized with this [Notion page](https://www.notion.so/PR-10572-Mark-failing-test-as-in-need-of-fixing-32f6d73d3650815cba72d19ebc54b0b3) by [Unito](https://www.unito.io)
111 lines
3.4 KiB
TypeScript
111 lines
3.4 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`
|
|
)
|
|
})
|
|
})
|
|
|
|
test.fixme('Load workflow from URL dropped onto Vue node', async ({
|
|
comfyPage
|
|
}) => {
|
|
const fakeUrl = 'https://example.com/workflow.png'
|
|
await comfyPage.page.route(fakeUrl, (route) =>
|
|
route.fulfill({
|
|
path: comfyPage.assetPath('workflowInMedia/workflow_itxt.png')
|
|
})
|
|
)
|
|
|
|
await comfyPage.settings.setSetting('Comfy.VueNodes.Enabled', true)
|
|
await comfyPage.vueNodes.waitForNodes()
|
|
|
|
const initialNodeCount = await comfyPage.nodeOps.getGraphNodesCount()
|
|
|
|
const node = comfyPage.vueNodes.getNodeByTitle('KSampler')
|
|
const box = await node.boundingBox()
|
|
expect(box).not.toBeNull()
|
|
|
|
const dropPosition = {
|
|
x: box!.x + box!.width / 2,
|
|
y: box!.y + box!.height / 2
|
|
}
|
|
|
|
await comfyPage.dragDrop.dragAndDropURL(fakeUrl, {
|
|
dropPosition,
|
|
preserveNativePropagation: true
|
|
})
|
|
|
|
await comfyPage.page.waitForFunction(
|
|
(prevCount) => window.app!.graph.nodes.length !== prevCount,
|
|
initialNodeCount,
|
|
{ timeout: 10000 }
|
|
)
|
|
|
|
const newNodeCount = await comfyPage.nodeOps.getGraphNodesCount()
|
|
expect(newNodeCount).not.toBe(initialNodeCount)
|
|
})
|
|
}
|
|
)
|