Implemented shadowRoot functionality wrapper

Author: mialeska

src/main/java/aquality/selenium/browser/JavaScript.java

@@ -40,7 +40,8 @@ public enum JavaScript {
     GET_VIEWPORT_COORDINATES("getViewPortCoordinates.js"),
     GET_SCREEN_OFFSET("getScreenOffset.js"),
     OPEN_IN_NEW_TAB("openInNewTab.js"),
-    OPEN_NEW_TAB("openNewTab.js");
+    OPEN_NEW_TAB("openNewTab.js"),
+    EXPAND_SHADOW_ROOT("expandShadowRoot.js");
 
     private final String filename;
 
@@ -55,11 +56,12 @@ public enum JavaScript {
      */
     public String getScript() {
         URL scriptFile = getClass().getResource("/js/" + filename);
-        try {
-            InputStream stream = scriptFile.openStream();
-            return IOUtils.toString(stream, StandardCharsets.UTF_8.name());
-        } catch (IOException e) {
-            Logger.getInstance().fatal(format("Couldn't find the script \"%s\"", filename), e);
+        if (scriptFile != null) {
+            try (InputStream stream = scriptFile.openStream()) {
+                return IOUtils.toString(stream, StandardCharsets.UTF_8.name());
+            } catch (IOException e) {
+                Logger.getInstance().fatal(format("Couldn't find the script \"%s\"", filename), e);
+            }
         }
         return "";
     }

src/main/java/aquality/selenium/elements/Element.java

@@ -20,6 +20,7 @@
 import org.openqa.selenium.By;
 import org.openqa.selenium.Keys;
 import org.openqa.selenium.NoSuchElementException;
+import org.openqa.selenium.SearchContext;
 import org.openqa.selenium.remote.RemoteWebElement;
 
 import java.time.Duration;
@@ -28,6 +29,8 @@
  * Abstract class, describing wrapper of WebElement.
  */
 public abstract class Element extends aquality.selenium.core.elements.Element implements IElement {
+    private IElementFinder elementFinder;
+
     /**
      * The main constructor
      *
@@ -51,7 +54,14 @@ protected IElementFactory getElementFactory() {
 
     @Override
     protected IElementFinder getElementFinder() {
-        return AqualityServices.get(IElementFinder.class);
+        if (elementFinder == null) {
+            elementFinder = AqualityServices.get(IElementFinder.class);
+        }
+        return elementFinder;
+    }
+
+    void setElementFinder(IElementFinder elementFinder) {
+        this.elementFinder = elementFinder;
     }
 
     @Override
@@ -176,4 +186,10 @@ public void sendKeys(Keys key) {
         logElementAction("loc.text.sending.key", Keys.class.getSimpleName().concat(".").concat(key.name()));
         doWithRetry(() -> getElement().sendKeys(key));
     }
+
+    @Override
+    public SearchContext expandShadowRoot() {
+        logElementAction("loc.shadowroot.expand");
+        return getElement().getShadowRoot();
+    }
 }

src/main/java/aquality/selenium/elements/ElementFactory.java

@@ -3,6 +3,7 @@
 import aquality.selenium.browser.AqualityServices;
 import aquality.selenium.browser.JavaScript;
 import aquality.selenium.core.elements.interfaces.IElementFinder;
+import aquality.selenium.core.elements.interfaces.IElementSupplier;
 import aquality.selenium.core.localization.ILocalizationManager;
 import aquality.selenium.core.waitings.IConditionalWait;
 import aquality.selenium.elements.interfaces.*;
@@ -19,9 +20,12 @@
 
 public class ElementFactory extends aquality.selenium.core.elements.ElementFactory implements IElementFactory {
 
+    private final IElementFinder elementFinder;
+
     @Inject
     public ElementFactory(IConditionalWait conditionalWait, IElementFinder elementFinder, ILocalizationManager localizationManager) {
         super(conditionalWait, elementFinder, localizationManager);
+        this.elementFinder = elementFinder;
     }
 
     private static Map<Class<? extends By>, String> getLocatorToXPathTemplateMap() {
@@ -92,4 +96,16 @@ protected String extractXPathFromLocator(By locator) {
                 ? String.format(getLocatorToXPathTemplateMap().get(locatorClass), locValuableString)
                 : super.extractXPathFromLocator(locator);
     }
+
+    @Override
+    protected <T extends aquality.selenium.core.elements.interfaces.IElement> IElementSupplier<T> getDefaultElementSupplier(Class<T> clazz) {
+        IElementSupplier<T> baseSupplier = super.getDefaultElementSupplier(clazz);
+        return (locator, name, state) -> {
+            T element = baseSupplier.get(locator, name, state);
+            if (element instanceof Element) {
+                ((Element)element).setElementFinder(elementFinder);
+            }
+            return element;
+        };
+    }
 }

src/main/java/aquality/selenium/elements/actions/JsActions.java

@@ -6,12 +6,14 @@
 import aquality.selenium.core.utilities.IElementActionRetrier;
 import aquality.selenium.elements.HighlightState;
 import aquality.selenium.elements.interfaces.IElement;
+import aquality.selenium.elements.interfaces.IShadowRootExpander;
 import org.openqa.selenium.Point;
+import org.openqa.selenium.SearchContext;
 
 import java.util.ArrayList;
 import java.util.List;
 
-public class JsActions {
+public class JsActions implements IShadowRootExpander {
 
     protected IElement element;
     protected String type;
@@ -23,6 +25,12 @@ public JsActions(IElement element, String type) {
         this.name = element.getName();
     }
 
+    @Override
+    public SearchContext expandShadowRoot() {
+        logElementAction("loc.shadowroot.expand.js");
+        return (SearchContext) executeScript(JavaScript.EXPAND_SHADOW_ROOT, element);
+    }
+
     /**
      * Click via JS.
      */

src/main/java/aquality/selenium/elements/interfaces/IElement.java

@@ -11,7 +11,7 @@
 
 import java.util.List;
 
-public interface IElement extends aquality.selenium.core.elements.interfaces.IElement {
+public interface IElement extends aquality.selenium.core.elements.interfaces.IElement, IShadowRootExpander {
 
     /**
      * Send keys.