fix(e2e): use evaluate timeout instead of waitForFunction for persistence wait

Amp-Thread-ID: https://ampcode.com/threads/T-019c17aa-1890-762e-ab68-28cc797456b4
This commit is contained in:
bymyself
2026-01-31 22:03:45 -08:00
parent 734cdc47c8
commit fd0b5841dc

View File

@@ -702,23 +702,9 @@ test.describe('Load workflow', { tag: '@screenshot' }, () => {
await expect(comfyPage.canvas).toHaveScreenshot(
'single_ksampler_modified.png'
)
// Wait for V2 persistence to save the modified workflow (debounced at 512ms)
// Check that localStorage has a draft key with the collapsed node state
await comfyPage.page.waitForFunction(
() => {
for (let i = 0; i < window.localStorage.length; i++) {
const key = window.localStorage.key(i)
if (key?.startsWith('Comfy.Workflow.Draft.v2:')) {
const value = window.localStorage.getItem(key)
// Check that the draft contains collapsed node flags
if (value && value.includes('"flags":')) {
return true
}
}
}
return false
},
{ timeout: 2000 }
// Wait for V2 persistence debounce (512ms) to save the modified workflow
await comfyPage.page.evaluate(
() => new Promise((resolve) => setTimeout(resolve, 600))
)
await comfyPage.setup({ clearStorage: false })
await expect(comfyPage.canvas).toHaveScreenshot(