SeleniumGridHub.java

/*
 * (C) Copyright 2014 Kurento (http://kurento.org/)
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Lesser General Public License
 * (LGPL) version 2.1 which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/lgpl-2.1.html
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 */
package com.kurento.kmf.test.services;

import org.openqa.grid.internal.utils.GridHubConfiguration;
import org.openqa.grid.web.Hub;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Selenium Grid Hub.
 * 
 * @author Boni Garcia (bgarcia@gsyc.es)
 * @since 4.2.5
 */
public class SeleniumGridHub {

	public static Logger log = LoggerFactory.getLogger(SeleniumGridHub.class);

	private static final int DEFAULT_TIMEOUT = 60;

	private String ip;
	private int port;
	private int timeout;
	private Hub hub;

	public SeleniumGridHub(String ip, int port) {
		this.ip = ip;
		this.port = port;
		this.timeout = DEFAULT_TIMEOUT; // Default timeout
	}

	public void start() throws Exception {
		GridHubConfiguration config = new GridHubConfiguration();
		config.setHost(this.ip);
		config.setPort(this.port);
		config.setTimeout(getTimeout());

		hub = new Hub(config);
		log.info("Starting hub on {}:{}", this.ip, this.port);
		hub.start();
	}

	public void stop() throws Exception {
		if (hub != null) {
			hub.stop();
		}
	}

	public int getTimeout() {
		return timeout;
	}

	public void setTimeout(int timeout) {
		this.timeout = timeout;
	}

}