mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-02-09 01:20:09 +00:00
71 lines
2.3 KiB
TypeScript
71 lines
2.3 KiB
TypeScript
import {
|
|
comfyPageFixture as test,
|
|
comfyExpect as expect
|
|
} from './fixtures/ComfyPage'
|
|
|
|
// Old `nodeTemplate.ts` system
|
|
test.describe('Node Template', () => {
|
|
test.afterEach(async ({ comfyPage }) => {
|
|
await comfyPage.clearNodeTemplates()
|
|
})
|
|
|
|
test('Can create and use node template', async ({ comfyPage }) => {
|
|
const templateName = 'Can create node template template'
|
|
|
|
await comfyPage.clearNodeTemplates()
|
|
await comfyPage.reload()
|
|
|
|
// TODO: Flaky test. Right click requires delay after reload, but other interactions do not.
|
|
await comfyPage.page.waitForTimeout(500)
|
|
|
|
// Enter filename when prompt dialog shown
|
|
comfyPage.page.on('dialog', (dialog) => dialog.accept(templateName))
|
|
|
|
// Ctrl + drag over 3 nodes
|
|
await comfyPage.dragAndDrop(
|
|
{ x: 175, y: 252 },
|
|
{ x: 483, y: 564 },
|
|
'ControlOrMeta'
|
|
)
|
|
expect(await comfyPage.getGraphSelectedItemsCount()).toEqual(3)
|
|
|
|
await comfyPage.rightClickCanvas()
|
|
await comfyPage.clickContextMenuItem('Save Selected as Template')
|
|
await comfyPage.nextFrame()
|
|
|
|
await comfyPage.rightClickCanvas()
|
|
await comfyPage.clickContextMenuItem('Node Templates >')
|
|
await comfyPage.clickContextMenuItem(templateName)
|
|
|
|
await expect(comfyPage.canvas).toHaveScreenshot()
|
|
})
|
|
|
|
test('Can load old format template', async ({ comfyPage }) => {
|
|
await comfyPage.setNodeTemplates('vintage_clipboard_template.json')
|
|
await comfyPage.reload()
|
|
|
|
// TODO: Flaky test. Right click requires delay after reload, but other interactions do not.
|
|
await comfyPage.page.waitForTimeout(500)
|
|
|
|
await comfyPage.rightClickCanvas()
|
|
await comfyPage.clickContextMenuItem('Node Templates >')
|
|
await comfyPage.clickContextMenuItem('vintageClipboard Template')
|
|
|
|
await expect(comfyPage.canvas).toHaveScreenshot()
|
|
})
|
|
|
|
test('Can load new format template', async ({ comfyPage }) => {
|
|
await comfyPage.setNodeTemplates('node_template_templates.json')
|
|
await comfyPage.reload()
|
|
|
|
// TODO: Flaky test. Right click requires delay after reload, but other interactions do not.
|
|
await comfyPage.page.waitForTimeout(500)
|
|
|
|
await comfyPage.rightClickCanvas()
|
|
await comfyPage.clickContextMenuItem('Node Templates >')
|
|
await comfyPage.clickContextMenuItem('Three Nodes Template')
|
|
|
|
await expect(comfyPage.canvas).toHaveScreenshot()
|
|
})
|
|
})
|