mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-07-14 19:27:09 +00:00
3164e6ab61
Also adds proper typing for `onNodeOutputsUpdated` See also: #12877 and #13427, which include near equivalent changes for the bug itself, but different tests. If I had more time and had not already made my own fix, I would have liked to spend more time getting either of them cleaned up.
79 lines
2.8 KiB
TypeScript
79 lines
2.8 KiB
TypeScript
import { mergeTests } from '@playwright/test'
|
|
|
|
import {
|
|
comfyPageFixture as test,
|
|
comfyExpect as expect
|
|
} from '@e2e/fixtures/ComfyPage'
|
|
import { ExecutionHelper } from '@e2e/fixtures/helpers/ExecutionHelper'
|
|
import { webSocketFixture } from '@e2e/fixtures/ws'
|
|
|
|
const wstest = mergeTests(test, webSocketFixture)
|
|
|
|
test.describe('Preview as Text node', () => {
|
|
test('does not include preview widget values in the API prompt', async ({
|
|
comfyPage
|
|
}) => {
|
|
await comfyPage.page.evaluate(() => {
|
|
const node = window.LiteGraph!.createNode('PreviewAny')!
|
|
node.pos = [500, 200]
|
|
window.app!.graph.add(node)
|
|
})
|
|
|
|
// Simulate a previous execution: backend returned text and the frontend
|
|
// populated the preview widget values. The next prompt submission must
|
|
// NOT echo those values back as inputs (which would change the cache
|
|
// signature and trigger a redundant re-execution).
|
|
await comfyPage.page.evaluate(() => {
|
|
const node = window.app!.graph.nodes.find((n) => n.type === 'PreviewAny')!
|
|
for (const widget of node.widgets ?? []) {
|
|
if (widget.name?.startsWith('preview_')) {
|
|
widget.value = 'rendered preview content from previous execution'
|
|
}
|
|
}
|
|
})
|
|
|
|
const apiWorkflow = await comfyPage.workflow.getExportedWorkflow({
|
|
api: true
|
|
})
|
|
|
|
const previewEntry = Object.values(apiWorkflow).find(
|
|
(n) => n.class_type === 'PreviewAny'
|
|
)
|
|
expect(previewEntry).toBeDefined()
|
|
|
|
expect(previewEntry!.inputs).not.toHaveProperty('preview_markdown')
|
|
expect(previewEntry!.inputs).not.toHaveProperty('preview_text')
|
|
expect(previewEntry!.inputs).not.toHaveProperty('previewMode')
|
|
})
|
|
|
|
wstest(
|
|
'restoring workflow restores state',
|
|
{ tag: '@vue-nodes' },
|
|
async ({ comfyPage, getWebSocket }) => {
|
|
const execution = new ExecutionHelper(comfyPage, await getWebSocket())
|
|
|
|
await comfyPage.menu.topbar.newWorkflowButton.click()
|
|
await comfyPage.searchBoxV2.addNode('Preview as Text')
|
|
const node = await comfyPage.vueNodes.getFixtureByTitle('Preview as Text')
|
|
const preview = node.root.locator('textarea')
|
|
|
|
await test.step('node previews execution result', async () => {
|
|
const id = await comfyPage.vueNodes.getNodeIdByTitle('Preview as Text')
|
|
execution.executed('', id, { text: 'massive fennec ears' })
|
|
await expect(preview).toHaveValue('massive fennec ears')
|
|
})
|
|
|
|
await test.step('swap to a different workflow and back', async () => {
|
|
await comfyPage.menu.topbar.getTab(0).click()
|
|
await expect(node.root).toBeHidden()
|
|
await comfyPage.menu.topbar.getTab(1).click()
|
|
await expect(node.root).toBeVisible()
|
|
})
|
|
|
|
await expect(preview, 'previous output is restored').toHaveValue(
|
|
'massive fennec ears'
|
|
)
|
|
}
|
|
)
|
|
})
|