handle stale element reference exception

Author: NikolayAvramov

bellatrix.web/src/main/java/solutions/bellatrix/web/components/WebComponent.java

@@ -118,9 +118,13 @@ public <ComponentT extends WebComponent> ComponentT as(Class<ComponentT> compone
 
     public WebElement getWrappedElement() {
         try {
-            wrappedElement.isDisplayed(); // checking if getting property throws exception
-            return wrappedElement;
-        } catch (StaleElementReferenceException | NoSuchElementException | NullPointerException | ScriptTimeoutException ex ) {
+            if(wrappedElement != null) {
+                wrappedElement.isDisplayed(); // checking if getting property throws exception
+                return wrappedElement;
+            } else {
+                return findElement();
+            }
+        } catch (Exception ex ) {
             return findElement();
         }
     }

bellatrix.web/src/main/java/solutions/bellatrix/web/waitstrategies/ToExistWaitStrategy.java

@@ -13,9 +13,7 @@
 
 package solutions.bellatrix.web.waitstrategies;
 
-import org.openqa.selenium.By;
-import org.openqa.selenium.NoSuchElementException;
-import org.openqa.selenium.SearchContext;
+import org.openqa.selenium.*;
 import solutions.bellatrix.core.configuration.ConfigurationService;
 import solutions.bellatrix.web.configuration.WebSettings;
 
@@ -42,7 +40,7 @@ private boolean elementExists(SearchContext searchContext, By by) {
         try {
             var element = findElement(searchContext, by);
             return element != null;
-        } catch (NoSuchElementException e) {
+        } catch (NoSuchElementException | StaleElementReferenceException | ElementNotInteractableException e) {
             return false;
         }
     }