Comfy-Org/ComfyUI_frontend
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI_frontend.git synced 2026-05-04 13:12:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Test] [Performance] Apply perf monitor wrappers to test files (1/4) (#4124)

Browse Source
This commit is contained in:
Christian Byrne
2025-06-10 15:24:13 -07:00
committed by GitHub
parent 7e3c7b754f
commit 862a9d2396
4 changed files with 530 additions and 157 deletions
Download Patch File Download Diff File Expand all files Collapse all files

106
browser_tests/tests/primitiveNode.spec.ts
View File

@@ -2,46 +2,114 @@ import { expect } from '@playwright/test'
import { comfyPageFixture as test } from '../fixtures/ComfyPage'
import type { NodeReference } from '../fixtures/utils/litegraphUtils'
import { PerformanceMonitor } from '../helpers/performanceMonitor'
test.describe('Primitive Node', () => {
test('Can load with correct size', async ({ comfyPage }) => {
await comfyPage.loadWorkflow('primitive/primitive_node')
test('@perf Can load with correct size', async ({ comfyPage }) => {
const perfMonitor = new PerformanceMonitor(comfyPage.page)
const testName = 'primitive-node-load'
await perfMonitor.startMonitoring(testName)
await perfMonitor.measureOperation('load-workflow', async () => {
await comfyPage.loadWorkflow('primitive/primitive_node')
})
await expect(comfyPage.canvas).toHaveScreenshot('primitive_node.png')
await perfMonitor.finishMonitoring(testName)
})
// When link is dropped on widget, it should automatically convert the widget
// to input.
test('Can connect to widget', async ({ comfyPage }) => {
await comfyPage.loadWorkflow('primitive/primitive_node_unconnected')
const primitiveNode: NodeReference = await comfyPage.getNodeRefById(1)
const ksamplerNode: NodeReference = await comfyPage.getNodeRefById(2)
test('@perf Can connect to widget', async ({ comfyPage }) => {
const perfMonitor = new PerformanceMonitor(comfyPage.page)
const testName = 'primitive-node-connect-widget'
await perfMonitor.startMonitoring(testName)
await perfMonitor.measureOperation('load-workflow', async () => {
await comfyPage.loadWorkflow('primitive/primitive_node_unconnected')
})
let primitiveNode: NodeReference
let ksamplerNode: NodeReference
await perfMonitor.measureOperation('get-node-references', async () => {
primitiveNode = await comfyPage.getNodeRefById(1)
ksamplerNode = await comfyPage.getNodeRefById(2)
})
// Connect the output of the primitive node to the input of first widget of the ksampler node
await primitiveNode.connectWidget(0, ksamplerNode, 0)
await perfMonitor.measureOperation('connect-widget', async () => {
await primitiveNode!.connectWidget(0, ksamplerNode!, 0)
})
await expect(comfyPage.canvas).toHaveScreenshot(
'primitive_node_connected.png'
)
await perfMonitor.finishMonitoring(testName)
})
test('Can connect to dom widget', async ({ comfyPage }) => {
await comfyPage.loadWorkflow(
'primitive/primitive_node_unconnected_dom_widget'
)
const primitiveNode: NodeReference = await comfyPage.getNodeRefById(1)
const clipEncoderNode: NodeReference = await comfyPage.getNodeRefById(2)
await primitiveNode.connectWidget(0, clipEncoderNode, 0)
test('@perf Can connect to dom widget', async ({ comfyPage }) => {
const perfMonitor = new PerformanceMonitor(comfyPage.page)
const testName = 'primitive-node-connect-dom-widget'
await perfMonitor.startMonitoring(testName)
await perfMonitor.measureOperation('load-workflow', async () => {
await comfyPage.loadWorkflow(
'primitive/primitive_node_unconnected_dom_widget'
)
})
let primitiveNode: NodeReference
let clipEncoderNode: NodeReference
await perfMonitor.measureOperation('get-node-references', async () => {
primitiveNode = await comfyPage.getNodeRefById(1)
clipEncoderNode = await comfyPage.getNodeRefById(2)
})
await perfMonitor.measureOperation('connect-dom-widget', async () => {
await primitiveNode!.connectWidget(0, clipEncoderNode!, 0)
})
await expect(comfyPage.canvas).toHaveScreenshot(
'primitive_node_connected_dom_widget.png'
)
await perfMonitor.finishMonitoring(testName)
})
test('Can connect to static primitive node', async ({ comfyPage }) => {
await comfyPage.loadWorkflow('primitive/static_primitive_unconnected')
const primitiveNode: NodeReference = await comfyPage.getNodeRefById(1)
const ksamplerNode: NodeReference = await comfyPage.getNodeRefById(2)
await primitiveNode.connectWidget(0, ksamplerNode, 0)
test('@perf Can connect to static primitive node', async ({ comfyPage }) => {
const perfMonitor = new PerformanceMonitor(comfyPage.page)
const testName = 'primitive-node-connect-static'
await perfMonitor.startMonitoring(testName)
await perfMonitor.measureOperation('load-workflow', async () => {
await comfyPage.loadWorkflow('primitive/static_primitive_unconnected')
})
let primitiveNode: NodeReference
let ksamplerNode: NodeReference
await perfMonitor.measureOperation('get-node-references', async () => {
primitiveNode = await comfyPage.getNodeRefById(1)
ksamplerNode = await comfyPage.getNodeRefById(2)
})
await perfMonitor.measureOperation('connect-static-primitive', async () => {
await primitiveNode!.connectWidget(0, ksamplerNode!, 0)
})
await expect(comfyPage.canvas).toHaveScreenshot(
'static_primitive_connected.png'
)
await perfMonitor.finishMonitoring(testName)
})
test('Report missing nodes when connect to missing node', async ({