Fix test failures

Comfy-Org · Oct 11, 2024 · 10e774b · 10e774b
1 parent 8d751cf
commit 10e774b
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 5 deletions.
diff --git a/browser_tests/changeTracker.spec.ts b/browser_tests/changeTracker.spec.ts
@@ -87,12 +87,12 @@ test.describe('Change Tracker', () => {
 
     await multipleChanges()
 
-    await comfyPage.ctrlZ()
+    await comfyPage.executeCommand('Comfy.Undo')
     await expect(node).not.toBeBypassed()
     await expect(node).not.toBePinned()
     await expect(node).not.toBeCollapsed()
 
-    await comfyPage.ctrlY()
+    await comfyPage.executeCommand('Comfy.Redo')
     await expect(node).toBeBypassed()
     await expect(node).toBePinned()
     await expect(node).toBeCollapsed()

diff --git a/browser_tests/menu.spec.ts b/browser_tests/menu.spec.ts
@@ -499,7 +499,7 @@ test.describe('Menu', () => {
       const workflowMenuItem =
         await comfyPage.menu.topbar.getMenuItem('Workflow')
       await workflowMenuItem.click()
-      const exportTag = comfyPage.page.locator('.keybinding-tag', {
+      const exportTag = comfyPage.page.locator('.keybinding-tag:visible', {
         hasText: 'Ctrl + s'
       })
       expect(await exportTag.count()).toBe(1)

diff --git a/src/stores/coreKeybindings.ts b/src/stores/coreKeybindings.ts
@@ -166,14 +166,24 @@ export const CORE_KEYBINDINGS: Keybinding[] = [
       key: 'z',
       ctrl: true
     },
-    commandId: 'Comfy.Undo'
+    commandId: 'Comfy.Undo',
+    targetSelector: '#graph-canvas'
   },
   {
     combo: {
       key: 'Z',
       ctrl: true,
       shift: true
     },
-    commandId: 'Comfy.Redo'
+    commandId: 'Comfy.Redo',
+    targetSelector: '#graph-canvas'
+  },
+  {
+    combo: {
+      key: 'y',
+      ctrl: true
+    },
+    commandId: 'Comfy.Redo',
+    targetSelector: '#graph-canvas'
   }
 ]