Fix closing saved workflows (#1049)

* Fix workflow save. Resolves #996 Resolves #1048

* Add test on closing tabs

* Add test on closing open workflows in sidebar
This commit is contained in:
bymyself
2024-09-30 12:09:24 -07:00
committed by GitHub
parent 0117964ca5
commit 04e1344676
3 changed files with 66 additions and 0 deletions

View File

@@ -437,6 +437,19 @@ test.describe('Menu', () => {
comfyPage.page.locator('.comfy-missing-nodes')
).not.toBeVisible()
})
test('Can close saved-workflows from the open workflows section', async ({
comfyPage
}) => {
await comfyPage.menu.topbar.saveWorkflow('deault')
const closeButton = comfyPage.page.locator(
'.comfyui-workflows-open .p-button-icon.pi-times'
)
await closeButton.click()
expect(
await comfyPage.menu.workflowsTab.getOpenedWorkflowNames()
).toEqual(['*Unsaved Workflow (2).json'])
})
})
test.describe('Workflows topbar tabs', () => {
@@ -447,11 +460,30 @@ test.describe('Menu', () => {
)
})
test.afterEach(async ({ comfyPage }) => {
// Delete the saved workflow for cleanup.
await comfyPage.page.evaluate(async () => {
window['app'].workflowManager.activeWorkflow.delete()
})
})
test('Can show opened workflows', async ({ comfyPage }) => {
expect(await comfyPage.menu.topbar.getTabNames()).toEqual([
'Unsaved Workflow'
])
})
test('Can close saved-workflow tabs', async ({ comfyPage }) => {
const savedWorkflowName = 'default'
await comfyPage.menu.topbar.saveWorkflow(savedWorkflowName)
expect(await comfyPage.menu.topbar.getTabNames()).toEqual([
savedWorkflowName
])
await comfyPage.menu.topbar.closeWorkflowTab('default')
expect(await comfyPage.menu.topbar.getTabNames()).toEqual([
'Unsaved Workflow (2)'
])
})
})
// Only test 'Top' to reduce test time.