io.github.sukgu.support.BaseBy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of automation Show documentation

This project automates shadow-dom elements using java selenium

There is a newer version: 0.1.5

package io.github.sukgu.support;

import java.util.List;

import org.openqa.selenium.By;
import org.openqa.selenium.SearchContext;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.remote.RemoteWebElement;

import io.github.sukgu.Shadow;
import io.github.sukgu.support.FindElementBy.FindByBuilder.SelectorType;

abstract class BaseBy extends By {

    protected final String selector;
    protected final String selectorType;

    protected BaseBy(String selector, String selectorType) {
        this.selector = selector;
        this.selectorType = selectorType;
    }

    private final WebDriver getWebDriver(SearchContext context) {
        WebDriver webDriver;
        
        if (context instanceof RemoteWebElement) {
        	webDriver = ((RemoteWebElement) context).getWrappedDriver();
        } else {
        	webDriver = (WebDriver) context;
        }
        
        return webDriver;
    }

	@Override
    public WebElement findElement(SearchContext context) {
    	WebDriver webDriver = getWebDriver(context);
    	
        if (context instanceof WebDriver) {
            context = null;
        }
        
        Shadow shadow = new Shadow(webDriver);
        
        if(selectorType.equals(SelectorType.CSS_SELECTOR.toString())) {
        	return shadow.findElement(selector);
        }
        
        if(selectorType.equals(SelectorType.XPATH.toString())) {
        	return shadow.findElementByXPath(selector);
        }
        
        return null;
    }

	@Override
    public List findElements(SearchContext context) {
		WebDriver webDriver = getWebDriver(context);
		
        if (context instanceof WebDriver) {
            context = null;
        }
        
        Shadow shadow = new Shadow(webDriver);
        if(selectorType.equals(SelectorType.CSS_SELECTOR.toString())) {
        	return shadow.findElements(selector);
        }
        
        if(selectorType.equals(SelectorType.XPATH.toString())) {
        	return shadow.findElementsByXPath(selector);
        }
        
        return null;
    }
}