com.testvagrant.optimus.core.web.WebDriverManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of optimus-lite Show documentation
Optimus Lite API to manage test devices and create appium driver based on platform
There is a newer version: 0.1.7-beta
package com.testvagrant.optimus.core.web;

import com.testvagrant.optimus.commons.SystemProperties;
import com.testvagrant.optimus.core.exceptions.UnsupportedPlatform;
import com.testvagrant.optimus.core.models.OptimusSupportedPlatforms;
import com.testvagrant.optimus.core.models.TargetDetails;
import com.testvagrant.optimus.core.models.web.WebDriverDetails;
import com.testvagrant.optimus.core.parser.WebTestFeedParser;
import com.testvagrant.optimus.core.remote.CloudConfigBuilder;
import com.testvagrant.optimus.core.remote.RemoteUrlBuilder;
import io.appium.java_client.remote.MobileCapabilityType;
import org.openqa.selenium.Capabilities;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.remote.DesiredCapabilities;
import org.openqa.selenium.remote.RemoteWebDriver;

import java.net.URL;
import java.util.Arrays;

public class WebDriverManager {

  private static final ThreadLocal driverThreadLocal = new ThreadLocal<>();
  private WebTestFeedParser testFeedParser;
  private final DesiredCapabilities desiredCapabilities;

  public WebDriverManager() {
    testFeedParser = new WebTestFeedParser(SystemProperties.WEB_FEED);
    desiredCapabilities = testFeedParser.getDesiredCapabilities();
  }

  public WebDriverManager(DesiredCapabilities capabilities) {
    this.desiredCapabilities = capabilities;
  }

  public WebDriverDetails createDriverDetails() {
    String target = System.getProperty("runMode", "local").toLowerCase();
    return target.equals("remote") ? createRemoteDriver() : createLocalDriver();
  }

  public static void dispose() {
    if (driverThreadLocal.get() != null) {
      driverThreadLocal.get().quit();
    }
  }

  private WebDriverDetails createRemoteDriver() {
    URL url = RemoteUrlBuilder.build(new CloudConfigBuilder().build());

    String platform = (String) desiredCapabilities.getCapability(MobileCapabilityType.BROWSER_NAME);

    OptimusSupportedPlatforms optimusPlatform =
        Arrays.stream(OptimusSupportedPlatforms.values())
            .filter(
                optimusSupportedPlatform ->
                    optimusSupportedPlatform.name().equals(platform.toUpperCase()))
            .findFirst()
            .orElseThrow(UnsupportedPlatform::new);

    switch (optimusPlatform) {
      case CHROME:
      case FIREFOX:
      case SAFARI:
      case EDGE:
        driverThreadLocal.set(new RemoteWebDriver(url, desiredCapabilities));
        break;
      case UNSUPPORTED:
      default:
        throw new UnsupportedPlatform();
    }

    return getWebDriverDetails(driverThreadLocal.get());
  }

  private WebDriverDetails createLocalDriver() {
    String browser = testFeedParser.getBrowserName().trim().toLowerCase();
    WebDriver webDriver =
        browser.equals("firefox")
            ? new FirefoxDriverManager(testFeedParser).createDriver()
            : new ChromeDriverManager(testFeedParser).createDriver();

    driverThreadLocal.set(webDriver);
    return getWebDriverDetails(driverThreadLocal.get());
  }

  private WebDriverDetails getWebDriverDetails(WebDriver webDriver) {
    Capabilities capabilities = ((RemoteWebDriver) webDriver).getCapabilities();

    TargetDetails target =
        TargetDetails.builder()
            .name(capabilities.getBrowserName())
            .platformVersion(capabilities.getVersion())
            .platform(
                OptimusSupportedPlatforms.valueOf(capabilities.getBrowserName().toUpperCase()))
            .build();

    WebDriverDetails webDriverDetails =
        WebDriverDetails.builder()
            .driver((RemoteWebDriver) webDriver)
            .targetDetails(target)
            .capabilities(capabilities)
            .build();
    new WebLauncher().launch(testFeedParser.getSiteConfig(), webDriverDetails);
    return webDriverDetails;
  }
}