Lean BrowserFactory Pattern with Selenium WebDriver

1_pom.xml

<!-- https://mvnrepository.com/artifact/org.reflections/reflections -->
<dependency>
    <groupId>org.reflections</groupId>
    <artifactId>reflections</artifactId>
    <version>0.9.12</version>
</dependency>

2_Browser.java

public interface Browser {
    MutableCapabilities getOptions(boolean isHeadless);
    WebDriver getDriver(boolean isHeadless);
}

3_Chrome.java

public class Chrome implements Browser {
    // ChromeOptions extends MutableCapabilities satisfying interface declaration
    @Override
    public ChromeOptions getOptions(boolean isHeadless) {
        var options = new ChromeOptions();
        options.setHeadless(isHeadless);
        // Do Chrome specific settings
        return options;
    }

    @Override
    public WebDriver getDriver(boolean isHeadless) {
        return new ChromeDriver(getOptions(isHeadless));
    }
    
    // Add other methods
}

4_BrowserFactory.java

import org.reflections.Reflections;

public final class BrowserFactory {
    private boolean isHeadless;

    public BrowserFactory setHeadless(boolean isHeadless) {
        this.isHeadless = isHeadless;
        return this;
    }

    public WebDriver getInstance(String browserName) throws Exception {
        // Returns a List of classes that derives from Browser using Generics and Reflection
        // The ? is a bounded wildcard which denotes an unknown type
        List<Class<? extends Browser>> browsers = new ArrayList<>(
            new Reflections(Browser.class.getPackageName()).getSubTypesOf(Browser.class));

        // Filters the List for the class containing the browserName, instantiates the class, and gets the WebDriver
        return browsers.stream().filter(b -> b.getSimpleName().toLowerCase().contains(browserName.toLowerCase()))
            .findFirst().orElseThrow(() -> new RuntimeException("Browser not supported!")).getConstructor()
            .newInstance().getDriver(isHeadless);
    }
}

5_BaseTest.java

public abstract class BaseTest {
    protected WebDriver driver;

    @BeforeEach
    public void setUp() throws Exception {
        // Set environment variables
        driver = new BrowserFactory()
                    .setHeadless(Boolean.parseBoolean(System.getenv("SELENIUM_HEADLESS")))
                    .getInstance(System.getenv("SELENIUM_BROWSER"));
        // More code
    }

    @AfterEach
    public void tearDown() {
        driver.quit();
    }
}