Merge pull request #156 from layer5labs/CodeAhmedJamil-patch-1

Fix snapshot crashing at removing elements

Author: CodeAhmedJamil

cypress-action/cypress/e2e/e2e/loadDesign.js

@@ -70,17 +70,22 @@ const removeWidgets = () => {
     "kanvas-bottom-dock",
     "left-navigation-bar",
     "top-navigation-bar",
+    "hide-from-design-snapshot",
   ];
 
-  ids.forEach((className) => {
-    try {
-      cy.get(`#${className}`).invoke("remove");
-    } catch (error) {
-      console.log(`Error removing ${className}`);
-    }
+  ids.forEach((id) => {
+    cy.get(`#${id}`, { timeout: 0 }).then(($el) => {
+      if ($el.length) {
+        $el.remove();
+      }
+    }).catch(() => {
+      // Element not found, do nothing
+      console.log(`Element #${id} not found`);
+    });
   });
 };
 
+
 ["light", "dark"].forEach((theme) => {
   InfraShot(theme);
 });