30
30
import java .time .Duration ;
31
31
import java .util .Arrays ;
32
32
import java .util .Comparator ;
33
+ import java .util .List ;
33
34
import java .util .stream .Stream ;
34
35
35
36
public class BrowserFactoryTests {
@@ -53,14 +54,14 @@ private IBrowserFactory getCustomFactory() {
53
54
return () -> {
54
55
FirefoxSettings firefoxSettings = new FirefoxSettings (AqualityServices .get (ISettingsFile .class ));
55
56
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 );
64
65
return new Browser (driver );
65
66
};
66
67
}
0 commit comments