mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-02-06 08:00:05 +00:00
63 lines
2.2 KiB
TypeScript
63 lines
2.2 KiB
TypeScript
import { expect } from '@playwright/test'
|
|
|
|
import { comfyPageFixture as test } from '../fixtures/ComfyPage'
|
|
import { PerformanceMonitor } from '../helpers/performanceMonitor'
|
|
|
|
test.describe('Graph Canvas Menu', () => {
|
|
test.beforeEach(async ({ comfyPage }) => {
|
|
// Set link render mode to spline to make sure it's not affected by other tests'
|
|
// side effects.
|
|
await comfyPage.setSetting('Comfy.LinkRenderMode', 2)
|
|
})
|
|
|
|
test('@perf Can toggle link visibility', async ({ comfyPage }) => {
|
|
const perfMonitor = new PerformanceMonitor(comfyPage.page)
|
|
const testName = 'toggle-link-visibility'
|
|
|
|
await perfMonitor.startMonitoring(testName)
|
|
|
|
// Note: `Comfy.Graph.CanvasMenu` is disabled in comfyPage setup.
|
|
// so no cleanup is needed.
|
|
await perfMonitor.measureOperation('enable-canvas-menu', async () => {
|
|
await comfyPage.setSetting('Comfy.Graph.CanvasMenu', true)
|
|
})
|
|
|
|
const button = comfyPage.page.getByTestId('toggle-link-visibility-button')
|
|
|
|
await perfMonitor.markEvent('before-hide-links')
|
|
await perfMonitor.measureOperation('hide-links', async () => {
|
|
await button.click()
|
|
await comfyPage.nextFrame()
|
|
})
|
|
await perfMonitor.markEvent('after-hide-links')
|
|
|
|
// Screenshot assertions and validations stay outside performance monitoring
|
|
await expect(comfyPage.canvas).toHaveScreenshot(
|
|
'canvas-with-hidden-links.png'
|
|
)
|
|
const hiddenLinkRenderMode = await comfyPage.page.evaluate(() => {
|
|
return window['LiteGraph'].HIDDEN_LINK
|
|
})
|
|
expect(await comfyPage.getSetting('Comfy.LinkRenderMode')).toBe(
|
|
hiddenLinkRenderMode
|
|
)
|
|
|
|
await perfMonitor.markEvent('before-show-links')
|
|
await perfMonitor.measureOperation('show-links', async () => {
|
|
await button.click()
|
|
await comfyPage.nextFrame()
|
|
})
|
|
await perfMonitor.markEvent('after-show-links')
|
|
|
|
// Screenshot assertions and validations stay outside performance monitoring
|
|
await expect(comfyPage.canvas).toHaveScreenshot(
|
|
'canvas-with-visible-links.png'
|
|
)
|
|
expect(await comfyPage.getSetting('Comfy.LinkRenderMode')).not.toBe(
|
|
hiddenLinkRenderMode
|
|
)
|
|
|
|
await perfMonitor.finishMonitoring(testName)
|
|
})
|
|
})
|