stormy.pythian.features.web.support.WebConnector.java Source code

Introduction

Here is the source code for stormy.pythian.features.web.support.WebConnector.java

Source

/**
 * Copyright 2013-2015 Pierre Merienne
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package stormy.pythian.features.web.support;

import static java.util.concurrent.TimeUnit.SECONDS;
import static stormy.pythian.features.web.support.Environment.BASE_HTML_PATH;
import java.util.ArrayList;
import java.util.List;
import org.openqa.selenium.By;
import org.openqa.selenium.Dimension;
import org.openqa.selenium.Keys;
import org.openqa.selenium.NoSuchElementException;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.interactions.Actions;
import org.openqa.selenium.phantomjs.PhantomJSDriver;
import org.openqa.selenium.remote.RemoteWebDriver;
import org.openqa.selenium.support.ui.Select;
import com.google.common.base.Function;
import com.google.common.collect.FluentIterable;
import com.google.common.collect.Iterables;

public class WebConnector {

    private final static boolean HEADLESS = true;

    private final static RemoteWebDriver driver;
    private static final Dimension WINDOW_SIZE = new Dimension(1600, 900);

    /**
     * Static init to avoid lots of initialization causing bugs and timeouts!
     */
    static {
        if (HEADLESS) {
            driver = new PhantomJSDriver();
        } else {
            System.setProperty("webdriver.chrome.driver", "/opt/chromedriver/chromedriver");
            driver = new ChromeDriver();
        }

        driver.manage().window().setSize(WINDOW_SIZE);
        driver.manage().timeouts().implicitlyWait(2, SECONDS).setScriptTimeout(2, SECONDS).pageLoadTimeout(10,
                SECONDS);
    }

    public void press(Keys keys) {
        new Actions(driver).sendKeys(keys).build().perform();
        wait_for_angular_requests_to_finish();
    }

    public void fill(String id, String text) {
        WebElement element = driver.findElementById(id);
        fill(element, text);
    }

    public void fill(By by, String text) {
        WebElement element = driver.findElement(by);
        fill(element, text);
    }

    public void fill(WebElement element, String text) {
        element.clear();
        element.sendKeys(text);
    }

    public void scroll_into_view(WebElement element) {
        driver.executeScript("arguments[0].scrollIntoView(true);", element);
        wait_for_angular_requests_to_finish();
    }

    public void click(WebElement element) {
        element.click();
        wait_for_angular_requests_to_finish();
    }

    public void click(By by) {
        WebElement element = driver.findElement(by);
        click(element);
    }

    public void click(String selector) {
        click(By.id(selector));
    }

    public void drag_and_drop(WebElement source, WebElement destination) {
        new Actions(driver).clickAndHold(source).moveByOffset(1, 1).moveToElement(destination).moveByOffset(1, 1)
                .release().build().perform();
        wait_for_angular_requests_to_finish();
    }

    public void drag_and_drop(String elementId, int dx, int dy) {
        WebElement element = driver.findElement(By.id(elementId));
        drag_and_drop(element, dx, dy);
    }

    public void drag_and_drop(WebElement source, int dx, int dy) {
        new Actions(driver).clickAndHold(source).moveByOffset(dx, dy).moveByOffset(1, 1).release().build()
                .perform();

        wait_for_angular_requests_to_finish();
    }

    public WebElement retrieve_element(By by) {
        try {
            return driver.findElement(by);
        } catch (NoSuchElementException ex) {
            return null;
        }
    }

    public List<WebElement> retrieve_elements(By by) {
        try {
            return driver.findElements(by);
        } catch (NoSuchElementException ex) {
            return new ArrayList<>();
        }
    }

    public WebElement retrieve_last(By by) {
        try {
            return Iterables.getLast(driver.findElements(by));
        } catch (NoSuchElementException ex) {
            return null;
        }
    }

    public boolean element_exists(String id) {
        return element_exists(By.id(id));
    }

    public boolean element_exists(By by) {
        try {
            WebElement element = driver.findElement(by);
            return element != null;
        } catch (NoSuchElementException ex) {
            return false;
        }
    }

    public void select(WebElement selectElement, String value) {
        new Select(selectElement).selectByVisibleText(value);
        wait_for_angular_requests_to_finish();
    }

    public void select(By by, String value) {
        WebElement selectElement = retrieve_element(by);
        new Select(selectElement).selectByVisibleText(value);
        wait_for_angular_requests_to_finish();
    }

    public List<String> get_available_options(WebElement select_element) {
        return FluentIterable.from(new Select(select_element).getOptions())
                .transform(new Function<WebElement, String>() {
                    public String apply(WebElement input) {
                        return input.getText();
                    }
                }).toList();
    }

    public String retrieve_selected(By by) {
        WebElement selectElement = retrieve_element(by);
        return new Select(selectElement).getFirstSelectedOption().getText();
    }

    public String retrieve_selected(WebElement selectElement) {
        return new Select(selectElement).getFirstSelectedOption().getText();
    }

    public void open(String location) {
        driver.get(location);
        wait_for_angular_requests_to_finish();
    }

    public RemoteWebDriver driver() {
        return driver;
    }

    public String relative_location() {
        return driver.getCurrentUrl().replaceAll(BASE_HTML_PATH, "");
    }

    public void wait_for_angular_requests_to_finish() {
        try {
            Thread.sleep(200);
            driver.executeAsyncScript("var callback = arguments[arguments.length - 1];"
                    + "angular.element(document.body).injector().get('$browser').notifyWhenNoOutstandingRequests(callback);");
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        }
    }

    public Object executeScript(String script, Object... args) {
        return driver.executeScript(script, args);
    }
}