io.github.sukgu.Shadow Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of automation Show documentation
Show all versions of automation Show documentation
This project automates shadow-dom elements using java selenium
package io.github.sukgu;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.lang.reflect.Method;
import java.util.List;
import org.openqa.selenium.JavascriptExecutor;
import org.openqa.selenium.SearchContext;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebDriverException;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.ie.InternetExplorerDriver;
import org.openqa.selenium.remote.CommandExecutor;
import org.openqa.selenium.remote.RemoteWebDriver;
import org.openqa.selenium.remote.RemoteWebElement;
import org.openqa.selenium.remote.SessionId;
import org.openqa.selenium.support.ui.ExpectedCondition;
import org.openqa.selenium.support.ui.WebDriverWait;
public class Shadow {
WebDriver driver;
WebDriverException exception;
WebDriverWait wait;
CommandExecutor executer;
SessionId sessionId;
ChromeDriver chromeDriver;
FirefoxDriver firfoxDriver;
InternetExplorerDriver ieDriver;
RemoteWebDriver remoteWebDriver;
public Shadow(WebDriver driver) {
if (driver instanceof ChromeDriver) {
sessionId = ((ChromeDriver)driver).getSessionId();
chromeDriver = (ChromeDriver)driver;
} else if (driver instanceof FirefoxDriver) {
sessionId = ((FirefoxDriver)driver).getSessionId();
firfoxDriver = (FirefoxDriver)driver;
} else if (driver instanceof InternetExplorerDriver) {
sessionId = ((InternetExplorerDriver)driver).getSessionId();
ieDriver = (InternetExplorerDriver)driver;
} else if (driver instanceof RemoteWebDriver) {
sessionId = ((RemoteWebDriver)driver).getSessionId();
remoteWebDriver = (RemoteWebDriver)driver;
}
this.driver = driver;
}
private Object injectShadowExecuter(String javascript) {
if(chromeDriver!=null) {
JavascriptExecutor js = (JavascriptExecutor)chromeDriver;
waitForPageLoaded();
return js.executeScript(javascript);
} else if (firfoxDriver!=null) {
waitForPageLoaded();
return firfoxDriver.executeScript(javascript);
} else if (ieDriver!=null) {
waitForPageLoaded();
return ieDriver.executeScript(javascript);
} else if (remoteWebDriver!=null) {
JavascriptExecutor js = (JavascriptExecutor)remoteWebDriver;
waitForPageLoaded();
return js.executeScript(javascript);
} else {
return null;
}
}
private Object injectShadowExecuter(String javascript, WebElement element) {
if(chromeDriver!=null) {
JavascriptExecutor js = (JavascriptExecutor)chromeDriver;
waitForPageLoaded();
return js.executeScript(javascript, element);
} else if (firfoxDriver!=null) {
waitForPageLoaded();
return firfoxDriver.executeScript(javascript, element);
} else if (ieDriver!=null) {
waitForPageLoaded();
return ieDriver.executeScript(javascript, element);
} else if (remoteWebDriver!=null) {
JavascriptExecutor js = (JavascriptExecutor)remoteWebDriver;
waitForPageLoaded();
return js.executeScript(javascript, element);
} else {
return null;
}
}
private Object executerGetObject(String script) {
String javascript = convertJStoText().toString();
javascript += script;
return injectShadowExecuter(javascript);
}
private Object executerGetObject(String script, WebElement element) {
String javascript = convertJStoText().toString();
javascript += script;
return injectShadowExecuter(javascript, element);
}
private StringBuilder convertJStoText() {
InputStream in = getClass().getResourceAsStream("/querySelector.js");
BufferedReader reader = null;
//File jsFile = new File("querySelector.js");
//BufferedReader reader = null;
StringBuilder text = new StringBuilder();
//reader = new BufferedReader(new FileReader(jsFile));
reader = new BufferedReader(new InputStreamReader(in));
if(reader!=null) {
try {
while(reader.ready()) {
text.append(reader.readLine());
}
} catch (IOException e) {
e.printStackTrace();
}
}
if(reader!=null) {
try {
reader.close();
} catch (IOException e) {
e.printStackTrace();
}
}
return text;
}
private void fixLocator(SearchContext context, String cssLocator, WebElement element) {
if (element instanceof RemoteWebElement) {
try {
@SuppressWarnings("rawtypes")
Class[] parameterTypes = new Class[] { SearchContext.class,
String.class, String.class };
Method m = element.getClass().getDeclaredMethod(
"setFoundBy", parameterTypes);
m.setAccessible(true);
Object[] parameters = new Object[] { context, "cssSelector", cssLocator };
m.invoke(element, parameters);
} catch (Exception fail) {
//fail("Something bad happened when fixing locator");
}
}
}
private void waitForPageLoaded() {
ExpectedCondition expectation = new
ExpectedCondition() {
public Boolean apply(WebDriver driver) {
return ((JavascriptExecutor) driver).executeScript("return document.readyState").toString().equals("complete");
}
};
try {
Thread.sleep(1000);
WebDriverWait wait = new WebDriverWait(driver, 30);
wait.until(expectation);
} catch (Throwable error) {
//Assertions.fail("Timeout waiting for Page Load Request to complete.");
}
}
public WebElement findElement(String cssSelector) {
WebElement element = null;
element = (WebElement) executerGetObject("return getObject(\""+cssSelector+"\");");
fixLocator(driver, cssSelector, element);
return element;
}
public WebElement findElement(WebElement parent, String cssSelector) {
WebElement element = null;
element = (WebElement) executerGetObject("return getObject(\""+cssSelector+"\", arguments[0]);", parent);
fixLocator(driver, cssSelector, element);
return element;
}
@SuppressWarnings("unchecked")
public List findElements(String cssSelector) {
List element = null;
Object object = executerGetObject("return getAllObject(\""+cssSelector+"\");");
if(object!=null && object instanceof List) {
element = (List) object;
}
for (WebElement webElement : element) {
fixLocator(driver, cssSelector, webElement);
}
return element;
}
@SuppressWarnings("unchecked")
public List findElements(WebElement parent, String cssSelector) {
List element = null;
Object object = executerGetObject("return getAllObject(\""+cssSelector+"\", arguments[0]);", parent);
if(object!=null && object instanceof List) {
element = (List) object;
}
for (WebElement webElement : element) {
fixLocator(driver, cssSelector, webElement);
}
return element;
}
public WebElement getShadowElement(WebElement parent,String selector) {
WebElement element = null;
element = (WebElement) executerGetObject("return getShadowElement(arguments[0],\""+selector+"\");", parent);
fixLocator(driver, selector, element);
return element;
}
@SuppressWarnings("unchecked")
public List getAllShadowElement(WebElement parent,String selector) {
List elements = null;
Object object = executerGetObject("return getAllShadowElement(arguments[0],\""+selector+"\");", parent);
if(object!=null && object instanceof List) {
elements = (List) object;
}
for (WebElement element : elements) {
fixLocator(driver, selector, element);
}
return elements;
}
public WebElement getParentElement(WebElement element) {
return (WebElement) executerGetObject("return getParentElement(arguments[0]);", element);
}
@SuppressWarnings("unchecked")
public List getChildElements(WebElement parent) {
List elements = null;
Object object = executerGetObject("return getChildElements(arguments[0]);", parent);
if(object!=null && object instanceof List) {
elements = (List) object;
}
return elements;
}
public boolean isVisible(WebElement element) {
return (Boolean) executerGetObject("return isVisible(arguments[0]);", element);
}
public boolean isChecked(WebElement element) {
return (Boolean) executerGetObject("return isChecked(arguments[0]);", element);
}
public boolean isDisabled(WebElement element) {
return (Boolean) executerGetObject("return isDisabled(arguments[0]);", element);
}
public String getAttribute(WebElement element,String attribute) {
return (String) executerGetObject("return getAttribute(arguments[0],\""+attribute+"\");", element);
}
public void selectCheckbox(WebElement parentElement,String label) {
executerGetObject("return selectCheckbox(\""+label+"\",arguments[0]);", parentElement);
}
public void selectCheckbox(String label) {
executerGetObject("return selectCheckbox(\""+label+"\");");
}
public void selectRadio(WebElement parentElement,String label) {
executerGetObject("return selectRadio(\""+label+"\",arguments[0]);", parentElement);
}
public void selectRadio(String label) {
executerGetObject("return selectRadio(\""+label+"\");");
}
public void selectDropdown(WebElement parentElement,String label) {
executerGetObject("return selectDropdown(\""+label+"\",arguments[0]);", parentElement);
}
public void selectDropdown(String label) {
executerGetObject("return selectDropdown(\""+label+"\");");
}
}