org.apache.atlas.objectwrapper.WebDriverWrapper.java Source code

Introduction

Here is the source code for org.apache.atlas.objectwrapper.WebDriverWrapper.java

Source

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 org.apache.atlas.objectwrapper;

import org.apache.atlas.seleniumtests.HomePageTest;
import org.apache.atlas.seleniumtests.PaginationTest;
import org.apache.atlas.utilities.AtlasDriverUtility;
import org.apache.log4j.PropertyConfigurator;
import org.atlas.testHelper.AtlasConstants;
import org.atlas.ui.pages.HomePage;
import org.atlas.ui.pages.Pagination;
import org.openqa.selenium.Alert;
import org.openqa.selenium.Dimension;
import org.openqa.selenium.Platform;
import org.openqa.selenium.Point;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.remote.DesiredCapabilities;
import org.openqa.selenium.support.PageFactory;
import org.openqa.selenium.support.events.EventFiringWebDriver;
import org.testng.log4testng.Logger;
import org.testng.xml.XmlTest;

import com.apache.atlas.listener.AtlasDriverEventListener;

/**
 * Base class for UI test classes.
 */
public class WebDriverWrapper {
    private static final Logger LOGGER = Logger.getLogger(WebDriverWrapper.class);

    private static WebDriver webDriver;
    protected static EventFiringWebDriver driver;
    protected static HomePage homePage;
    protected static Pagination Pagination;
    protected static WebElementWrapper webElement;
    protected long testExecutionStartTime;

    public static void setDriver(EventFiringWebDriver driver) {
        WebDriverWrapper.driver = driver;
    }

    private static WebDriver appConfig(XmlTest config) {
        //      AtlasConstants.UI_URL = config.getParameter("app_url");
        String serverIP = config.getParameter("server_ip");
        String serverPort = config.getParameter("server_port");
        String browserName = config.getParameter("browserName");
        int browserHeight = Integer.parseInt(config.getParameter("browser_window_height"));
        int browserWidth = Integer.parseInt(config.getParameter("browser_window_width"));
        DesiredCapabilities capabilities = null;

        if (browserName.contains("firefox")) {
            capabilities = DesiredCapabilities.firefox();
            capabilities.setPlatform(Platform.ANY);
            webDriver = new FirefoxDriver(capabilities);
        } else if (browserName.contains("chrome") || browserName.equalsIgnoreCase("chrome")) {

            LOGGER.info("Configuring settings for Chrome");
            // Google Chrome Driver
            ChromeOptions options = new ChromeOptions();
            options.addArguments("--disable-web-security");
            options.addArguments("--start-maximized");

            // For use with RemoteWebDriver:
            capabilities = DesiredCapabilities.chrome();
            capabilities.setCapability(ChromeOptions.CAPABILITY, options);
            webDriver = new ChromeDriver(capabilities);
        }
        AtlasConstants.UI_URL = "http://" + serverIP + ":" + serverPort;
        webDriver.manage().window().setPosition(new Point(0, 0));
        Dimension dimenssion = new Dimension(browserHeight, browserWidth);
        webDriver.manage().window().setSize(dimenssion);
        return webDriver;
    }

    public static WebDriver getDriver() {
        if (driver != null) {
            return driver;
        }
        return null;
    }

    public static WebDriver setupConfig(XmlTest config) {
        driver = new EventFiringWebDriver(appConfig(config));
        AtlasDriverEventListener driverLister = new AtlasDriverEventListener();
        driver.register(driverLister);
        driver.manage().window().maximize();
        webElement = new WebElementWrapper();
        PropertyConfigurator.configure("log4j.properties");
        homePage = new HomePage();
        Pagination = new Pagination();
        return driver;
    }

    public HomePageTest launchApp() {
        long startTime = System.currentTimeMillis();
        driver.get(AtlasConstants.UI_URL);
        LOGGER.info("Opened a URL: " + AtlasConstants.UI_URL);
        AtlasDriverUtility.waitForPageLoad(driver, 30);
        AtlasDriverUtility.pageLoadedTime(startTime, AtlasConstants.UI_URL);
        return PageFactory.initElements(driver, HomePageTest.class);
    }

    public PaginationTest launchApp1() {
        long startTime = System.currentTimeMillis();
        driver.get(AtlasConstants.UI_URL);
        LOGGER.info("Opened a URL: " + AtlasConstants.UI_URL);
        AtlasDriverUtility.waitForPageLoad(driver, 30);
        AtlasDriverUtility.pageLoadedTime(startTime, AtlasConstants.UI_URL);
        return PageFactory.initElements(driver, PaginationTest.class);
    }

    public static void closeBrowser() {
        if (driver != null) {
            AtlasDriverUtility.waitForPageLoad(driver, AtlasConstants.DRIVER_TIMEOUT);
            driver.close();
            driver.quit();
            driver = null;
        }
    }

    public void switchToWindow(String browserName) {
        driver.switchTo().window(browserName);
    }

    public Alert switchToAlert() {
        return driver.switchTo().alert();
    }

}