SeleniumContext.java

package org.lightadmin;

import org.lightadmin.util.ExtendedWebDriver;
import org.lightadmin.util.ExtendedWebDriverImpl;
import org.openqa.selenium.remote.RemoteWebDriver;

import java.net.URL;

public class SeleniumContext {

	private final ExtendedWebDriver webDriver;
	private final URL baseUrl;
	private long webDriverTimeout;

	public SeleniumContext( final RemoteWebDriver webDriver, final URL baseUrl, final long webDriverWaitTimeout ) {
		this.webDriver = new ExtendedWebDriverImpl( webDriver, webDriverWaitTimeout );
		this.baseUrl = baseUrl;
		this.webDriverTimeout = webDriverWaitTimeout;
	}

	public ExtendedWebDriver getWebDriver() {
		return webDriver;
	}

	public URL getBaseUrl() {
		return baseUrl;
	}

	public void destroy() {
		webDriver.quit();
	}

	public long getWebDriverWaitTimeout() {
		return webDriverTimeout;
	}
}