com.google.testing.web.Browser.java Source code

Introduction

Here is the source code for com.google.testing.web.Browser.java

Source

// Copyright 2016 Google Inc.
//
// 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 com.google.testing.web;

import java.net.MalformedURLException;
import java.net.URL;
import javax.annotation.Nullable;
import org.openqa.selenium.Capabilities;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.remote.Augmenter;
import org.openqa.selenium.remote.DesiredCapabilities;
import org.openqa.selenium.remote.RemoteWebDriver;

/**
 * Browser provisioning and information API.
 *
 * <p>Provision a browser:
 *
 * <pre class="code">
 * WebDriver driver = new Browser().newSession();
 * </pre>
 *
 * <p>Provision a browser with capabilities (as an example, profiling):
 *
 * <pre class="code">
 * DesiredCapabilities capabilities = new DesiredCapabilities();
 * capabilities.setCapability(CapabilityType.ENABLE_PROFILING_CAPABILITY, true);
 *
 * WebDriver driver = new Browser().newSession(capabilities);
 * </pre>
 */
public class Browser {

    @Nullable
    private final String address;

    public Browser() {
        this(System.getenv("REMOTE_WEBDRIVER_SERVER"));
    }

    private Browser(String address) {
        this.address = address;
    }

    /** Provisions and returns a new default {@link WebDriver} session. */
    public WebDriver newSession() {
        return newSession(new DesiredCapabilities());
    }

    /**
     * Provisions and returns a new {@link WebDriver} session.
     *
     * @param capabilities Configuration of the browser.
     */
    public WebDriver newSession(Capabilities capabilities) {
        DesiredCapabilities desired = new DesiredCapabilities(capabilities);
        WebDriver driver = new Augmenter().augment(new RemoteWebDriver(constructUrl(address), desired));

        return driver;
    }

    private static URL constructUrl(String hostAndPort) {
        try {
            return new URL(String.format("http://%s/wd/hub", hostAndPort));
        } catch (MalformedURLException e) {
            throw new RuntimeException(e);
        }
    }
}