org.fluentlenium.adapter.sharedwebdriver.ClassDriver Maven / Gradle / Ivy

package org.fluentlenium.adapter.sharedwebdriver;

import java.util.HashMap;
import java.util.Map;

/**
 * Stores and handles {@link SharedWebDriver} instances for test {@link Class}es.
 *
 * @see org.fluentlenium.configuration.ConfigurationProperties.DriverLifecycle#CLASS
 * @see SharedWebdriverSingletonImpl
 */
public class ClassDriver implements FluentLeniumDriver {

    private final Map, SharedWebDriver> classDrivers = new HashMap<>();

    Map, SharedWebDriver> getClassDrivers() {
        return classDrivers;
    }

    @Override
    public void quitDriver(SharedWebDriver sharedWebDriver) {
        SharedWebDriver classDriver = classDrivers.remove(sharedWebDriver.getTestClass());
        quitDriver(sharedWebDriver, classDriver);
    }

    @Override
    public void addDriver(SharedWebDriver driver) {
        classDrivers.put(driver.getTestClass(), driver);
    }

    public  SharedWebDriver getDriver(Class testClass) {
        return classDrivers.get(testClass);
    }
}