parent
6859cb6493
commit
34c52c0023
@ -0,0 +1,283 @@
|
||||
package com.olexyn.tabdriver;
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Map.Entry;
|
||||
import java.util.Set;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
import com.olexyn.min.log.LogU;
|
||||
import org.openqa.selenium.By;
|
||||
import org.openqa.selenium.Capabilities;
|
||||
import org.openqa.selenium.JavascriptExecutor;
|
||||
import org.openqa.selenium.Keys;
|
||||
import org.openqa.selenium.NoSuchFrameException;
|
||||
import org.openqa.selenium.SearchContext;
|
||||
import org.openqa.selenium.WebElement;
|
||||
import org.openqa.selenium.chrome.ChromeDriver;
|
||||
import org.openqa.selenium.chrome.ChromeDriverService;
|
||||
|
||||
public class TabDriver extends ChromeDriver implements JavascriptExecutor {
|
||||
|
||||
private final Map<String, Tab> TABS = new HashMap<>();
|
||||
|
||||
public TabDriver(ChromeDriverService service, Capabilities capabilities) {
|
||||
super(service, capabilities);
|
||||
manage().timeouts().implicitlyWait(10, TimeUnit.SECONDS);
|
||||
}
|
||||
|
||||
public synchronized void registerCurrentTab(String purpose) {
|
||||
Tab tab = new Tab(getWindowHandle());
|
||||
tab.setName(getTitle());
|
||||
tab.setUrl(getCurrentUrl());
|
||||
tab.setPurpose(purpose);
|
||||
TABS.put(tab.getHandle(), tab);
|
||||
}
|
||||
|
||||
public synchronized Tab getCurrentTab() {
|
||||
return TABS.get(getWindowHandle());
|
||||
}
|
||||
|
||||
public synchronized List<Tab> getTabByPurpose(String purpose) {
|
||||
return TABS.values().stream()
|
||||
.filter(x -> x.getPurpose() == purpose)
|
||||
.collect(Collectors.toList());
|
||||
}
|
||||
|
||||
public synchronized String registerBlankTab(String purpose) {
|
||||
Set<String> openTabHandles = getWindowHandles();
|
||||
for (String openTabHandle : openTabHandles) {
|
||||
if (!TABS.containsKey(openTabHandle)) {
|
||||
Tab blankTab = new Tab(openTabHandle);
|
||||
blankTab.setName("about:blank");
|
||||
blankTab.setUrl("about:blank");
|
||||
blankTab.setPurpose(purpose);
|
||||
TABS.put(openTabHandle, blankTab);
|
||||
return openTabHandle;
|
||||
}
|
||||
}
|
||||
LogU.warnPlain("Not unregistered tab found.");
|
||||
return null;
|
||||
}
|
||||
|
||||
public synchronized String registerExistingTab(String purpose) {
|
||||
String handle = getWindowHandle();
|
||||
Tab tab = new Tab(handle);
|
||||
tab.setName(getTitle());
|
||||
tab.setUrl(getCurrentUrl());
|
||||
tab.setPurpose(purpose);
|
||||
TABS.put(handle, tab);
|
||||
return handle;
|
||||
}
|
||||
|
||||
public synchronized void switchToTab(String handle) {
|
||||
for (Entry<String, Tab> entry : TABS.entrySet()) {
|
||||
String tabHandle = entry.getKey();
|
||||
if (tabHandle.equals(handle)) {
|
||||
switchTo().window(tabHandle);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public synchronized void switchToTab(Tab tab) {
|
||||
switchTo().window(tab.getHandle());
|
||||
}
|
||||
|
||||
/**
|
||||
* Opens a new tab, and "moves" the WebDriver to the new tab.
|
||||
* If the current tab is empty, it is registered - this happens usually only with the initial tab of the session.
|
||||
*/
|
||||
public synchronized void newTab(String purpose) {
|
||||
String currentUrl = getCurrentUrl();
|
||||
if (currentUrl.isEmpty()
|
||||
|| currentUrl.equals("data:,")
|
||||
|| currentUrl.equals("about:blank")) {
|
||||
registerExistingTab(purpose);
|
||||
} else {
|
||||
executeScript("window.open(arguments[0])");
|
||||
switchToTab(registerBlankTab(purpose));
|
||||
}
|
||||
}
|
||||
|
||||
public synchronized void switchToTabByPurpose(String purpose) {
|
||||
List<Tab> existingTabs = getTabByPurpose(purpose);
|
||||
if (!existingTabs.isEmpty()) {
|
||||
switchToTab(existingTabs.get(0));
|
||||
} else {
|
||||
Tab currentTab = getCurrentTab();
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
public synchronized void refresh() {
|
||||
navigate().refresh();
|
||||
}
|
||||
|
||||
@Override
|
||||
public synchronized void get(String url) {
|
||||
super.get(url);
|
||||
}
|
||||
|
||||
@Override
|
||||
public synchronized WebElement findElement(By by) {
|
||||
return super.findElement(by);
|
||||
}
|
||||
|
||||
public synchronized void executeScript(String script) {
|
||||
((JavascriptExecutor) this).executeScript(script);
|
||||
}
|
||||
|
||||
public synchronized void sendDeleteKeys(WebElement element, int n) {
|
||||
for (int i = 0; i < n; i++) {
|
||||
element.sendKeys(Keys.BACK_SPACE);
|
||||
}
|
||||
}
|
||||
|
||||
public synchronized WebElement filterElementListBy(List<WebElement> list, CRITERIA criteria, String text) {
|
||||
for (WebElement element : list) {
|
||||
String toEvaluate = switch (criteria) {
|
||||
case CLASS -> element.getClass().getName();
|
||||
case TEXT -> element.getText();
|
||||
case TAG -> element.getTagName();
|
||||
case HREF -> element.getAttribute("href");
|
||||
case ID -> element.getAttribute("id");
|
||||
case TITLE -> element.getAttribute("title");
|
||||
case NONE -> text;
|
||||
};
|
||||
if (toEvaluate != null && toEvaluate.contains(text)) { return element; }
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
private static final String FRAME_ID_DEFAULT_CONTENT = "defaultContent";
|
||||
private static final String FRAME_ID_NONE_FOUND = "noneFound";
|
||||
|
||||
/**
|
||||
* Collects all frames accessible to WebDriver.
|
||||
*/
|
||||
public synchronized Map<String, String> collectAllFrames() throws NoSuchFrameException {
|
||||
Map<String, String> mapOfCollectedSources = new HashMap<>();
|
||||
switchTo().defaultContent();
|
||||
mapOfCollectedSources.put(FRAME_ID_DEFAULT_CONTENT, getPageSource());
|
||||
for (int i = 0; i < 10; i++) {
|
||||
try {
|
||||
switchTo().defaultContent();
|
||||
switchTo().frame(i);
|
||||
mapOfCollectedSources.put(String.valueOf(i), getPageSource());
|
||||
} catch (NoSuchFrameException e) {
|
||||
return mapOfCollectedSources;
|
||||
}
|
||||
}
|
||||
return mapOfCollectedSources;
|
||||
}
|
||||
|
||||
public synchronized String findFrameContainingCharSeq(Map<String, String> mapOfCollectedSources, String string) {
|
||||
for (Entry<String, String> entry : mapOfCollectedSources.entrySet()) {
|
||||
if (entry.getValue().contains(string)) {
|
||||
return entry.getKey();
|
||||
}
|
||||
}
|
||||
return FRAME_ID_NONE_FOUND;
|
||||
}
|
||||
|
||||
// @Override
|
||||
// public boolean isJavascriptEnabled() {
|
||||
// return false;
|
||||
// }
|
||||
|
||||
public enum CRITERIA {
|
||||
CLASS,
|
||||
TEXT,
|
||||
TAG,
|
||||
HREF,
|
||||
NONE,
|
||||
ID,
|
||||
TITLE
|
||||
}
|
||||
|
||||
public synchronized void switchToFrameContainingCharSeq(String charSeq) {
|
||||
switchTo().defaultContent();
|
||||
sleep(500);
|
||||
final String frameId = findFrameContainingCharSeq(collectAllFrames(), charSeq);
|
||||
sleep(400);
|
||||
switch (frameId) {
|
||||
case FRAME_ID_DEFAULT_CONTENT:
|
||||
switchTo().defaultContent();
|
||||
break;
|
||||
case FRAME_ID_NONE_FOUND:
|
||||
break;
|
||||
default:
|
||||
switchTo().frame(Integer.parseInt(frameId));
|
||||
}
|
||||
}
|
||||
|
||||
public static void sleep(long milli) {
|
||||
try {
|
||||
Thread.sleep(milli);
|
||||
} catch (InterruptedException e) {
|
||||
LogU.warnPlain("SLEEP was INTERRUPED.");
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Return the first occurrence of specified class that has specified label.
|
||||
*/
|
||||
public synchronized WebElement getWhere(String className, CRITERIA criteria, String text) {
|
||||
switchToFrameContainingCharSeq(text);
|
||||
List<WebElement> elements = findElements(By.className(className));
|
||||
return filterElementListBy(elements, criteria, text);
|
||||
}
|
||||
|
||||
public synchronized WebElement getWhere(String className) {
|
||||
switchToFrameContainingCharSeq(className);
|
||||
List<WebElement> elements = findElements(By.className(className));
|
||||
return filterElementListBy(elements, CRITERIA.NONE, Constants.EMPTY);
|
||||
}
|
||||
|
||||
public synchronized void followContainedLink(WebElement element) {
|
||||
String link = element.getAttribute("href");
|
||||
if (link != null) { navigate().to(link); }
|
||||
}
|
||||
|
||||
public synchronized void setRadio(By by, boolean checked) {
|
||||
((JavascriptExecutor) this).executeScript("arguments[0].checked = " + checked + ";", findElement(by));
|
||||
}
|
||||
|
||||
public synchronized void setComboByDataValue(By comboBy, String dataValue) {
|
||||
WebElement combo = findElement(comboBy);
|
||||
combo.click();
|
||||
combo.findElement(By.cssSelector("li[data-value='" + dataValue + "']")).click();
|
||||
}
|
||||
|
||||
/**
|
||||
* Any-Match.
|
||||
*/
|
||||
public synchronized WebElement getByFieldValue(String type, String field, String value) {
|
||||
return findElement(By.cssSelector(type + "[" + field + "*='" + value + "']"));
|
||||
}
|
||||
|
||||
public synchronized void clickByFieldValue(String type, String field, String value) {
|
||||
var we = getByFieldValue(type, field, value);
|
||||
click(we);
|
||||
}
|
||||
|
||||
/**
|
||||
* @param field can contain wildcards like "class*" or "class^"
|
||||
* @param value can contain partial matches like "last-"
|
||||
*/
|
||||
public synchronized WebElement getByFieldValue(SearchContext searchContext, String type, String field, String value) {
|
||||
|
||||
return searchContext.findElement(By.cssSelector(type + "[" + field + "='" + value + "']"));
|
||||
}
|
||||
|
||||
public synchronized WebElement getByText(String text) {
|
||||
return findElement(By.xpath("//*[contains(text(),'" + text + "')]"));
|
||||
}
|
||||
|
||||
public synchronized void click(WebElement we) {
|
||||
executeScript("arguments[0].click();", we);
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,59 @@
|
||||
package com.olexyn.tabdriver;
|
||||
|
||||
import java.nio.file.Path;
|
||||
import java.util.HashMap;
|
||||
|
||||
import com.olexyn.PropConf;
|
||||
import lombok.experimental.UtilityClass;
|
||||
import org.openqa.selenium.chrome.ChromeDriverService;
|
||||
import org.openqa.selenium.chrome.ChromeOptions;
|
||||
import org.openqa.selenium.remote.CapabilityType;
|
||||
import org.openqa.selenium.remote.DesiredCapabilities;
|
||||
|
||||
@UtilityClass
|
||||
public class TabDriverBuilder {
|
||||
|
||||
private static Path CONFIG_PATH;
|
||||
|
||||
public static TabDriver build(Path path) {
|
||||
CONFIG_PATH = path;
|
||||
if (CONFIG_PATH == null) {
|
||||
throw new RuntimeException("CONFIG_PATH must be set to an absolute path.");
|
||||
}
|
||||
return new TabDriver(configureService(), configureCapabilities());
|
||||
}
|
||||
|
||||
|
||||
private static ChromeDriverService configureService() {
|
||||
PropConf.loadProperties(CONFIG_PATH.toString());
|
||||
var path = Path.of(PropConf.get("chrome.driver.path"));
|
||||
return new ChromeDriverService.Builder()
|
||||
.usingDriverExecutable(path.toFile())
|
||||
.usingAnyFreePort()
|
||||
.build();
|
||||
}
|
||||
|
||||
private static DesiredCapabilities configureCapabilities() {
|
||||
var cap = DesiredCapabilities.chrome();
|
||||
cap.setCapability(CapabilityType.ACCEPT_SSL_CERTS, true);
|
||||
|
||||
ChromeOptions options = new ChromeOptions();
|
||||
options.addArguments("--start-maximized");
|
||||
if (PropConf.is("headless")) {
|
||||
options.addArguments("--window-size=1920,1080");
|
||||
options.addArguments("--headless");
|
||||
}
|
||||
// see also https://chromium.googlesource.com/chromium/src/+/master/chrome/common/pref_names.cc
|
||||
HashMap<String, Object> chromePrefs = new HashMap<>();
|
||||
chromePrefs.put("profile.default_content_settings.popups", 0);
|
||||
chromePrefs.put("download.default_directory", PropConf.get("download.dir"));
|
||||
chromePrefs.put("download.prompt_for_download", false);
|
||||
options.setExperimentalOption("prefs", chromePrefs);
|
||||
cap.setCapability(ChromeOptions.CAPABILITY, options);
|
||||
return cap;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
}
|
||||
@ -1,3 +0,0 @@
|
||||
chrome.driver.path=
|
||||
headless=false
|
||||
download.dir=
|
||||
Loading…
Reference in new issue