mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-04-19 22:09:37 +00:00
5899a9392e
## Summary Simplifies test setup for common settings ## Changes - **What**: - add vue-nodes tag to auto enable nodes 2.0 - remove UseNewMenu Top as this is default ┆Issue is synchronized with this [Notion page](https://www.notion.so/PR-11184-test-Simplify-vue-node-menu-test-setup-3416d73d3650815487e0c357d28761fe) by [Unito](https://www.unito.io)
55 lines
1.6 KiB
TypeScript
55 lines
1.6 KiB
TypeScript
import { expect } from '@playwright/test'
|
|
|
|
import { comfyPageFixture as test } from '@e2e/fixtures/ComfyPage'
|
|
|
|
test.describe(
|
|
'Paste Image context menu option',
|
|
{ tag: ['@node', '@vue-nodes'] },
|
|
() => {
|
|
test('shows Paste Image in LoadImage node context menu', async ({
|
|
comfyPage
|
|
}) => {
|
|
await comfyPage.workflow.loadWorkflow('widgets/load_image_widget')
|
|
|
|
const loadImageNode = (
|
|
await comfyPage.nodeOps.getNodeRefsByType('LoadImage')
|
|
)[0]
|
|
|
|
const nodeEl = comfyPage.page.locator(
|
|
`[data-node-id="${loadImageNode.id}"]`
|
|
)
|
|
await nodeEl.click({ button: 'right' })
|
|
const menu = comfyPage.page.locator('.p-contextmenu')
|
|
await menu.waitFor({ state: 'visible' })
|
|
const menuLabels = await menu
|
|
.locator('[role="menuitem"] span.flex-1')
|
|
.allInnerTexts()
|
|
|
|
expect(menuLabels).toContain('Paste Image')
|
|
})
|
|
|
|
test('does not show Paste Image on output-only image nodes', async ({
|
|
comfyPage
|
|
}) => {
|
|
await comfyPage.workflow.loadWorkflow('nodes/single_save_image_node')
|
|
|
|
const saveImageNode = (
|
|
await comfyPage.nodeOps.getNodeRefsByType('SaveImage')
|
|
)[0]
|
|
|
|
const nodeEl = comfyPage.page.locator(
|
|
`[data-node-id="${saveImageNode.id}"]`
|
|
)
|
|
await nodeEl.click({ button: 'right' })
|
|
const menu = comfyPage.page.locator('.p-contextmenu')
|
|
await menu.waitFor({ state: 'visible' })
|
|
const menuLabels = await menu
|
|
.locator('[role="menuitem"] span.flex-1')
|
|
.allInnerTexts()
|
|
|
|
expect(menuLabels).not.toContain('Paste Image')
|
|
expect(menuLabels).not.toContain('Open Image')
|
|
})
|
|
}
|
|
)
|