fix(e2e): wait for V2 persistence debounce before reload in workflow modify test

Amp-Thread-ID: https://ampcode.com/threads/T-019c17aa-1890-762e-ab68-28cc797456b4
This commit is contained in:
bymyself
2026-01-31 21:47:01 -08:00
parent c8b3a0fbd9
commit 41e43bf237

View File

@@ -702,6 +702,24 @@ 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.Draft:')) {
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 }
)
await comfyPage.setup({ clearStorage: false })
await expect(comfyPage.canvas).toHaveScreenshot(
'single_ksampler_modified.png'