3030import java .time .Duration ;
3131import java .util .Arrays ;
3232import java .util .Comparator ;
33+ import java .util .List ;
3334import java .util .stream .Stream ;
3435
3536public class BrowserFactoryTests {
@@ -53,14 +54,14 @@ private IBrowserFactory getCustomFactory() {
5354 return () -> {
5455 FirefoxSettings firefoxSettings = new FirefoxSettings (AqualityServices .get (ISettingsFile .class ));
5556 WebDriverManager .firefoxdriver ().setup ();
56- FirefoxDriver driver = AqualityServices . get ( IActionRetrier . class ). doWithRetry (
57- () -> new FirefoxDriver ((( FirefoxOptions ) firefoxSettings . getDriverOptions ()). setHeadless ( true )),
58- Arrays . asList (
59- SessionNotCreatedException .class ,
60- UnreachableBrowserException .class ,
61- WebDriverException .class ,
62- TimeoutException . class ));
63-
57+ FirefoxOptions options = (( FirefoxOptions ) firefoxSettings . getDriverOptions ()). addArguments ( "--headless" );
58+ final List < Class <? extends Throwable >> handledExceptions = Arrays . asList (
59+ SessionNotCreatedException . class ,
60+ UnreachableBrowserException .class ,
61+ WebDriverException .class ,
62+ TimeoutException .class );
63+ FirefoxDriver driver = AqualityServices . get ( IActionRetrier . class )
64+ . doWithRetry (() -> new FirefoxDriver ( options ), handledExceptions );
6465 return new Browser (driver );
6566 };
6667 }
0 commit comments