mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-02-04 07:00:23 +00:00
92 lines
3.1 KiB
TypeScript
92 lines
3.1 KiB
TypeScript
import { comfyPageFixture as test } from '../../../fixtures/comfyPageFixture'
|
|
import { comfyExpect as expect } from '../../../fixtures/ComfyPage'
|
|
|
|
test.describe('Vue Node Collapse', () => {
|
|
test.beforeEach(async ({ comfyPage }) => {
|
|
await comfyPage.setSetting('Comfy.Graph.CanvasMenu', false)
|
|
await comfyPage.setSetting('Comfy.EnableTooltips', true)
|
|
await comfyPage.setSetting('Comfy.VueNodes.Enabled', true)
|
|
await comfyPage.setup()
|
|
await comfyPage.vueNodes.waitForNodes()
|
|
})
|
|
|
|
test('should allow collapsing node with collapse icon', async ({
|
|
comfyPage
|
|
}) => {
|
|
const vueNode = await comfyPage.vueNodes.getFixtureByTitle('KSampler')
|
|
await expect(vueNode.root).toBeVisible()
|
|
|
|
// Initially should not be collapsed
|
|
const body = vueNode.body
|
|
await expect(body).toBeVisible()
|
|
const expandedBoundingBox = await vueNode.boundingBox()
|
|
if (!expandedBoundingBox)
|
|
throw new Error('Failed to get node bounding box before collapse')
|
|
|
|
// Collapse the node
|
|
await vueNode.toggleCollapse()
|
|
await comfyPage.nextFrame()
|
|
|
|
// Verify node content is hidden
|
|
await expect(body).not.toBeVisible()
|
|
const collapsedBoundingBox = await vueNode.boundingBox()
|
|
if (!collapsedBoundingBox)
|
|
throw new Error('Failed to get node bounding box after collapse')
|
|
expect(collapsedBoundingBox.height).toBeLessThan(expandedBoundingBox.height)
|
|
|
|
// Expand again
|
|
await vueNode.toggleCollapse()
|
|
await comfyPage.nextFrame()
|
|
await expect(body).toBeVisible()
|
|
|
|
// Size should be restored
|
|
const expandedBoundingBoxAfter = await vueNode.boundingBox()
|
|
if (!expandedBoundingBoxAfter)
|
|
throw new Error('Failed to get node bounding box after expand')
|
|
expect(expandedBoundingBoxAfter.height).toBeGreaterThanOrEqual(
|
|
collapsedBoundingBox.height
|
|
)
|
|
})
|
|
|
|
test('should show collapse/expand icon state', async ({ comfyPage }) => {
|
|
const vueNode = await comfyPage.vueNodes.getFixtureByTitle('KSampler')
|
|
await expect(vueNode.root).toBeVisible()
|
|
|
|
// Check initial expanded state icon
|
|
let iconClass = await vueNode.getCollapseIconClass()
|
|
expect(iconClass).not.toContain('-rotate-90')
|
|
|
|
// Collapse and check icon
|
|
await vueNode.toggleCollapse()
|
|
iconClass = await vueNode.getCollapseIconClass()
|
|
expect(iconClass).toContain('-rotate-90')
|
|
|
|
// Expand and check icon
|
|
await vueNode.toggleCollapse()
|
|
iconClass = await vueNode.getCollapseIconClass()
|
|
expect(iconClass).not.toContain('-rotate-90')
|
|
})
|
|
|
|
test('should preserve title when collapsing/expanding', async ({
|
|
comfyPage
|
|
}) => {
|
|
const vueNode = await comfyPage.vueNodes.getFixtureByTitle('KSampler')
|
|
await expect(vueNode.root).toBeVisible()
|
|
|
|
// Set custom title
|
|
await vueNode.setTitle('Test Sampler')
|
|
expect(await vueNode.getTitle()).toBe('Test Sampler')
|
|
|
|
// Collapse
|
|
await vueNode.toggleCollapse()
|
|
expect(await vueNode.getTitle()).toBe('Test Sampler')
|
|
|
|
// Expand
|
|
await vueNode.toggleCollapse()
|
|
expect(await vueNode.getTitle()).toBe('Test Sampler')
|
|
|
|
// Verify title is still displayed
|
|
await expect(vueNode.header).toContainText('Test Sampler')
|
|
})
|
|
})
|