net.thucydides.core.webdriver.WebdriverCloseBrowser Maven / Gradle / Ivy
package net.thucydides.core.webdriver;
import com.google.inject.Inject;
import net.serenitybdd.core.webdriver.configuration.RestartBrowserForEach;
import net.thucydides.core.guice.Injectors;
import net.thucydides.core.junit.SerenityJUnitTestCase;
import net.thucydides.core.statistics.TestCount;
import net.thucydides.core.util.EnvironmentVariables;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebDriverException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class WebdriverCloseBrowser implements CloseBrowser {
private final EnvironmentVariables environmentVariables;
private static final Logger LOGGER = LoggerFactory.getLogger(WebDriverFacade.class);
private final TestCount testCount;
private final PeriodicRestart shouldRestartBrowser;
@Inject
public WebdriverCloseBrowser(EnvironmentVariables environmentVariables) {
this.environmentVariables = environmentVariables;
this.testCount = Injectors.getInjector().getInstance(TestCount.class);
this.shouldRestartBrowser = new PeriodicRestart(environmentVariables);
}
@Override
public void closeIfConfiguredForANew(RestartBrowserForEach event) {
if (restartBrowserForANew(event) || shouldRestartBrowserPeriodicallyNow()) {
ThucydidesWebDriverSupport.closeCurrentDrivers();
}
}
private boolean shouldRestartBrowserPeriodicallyNow() {
return shouldRestartBrowser.forTestNumber(testCount.getCurrentTestNumber());
}
private boolean restartBrowserForANew(RestartBrowserForEach event) {
return RestartBrowserForEach.configuredIn(environmentVariables).restartBrowserForANew(event);
}
@Override
public void closeWhenTheTestsAreFinished(final WebDriver driver) {
Runtime.getRuntime().addShutdownHook(new Thread() {
@Override
public void run() {
if (driver != null) {
try {
driver.quit();
} catch (WebDriverException mostLikelyLostContactWithTheBrowser) {
LOGGER.debug("Failed to close a browser: {}", mostLikelyLostContactWithTheBrowser.getMessage());
}
}
}
});
}
@Override
public CloseBrowser forTestSuite(Class testSuite) {
if (testSuite == null) {
return this;
}
if (SerenityJUnitTestCase.inClass(testSuite).isAWebTest()) {
return new TestSuiteCloseBrowser(environmentVariables, testSuite);
}
return new DeactivatedCloseBrowser();
}
}