com.atanas.kanchev.testframework.selenium.driverfactory.RemoteWebDriverFactory.java Source code

Introduction

Here is the source code for com.atanas.kanchev.testframework.selenium.driverfactory.RemoteWebDriverFactory.java

Source

/*
 * Copyright 2016 Atanas Stoychev Kanchev
 * 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.atanas.kanchev.testframework.selenium.driverfactory;

import org.openqa.selenium.Platform;
import org.openqa.selenium.remote.DesiredCapabilities;
import org.openqa.selenium.remote.RemoteWebDriver;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.net.MalformedURLException;
import java.net.URL;

/**
 * <p>RemoteWebDriverFactory class.</p>
 *
 * @author Atanas Kanchev
 */
public class RemoteWebDriverFactory extends DriverConfig {

    private static final Logger logger = LoggerFactory.getLogger(RemoteWebDriverFactory.class);

    /**
     * <p>getRemoteWebDriver.</p>
     *
     * @param desiredCapabilities a {@link org.openqa.selenium.remote.DesiredCapabilities} object.
     * @return a {@link org.openqa.selenium.remote.RemoteWebDriver} object.
     */
    public RemoteWebDriver getRemoteWebDriver(DesiredCapabilities desiredCapabilities) {

        URL url = null;
        try {
            url = new URL(getHubURL());
        } catch (MalformedURLException e) {
            e.printStackTrace();
        }
        String desiredBrowserVersion = getBrowserVersion();
        String desiredPlatform = getPlatform();

        if (null != desiredPlatform && !desiredPlatform.isEmpty()) {
            desiredCapabilities.setPlatform(Platform.valueOf(desiredPlatform.toUpperCase()));
        }

        if (null != desiredBrowserVersion && !desiredBrowserVersion.isEmpty()) {
            desiredCapabilities.setVersion(desiredBrowserVersion);
        }

        return new RemoteWebDriver(url, desiredCapabilities);
    }

    //    /**
    //     * Method that relieves the remote browser version
    //     *
    //     * @return remote browser version
    //     */
    //    public String getRemoteBrowserVersion() {
    //        return ((RemoteWebDriver) ((WebContext) getCurrentContext()).getDriver()).getCapabilities().getVersion();
    //    }
    //
    //    /**
    //     * Method that retrieves to the remote browser name
    //     *
    //     * @return remote browser name
    //     */
    //    public String getRemoteBrowserName() {
    //        return ((RemoteWebDriver) ((WebContext) getCurrentContext()).getDriver()).getCapabilities().getBrowserName();
    //    }
    //
    //    /**
    //     * Method that retrieves to the remote platform name
    //     *
    //     * @return remote browser name
    //     */
    //    public String getRemotePlatformName() {
    //        return ((RemoteWebDriver) ((WebContext) getCurrentContext()).getDriver()).getCapabilities().getPlatform().name();
    //    }

    //    public String getExecutionIP() {
    //
    //        String ip = null;
    //        try {
    //            HttpCommandExecutor ce = (HttpCommandExecutor) ((RemoteWebDriver) getCurrentContext().getDriver()).getCommandExecutor();
    //            String hostName = ce.getAddressOfRemoteServer().getHost();
    //            int port = ce.getAddressOfRemoteServer().getPort();
    //            HttpHost host = new HttpHost(hostName, port);
    //            HttpClient client = HttpClientBuilder.create().build();
    //            URL sessionURL = new URL("http://" + hostName + ":" + port + "/grid/api/testsession?session=" + ((RemoteWebDriver) getCurrentContext().getDriver()).getSessionId());
    //            logger.debug("Remote session URL: " + sessionURL);
    //            BasicHttpEntityEnclosingRequest r = new BasicHttpEntityEnclosingRequest("POST", sessionURL.toExternalForm());
    //            HttpResponse response = client.execute(host, r);
    //            JSONObject object = extractObject(response);
    //            URL myURL = new URL(object.getString("proxyId"));
    //            if ((myURL.getHost() != null) && (myURL.getPort() != -1))
    //                ip = myURL.getHost();
    //        } catch (IOException e) {
    //            logger.error(e.getMessage());
    //        }
    //
    //        return ip;
    //    }
    //
    //    /**
    //     * @param resp HttpResponse
    //     * @return JSONObject
    //     * @throws IOException
    //     * @throws JSONException
    //     */
    //    private static JSONObject extractObject(HttpResponse resp) throws IOException, JSONException {
    //
    //        InputStream contents = resp.getEntity().getContent();
    //        StringWriter writer = new StringWriter();
    //        IOUtils.copy(contents, writer, "UTF8");
    //
    //        return new JSONObject(writer.toString());
    //    }
}