com.rationaleemotions.server.JvmBasedSeleniumServer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of just-ask Show documentation
A simple on-demand Grid
The newest version!
package com.rationaleemotions.server;

import com.rationaleemotions.servlets.EnrollServlet;
import org.openqa.grid.internal.TestSession;
import org.openqa.grid.selenium.GridLauncherV3;
import org.openqa.selenium.net.PortProber;

import java.io.File;
import java.util.Arrays;
import java.util.logging.Level;
import java.util.logging.Logger;

/**
 * Represents a {@link ISeleniumServer} implementation that is backed by a new JVM which executes the
 * selenium server as a separate process.
 */
public class JvmBasedSeleniumServer implements ISeleniumServer {
    private static final Logger LOG = Logger.getLogger(Marker.class.getEnclosingClass().getName());
    private static final String JAVA = System.getProperty("java.home") + File.separator + "bin" + File.separator +
        "java";
    private static final String CP = "-cp";
    private static final String CLASSPATH = System.getProperty("java.class.path");
    private static final String PORT_ARG = "-port";
    private static final String MAIN_CLASS = GridLauncherV3.class.getCanonicalName();
    private Process process;
    private int port;

    private static String[] getArgs(int port) {
        return new String[] {
            JAVA,
            CP,
            CLASSPATH,
            MAIN_CLASS,
            PORT_ARG,
            Integer.toString(port)
        };
    }

    @Override
    public int getPort() {
        return port;
    }

    @Override
    public int startServer(TestSession session) throws ServerException {
        port = PortProber.findFreePort();
        String[] args = getArgs(port);
        if (LOG.isLoggable(Level.INFO)) {
            LOG.info(String.format("Spawning a Selenium server using the arguments [%s]", Arrays.toString(args)));
        }
        ProcessBuilder pb = new ProcessBuilder(getArgs(port));
        try {
            this.process = pb.start();
            return port;
        } catch (Exception e) {
            throw new ServerException(e.getMessage(), e);
        }
    }

    @Override
    public void shutdownServer() {
        if (process != null) {
            process.destroyForcibly();
            LOG.warning("***Server shutdown****");
            process = null;
        }
    }

    @Override
    public String getHost() {
        return EnrollServlet.getHubHost();
    }

    private interface Marker {
    }

}