[bazel] Get the bidi tests working on the EngFlow grid

shs96c · shs96c · commit e5b46711176f · 2023-05-23T17:18:25.000+01:00
diff --git a/java/test/org/openqa/selenium/bidi/BiDiSessionCleanUpTest.java b/java/test/org/openqa/selenium/bidi/BiDiSessionCleanUpTest.java
@@ -26,14 +26,15 @@
 import org.openqa.selenium.WindowType;
 import org.openqa.selenium.firefox.FirefoxDriver;
 import org.openqa.selenium.firefox.FirefoxOptions;
+import org.openqa.selenium.testing.drivers.Browser;
 
 class BiDiSessionCleanUpTest {
 
   private FirefoxDriver driver;
 
   @Test
   void shouldNotCloseBiDiSessionIfOneWindowIsClosed() {
-    FirefoxOptions options = new FirefoxOptions();
+    FirefoxOptions options = (FirefoxOptions) Browser.FIREFOX.getCapabilities();
     // Enable BiDi
     options.setCapability("webSocketUrl", true);
 
@@ -61,7 +62,7 @@ void shouldNotCloseBiDiSessionIfOneWindowIsClosed() {
 
   @Test
   void shouldCloseBiDiSessionIfLastWindowIsClosed() {
-    FirefoxOptions options = new FirefoxOptions();
+    FirefoxOptions options = (FirefoxOptions) Browser.FIREFOX.getCapabilities();
     // Enable BiDi
     options.setCapability("webSocketUrl", true);
 
diff --git a/java/test/org/openqa/selenium/bidi/BiDiSessionTest.java b/java/test/org/openqa/selenium/bidi/BiDiSessionTest.java
@@ -24,14 +24,15 @@
 import org.junit.jupiter.api.Test;
 import org.openqa.selenium.firefox.FirefoxDriver;
 import org.openqa.selenium.firefox.FirefoxOptions;
+import org.openqa.selenium.testing.drivers.Browser;
 
 class BiDiSessionTest {
 
   private FirefoxDriver driver;
 
   @Test
   void shouldBeAbleToCreateABiDiSession() {
-    FirefoxOptions options = new FirefoxOptions();
+    FirefoxOptions options = (FirefoxOptions) Browser.FIREFOX.getCapabilities();
     // Enable BiDi
     options.setCapability("webSocketUrl", true);
 
diff --git a/java/test/org/openqa/selenium/bidi/BiDiTest.java b/java/test/org/openqa/selenium/bidi/BiDiTest.java
@@ -36,6 +36,7 @@
 import org.openqa.selenium.environment.webserver.NettyAppServer;
 import org.openqa.selenium.firefox.FirefoxDriver;
 import org.openqa.selenium.firefox.FirefoxOptions;
+import org.openqa.selenium.testing.drivers.Browser;
 
 class BiDiTest {
 
@@ -45,7 +46,7 @@ class BiDiTest {
 
   @BeforeEach
   public void setUp() {
-    FirefoxOptions options = new FirefoxOptions();
+    FirefoxOptions options = (FirefoxOptions) Browser.FIREFOX.getCapabilities();
     options.setCapability("webSocketUrl", true);
 
     driver = new FirefoxDriver(options);
