package net.thucydides.core.bootstrap;
import com.google.common.base.Optional;
import net.thucydides.core.guice.Injectors;
import net.thucydides.core.model.TestOutcome;
import net.thucydides.core.pages.Pages;
import net.thucydides.core.reports.ReportService;
import net.thucydides.core.steps.*;
import net.thucydides.core.webdriver.Configuration;
import net.thucydides.core.webdriver.WebDriverFactory;
import net.thucydides.core.webdriver.WebdriverManager;
import org.openqa.selenium.WebDriver;
import java.io.File;
import java.util.List;
/**
* Container holding thread-local data related to a Thucydides test run.
* This includes the StepFactory and the associated step listeners. These need to be thread-local so that,
* if tests are run in parallel in different threads, the step listeners will still build up correct result trees
* and report data.
*/
class ThucydidesContext {
private static final ThreadLocal<ThucydidesContext> contextThreadLocal = new ThreadLocal<ThucydidesContext>();
/**
* Instruments step libraries to that any @Step-annotated methods called will appear in the test reports.
*/
private final StepFactory stepFactory;
/**
* The main step listener used to record test results and outcomes.
*/
private BaseStepListener stepListener;
/**
* Generates reports once the test outcomes have been recorded and built.
*/
private final ReportService reportService;
/**
* Where are the Thucydides reports written to.
* Normally defined in the system properties.
*/
private File outputDirectory;
private String defaultDriver;
private WebDriverFactory webDriverFactory;
private Pages pages;
/**
* Thucydides configuration data
*/
private Configuration configuration;
private WebdriverManager webdriverManager;
private ThucydidesContext(StepListener... additionalListeners) {
this(null, additionalListeners);
}
private ThucydidesContext(String defaultDriver, StepListener... additionalListeners) {
configuration = Injectors.getInjector().getInstance(Configuration.class);
webdriverManager = Injectors.getInjector().getInstance(WebdriverManager.class);
outputDirectory = configuration.getOutputDirectory();
this.defaultDriver = defaultDriver;
if (defaultDriver != null) {
webDriverFactory = new WebDriverFactory();
pages = new Pages(getDriver());
stepFactory = new StepFactory(pages);
} else {
stepFactory = new StepFactory();
}
registerStepListeners(additionalListeners);
reportService = new ReportService(outputDirectory,
ReportService.getDefaultReporters());
}
protected WebDriver getDriver() {
return webdriverManager.getWebdriver(defaultDriver);
}
public void setOutputDirectory(File outputDirectory) {
this.outputDirectory = outputDirectory;
}
private void registerStepListeners(StepListener... additionalListeners) {
stepListener = buildBaseStepListener();
StepEventBus.getEventBus().registerListener(stepListener);
for (StepListener listener : additionalListeners) {
StepEventBus.getEventBus().registerListener(listener);
}
}
public static ThucydidesContext newContext() {
return new ThucydidesContext();
}
public static ThucydidesContext newContext(Optional<String> driver, StepListener... listeners) {
ThucydidesContext context = null;
context = (driver.isPresent()) ? new ThucydidesContext(driver.get(), listeners) : new ThucydidesContext(listeners);
contextThreadLocal.set(context);
return context;
}
public static ThucydidesContext getCurrentContext() {
return contextThreadLocal.get();
}
/**
* Injects instrumented step classes into any @Step annotated fields of the specified class.
*
* @param testCase
*/
public void initialize(Object testCase) {
StepAnnotations.injectScenarioStepsInto(testCase, stepFactory);
}
public void generateReports() {
reportService.generateReportsFor(latestTestOutcomes());
}
private List<TestOutcome> latestTestOutcomes() {
return stepListener.getTestOutcomes();
}
public void dropListeners() {
StepEventBus.getEventBus().dropAllListeners();
}
private BaseStepListener buildBaseStepListener() {
return Listeners.getBaseStepListener().withOutputDirectory(outputDirectory);
}
// private BaseStepListener buildBaseStepListener() {
// if (pageFactory != null) {
// return Listeners.getBaseStepListener()
// .withPages(pageFactory)
// .and().withOutputDirectory(outputDirectory);
// } else {
// return Listeners.getBaseStepListener()
// .withOutputDirectory(outputDirectory);
// }
// }
}