com.automationrockstars.gir.ui.part.WebUiPartDelegate Maven / Gradle / Ivy

Go to download
/*
 * 
 */

package com.automationrockstars.gir.ui.part;

import com.automationrockstars.base.ConfigLoader;
import com.automationrockstars.design.gir.webdriver.*;
import com.automationrockstars.gir.ui.FilteredBy;
import com.automationrockstars.gir.ui.Timeout;
import com.automationrockstars.gir.ui.UiPart;
import com.google.common.base.Predicate;
import com.google.common.collect.FluentIterable;
import com.google.common.collect.Lists;
import org.openqa.selenium.By;
import org.openqa.selenium.SearchContext;
import org.openqa.selenium.WebDriverException;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.support.pagefactory.ByChained;
import org.openqa.selenium.support.ui.FluentWait;
import ru.yandex.qatools.htmlelements.element.Link;

import java.util.concurrent.TimeUnit;
import java.util.function.Function;


public class WebUiPartDelegate extends AbstractUiPartDelegate {

    public WebUiPartDelegate(Class view) {
        super(view);
        initialPageSetUp();
    }

    public WebUiPartDelegate(Class generic, UiObject toWrap) {
        super(generic, toWrap);
    }

    public FluentIterable children() {
        return FluentIterable.from(this.findElements(org.openqa.selenium.By.xpath(".//*"))).transform(new com.google.common.base.Function() {
            int instance = 0;

            public UiObject apply(WebElement input) {
                return UiObject.wrap(input, org.openqa.selenium.By.xpath(".//*[" + (instance++) + "]"));
            }
        });

    }

    public boolean has(org.openqa.selenium.By element) {
        return Waits.visible(element).apply(FilterableSearchContext.unwrap(this)) != null;
    }

    public boolean hasText(String text) {
        return getText().contains(text);
    }

    public boolean isPresent() {
        if (Page.isElementPresent(getLocator())) {
            return !new FilterableSearchContext(DriverFactory.getUnwrappedDriver()).findElements(getLocator()).isEmpty();
        } else return false;

    }

    public boolean isVisible() {
        try {
            return Page.isElementVisible(getLocator());
        } catch (WebDriverException e) {
            return false;
        }
    }


    public WebElement getWrappedElement() {
        if (!loaded.get()) {
            initialPageSetUp();
        }
        if (wrapped == null) {
            wrapped = new UiObject(getLocator());
            wrapped.setName(name());
            if (view.getAnnotation(Timeout.class) != null) {
                wrapped.setTimeout(view.getAnnotation(Timeout.class).value());
            } else if (view.getAnnotation(ru.yandex.qatools.htmlelements.annotations.Timeout.class) != null) {
                wrapped.setTimeout(view.getAnnotation(ru.yandex.qatools.htmlelements.annotations.Timeout.class).value());
            }

        }
        return wrapped;
    }

    @SuppressWarnings({"unchecked", "rawtypes"})
    void initialPageSetUp() {
        if (view.getAnnotation(InitialPage.class) != null) {
            if (view.getAnnotation(InitialPage.class).restartBrowser()) {
                DriverFactory.closeDriver();
            }
            if (!DriverFactory.canScreenshot() ||
                    view.getAnnotation(InitialPage.class).reload()
                    || !DriverFactory.getDriver().getCurrentUrl().startsWith("http")) {
                String url = (String) FluentIterable.from(Lists.newArrayList(DriverFactory.url(),
                        ConfigLoader.config().getString("url"),
                        view.getAnnotation(InitialPage.class).url())).firstMatch(new Predicate() {

                    @Override
                    public boolean apply(Object o) {
                        return o != null;
                    }
                }).get();
                DriverFactory.getDriver().get(url);
            }
            if (!loaded.get()) {
                String desiredUrl = ConfigLoader.config().getString("url", view.getAnnotation(InitialPage.class).url());
                if (desiredUrl != null && !DriverFactory.getDriver().getCurrentUrl().contains(desiredUrl)) {
                    DriverFactory.getDriver().get(desiredUrl);
                }
                loaded.set(Boolean.TRUE);
            }
            handleBrowserSpecifics();
        }
    }

    public void handleIeSpecifics() {
        if (DriverFactory.getDriver().getTitle().contains("Certificate Error")) {
            DriverFactory.getDriver().findElement(By.name("overridelink")).click();
        }

    }

    public void handleEdgeSpecifics() {
        if (DriverFactory.getDriver().getTitle().contains("Certificate error")) {
            DriverFactory.getDriver().findElement(By.id("invalidcert_continue")).click();
        }

    }

    public void handleBrowserSpecifics() {
        if (DriverFactory.isIe()) {
            handleIeSpecifics();
        } else if (DriverFactory.isEdge()) {
            handleEdgeSpecifics();
        }
    }


    @Override
    public void waitForHidden() {

        new FluentWait(this)
                .withTimeout(60, TimeUnit.SECONDS)
                .pollingEvery(200, TimeUnit.MILLISECONDS)
                .until(new Function() {

                    @Override
                    public Boolean apply(UiPart input) {
                        if (input.isPresent()) {
                            return !input.isVisible();
                        }
                        return true;
                    }

                    @Override
                    public String toString() {
                        return String.format("UiPart %s hidden", name());
                    }
                });

    }

    @Override
    public UiObject childWithText(String text) {
        return waitForChild(new FilteredBy(By.xpath(".//*"), "text.contains('" + text + "')"));

    }

    private UiObject waitForChild(By childBy) {
        final By theChildBy = new ByChained(getLocator(), childBy);
        return UiObject.wrap(delay().until(new Function() {
            @Override
            public WebElement apply(SearchContext input) {
                return input.findElement(theChildBy);
            }
        }), theChildBy);
    }

    @Override
    public Link childLinkWithText(String text) {
        return new Link(waitForChild(By.partialLinkText(text)));
    }


}