Fix missed awaits

browser_tests/changeTracker.spec.ts

@@ -53,6 +53,38 @@ test.describe('Change Tracker', () => {
     })
   })
 
+  test('Can undo & redo user changes', async ({ comfyPage }) => {
+    await comfyPage.clickEmptySpace()
+    await expect(comfyPage.canvas).toHaveScreenshot('undo-initial-state.png')
+
+    await comfyPage.dragAndDrop(
+      comfyPage.clipTextEncodeNode2OutputSlot,
+      comfyPage.kSamplerTitlebar
+    )
+    await expect(comfyPage.canvas).toHaveScreenshot('undo-step1.png')
+
+    await comfyPage.ctrlZ()
+    await expect(comfyPage.canvas).toHaveScreenshot('undo-initial-state.png')
+
+    await comfyPage.dragAndDrop(
+      comfyPage.clipTextEncodeNode1OutputSlot,
+      comfyPage.kSamplerTitlebar
+    )
+    await expect(comfyPage.canvas).toHaveScreenshot('undo-step3.png')
+
+    // Test what a real person does
+    await comfyPage.ctrlZ()
+    await comfyPage.ctrlZ()
+    await comfyPage.ctrlZ()
+    await comfyPage.ctrlZ()
+    await expect(comfyPage.canvas).toHaveScreenshot('undo-initial-state.png')
+
+    // Test past the bounds of the buffer
+    await comfyPage.ctrlY()
+    await comfyPage.ctrlY()
+    await expect(comfyPage.canvas).toHaveScreenshot('undo-step3.png')
+  })
+
   test('Can group multiple change actions into a single transaction', async ({
     comfyPage
   }) => {

browser_tests/fixtures/ComfyPage.ts

@@ -458,10 +458,20 @@ export class ComfyPage {
   get clipTextEncodeNode1InputSlot(): Position {
     return { x: 427, y: 198 }
   }
+  get clipTextEncodeNode1OutputSlot(): Position {
+    return { x: 828, y: 197 }
+  }
 
   get clipTextEncodeNode2InputSlot(): Position {
     return { x: 422, y: 402 }
   }
+  get clipTextEncodeNode2OutputSlot(): Position {
+    return { x: 828, y: 400 }
+  }
+
+  get kSamplerTitlebar(): Position {
+    return { x: 1058, y: 175 }
+  }
 
   // A point on input edge.
   get clipTextEncodeNode2InputLinkPath(): Position {