From 62be958d47841347ad5eeca4cfb4a51713a5cd61 Mon Sep 17 00:00:00 2001
From: bymyself <abolkonsky.rem@gmail.com>
Date: Tue, 8 Oct 2024 06:11:22 -0700
Subject: [PATCH] Add tests on node multi-select (#1163)

---
 browser_tests/interaction.spec.ts | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/browser_tests/interaction.spec.ts b/browser_tests/interaction.spec.ts
index 743884e02d..dd19d3e1db 100644
--- a/browser_tests/interaction.spec.ts
+++ b/browser_tests/interaction.spec.ts
@@ -12,9 +12,19 @@ test.describe('Node Interaction', () => {
   })
 
   test.describe('Node Selection', () => {
-    test.afterEach(async ({ comfyPage }) => {
-      // Deselect all nodes
-      await comfyPage.clickEmptySpace()
+    const multiSelectModifiers = ['Control', 'Shift', 'Meta']
+
+    multiSelectModifiers.forEach((modifier) => {
+      test(`Can add multiple nodes to selection using ${modifier}+Click`, async ({
+        comfyPage
+      }) => {
+        const clipNodes = await comfyPage.getNodeRefsByType('CLIPTextEncode')
+        for (const node of clipNodes) {
+          await node.click('title', { modifiers: [modifier] })
+        }
+        const selectedNodeCount = await comfyPage.getSelectedGraphNodesCount()
+        expect(selectedNodeCount).toBe(clipNodes.length)
+      })
     })
 
     test('Can highlight selected', async ({ comfyPage }) => {