Create Selenium Driver

gistfile1.java

@Before
public void createDriver() throws MalformedURLException {
    String remoteTest = System.getProperty("selenium.remote");
    if ("true".equals(remoteTest)) {
        String browser = System.getProperty("selenium.browser", "firefox");
        String version = System.getProperty("selenium.browser.version", "");
        Platform platform = Platform.valueOf(System.getProperty("selenium.platform", "ANY"));
        
        DesiredCapabilities capabilities = new DesiredCapabilities(browser, version, platform);
        capabilities.setCapability("max-duration", "120"); // tests can't run more than 2 mins
        capabilities.setCapability("name", getClass().getName() + "." + name.getMethodName());
        driver = new RemoteWebDriver(new URL("http://sauceuserinfo@ondemand.saucelabs.com:80/wd/hub"),
                                     capabilities);
        // longer timeouts for remote invocation
        driver.manage().timeouts().implicitlyWait(120, TimeUnit.SECONDS);
    } else {
        String browser = System.getProperty("selenium.browser", "firefox");
        if ("firefox".equals(browser)) {
            driver = new FirefoxDriver();
        } else if ("ie".equals(browser)) {
            driver = new InternetExplorerDriver();
        }
        
        driver.manage().timeouts().implicitlyWait(20, TimeUnit.SECONDS);
    }
}

@After
public void closeDriver() {
    if (driver != null) {
        driver.quit();
    }
}