diff --git a/browser_tests/fixtures/ComfyPage.ts b/browser_tests/fixtures/ComfyPage.ts index c32dd3937..0a855b2ff 100644 --- a/browser_tests/fixtures/ComfyPage.ts +++ b/browser_tests/fixtures/ComfyPage.ts @@ -1540,7 +1540,7 @@ export class ComfyPage { } async convertOffsetToCanvas(pos: [number, number]) { - return this.page.evaluate((pos) => { + return await this.page.evaluate((pos) => { return window['app'].canvas.ds.convertOffsetToCanvas(pos) }, pos) } @@ -1592,21 +1592,21 @@ export class ComfyPage { await this.page.mouse.move(10, 10) } async getUndoQueueSize() { - return this.page.evaluate(() => { + return await this.page.evaluate(() => { const workflow = (window['app'].extensionManager as WorkspaceStore) .workflow.activeWorkflow return workflow?.changeTracker.undoQueue.length }) } async getRedoQueueSize() { - return this.page.evaluate(() => { + return await this.page.evaluate(() => { const workflow = (window['app'].extensionManager as WorkspaceStore) .workflow.activeWorkflow return workflow?.changeTracker.redoQueue.length }) } async isCurrentWorkflowModified() { - return this.page.evaluate(() => { + return await this.page.evaluate(() => { return (window['app'].extensionManager as WorkspaceStore).workflow .activeWorkflow?.isModified }) diff --git a/browser_tests/fixtures/components/SidebarTab.ts b/browser_tests/fixtures/components/SidebarTab.ts index 7baaa1ef9..f4a840b3b 100644 --- a/browser_tests/fixtures/components/SidebarTab.ts +++ b/browser_tests/fixtures/components/SidebarTab.ts @@ -237,7 +237,7 @@ export class QueueSidebarTab extends SidebarTab { if (width < 0 || width > 100) { throw new Error('Width must be between 0 and 100') } - return this.page.evaluate((width) => { + return await this.page.evaluate((width) => { localStorage.setItem('queue', JSON.stringify([width, 100 - width])) }, width) } diff --git a/browser_tests/tests/remoteWidgets.spec.ts b/browser_tests/tests/remoteWidgets.spec.ts index 4a390af96..b99104c5f 100644 --- a/browser_tests/tests/remoteWidgets.spec.ts +++ b/browser_tests/tests/remoteWidgets.spec.ts @@ -37,8 +37,8 @@ test.describe('Remote COMBO Widget', () => { }, nodeName) } - const clickRefreshButton = (comfyPage: ComfyPage, nodeName: string) => { - return comfyPage.page.evaluate((name) => { + const clickRefreshButton = async (comfyPage: ComfyPage, nodeName: string) => { + return await comfyPage.page.evaluate((name) => { const node = window['app'].graph.nodes.find((node) => node.title === name) const buttonWidget = node.widgets.find((w) => w.name === 'refresh') return buttonWidget?.callback()