objective.taskboard.it.AbstractUIIntegrationTest.java Source code

Introduction

Here is the source code for objective.taskboard.it.AbstractUIIntegrationTest.java

Source

/*-
 * [LICENSE]
 * Taskboard
 * ---
 * Copyright (C) 2015 - 2017 Objective Solutions
 * ---
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 * [/LICENSE]
 */
package objective.taskboard.it;

import java.io.File;
import java.util.ArrayList;

import org.apache.commons.lang3.SystemUtils;
import org.junit.After;
import org.junit.Before;
import org.junit.Rule;
import org.openqa.selenium.By;
import org.openqa.selenium.Dimension;
import org.openqa.selenium.JavascriptExecutor;
import org.openqa.selenium.Keys;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebDriverException;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.firefox.FirefoxOptions;

import com.safaribooks.junitattachments.CaptureFile;
import com.safaribooks.junitattachments.RecordAttachmentRule;

public abstract class AbstractUIIntegrationTest extends AbstractIntegrationTest {

    protected WebDriver webDriver;

    @Before
    public final void setupUIIntegrationTest() {

        String driverPath = "drivers/" + getOs() + "/marionette/64bit/geckodriver";

        if (System.getProperty("webdriver.gecko.driver") == null)
            System.setProperty("webdriver.gecko.driver", driverPath);

        if (!new File(driverPath).exists())
            throw new IllegalStateException(
                    "To run integration tests, you must run 'mvn clean install' at least once to download gecko driver");

        FirefoxOptions options = new FirefoxOptions();

        options.addPreference("dom.file.createInChild", true);
        options.addPreference("browser.link.open_newwindow", 3);
        options.addPreference("browser.link.open_newwindow.restriction", 2);
        options.addPreference("intl.accept_languages", "en");

        try {
            webDriver = new FirefoxDriver(options);
        } catch (WebDriverException ex) {
            System.err.println("UI Integration Tests Aborted: It wasn't possible to instantiate the WebDriver.");
            System.err.println("You could check if Firefox AND geckodriver are up to date.");
            System.exit(1);
        }
        webDriver.manage().window().setSize(new Dimension(1280, 1080));
    }

    private String getOs() {
        if (SystemUtils.IS_OS_MAC)
            return "osx";
        else if (SystemUtils.IS_OS_WINDOWS)
            return "windows";
        return "linux";
    }

    @CaptureFile(extension = "html")
    public String capturedDom = null;

    @CaptureFile(extension = "png")
    public byte[] capturePage = null;

    @After
    public final void cleanupThread() {
        if (webDriver == null)
            return;
        try {
            // capture the dom
            capturedDom = webDriver.getPageSource();

            // capture a screenshot
            if (webDriver instanceof TakesScreenshot) {
                capturePage = ((TakesScreenshot) webDriver).getScreenshotAs(OutputType.BYTES);
            }

            ((JavascriptExecutor) webDriver).executeScript("return window.jsErrors");

            webDriver.quit();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    @Rule
    public RecordAttachmentRule recordArtifactRule = new RecordAttachmentRule(this);

    protected void switchToFirstTab() {
        webDriver.findElement(By.cssSelector("body")).sendKeys(Keys.CONTROL + "\t");
        ArrayList<String> tabs = new ArrayList<>(webDriver.getWindowHandles());
        webDriver.switchTo().window(tabs.get(0));
    }

    protected ArrayList<String> createAndSwitchToNewTab() {
        ((JavascriptExecutor) webDriver).executeScript("window.open('about:blank','_blank');");
        ArrayList<String> tabs = new ArrayList<>(webDriver.getWindowHandles());

        webDriver.switchTo().window(tabs.get(1));
        return tabs;
    }

}