[java] Add convenience for enabling BiDi (#14029)

Related to #13991

Author: pujagani

java/src/org/openqa/selenium/chromium/ChromiumOptions.java

@@ -186,6 +186,11 @@ public T addEncodedExtensions(List<String> encoded) {
     return (T) this;
   }
 
+  public T enableBiDi() {
+    setCapability("webSocketUrl", true);
+    return (T) this;
+  }
+
   /**
    * Sets an experimental option. Useful for new ChromeDriver options not yet exposed through the
    * {@link ChromiumOptions} API.

java/src/org/openqa/selenium/firefox/FirefoxOptions.java

@@ -294,6 +294,11 @@ private FirefoxOptions setFirefoxOption(String key, Object value) {
     return this;
   }
 
+  public FirefoxOptions enableBiDi() {
+    setCapability("webSocketUrl", true);
+    return this;
+  }
+
   @Override
   protected Set<String> getExtraCapabilityNames() {
     Set<String> names = new TreeSet<>();

java/test/org/openqa/selenium/bidi/BiDiSessionCleanUpTest.java

@@ -36,7 +36,7 @@ class BiDiSessionCleanUpTest {
   void shouldNotCloseBiDiSessionIfOneWindowIsClosed() {
     FirefoxOptions options = (FirefoxOptions) Browser.FIREFOX.getCapabilities();
     // Enable BiDi
-    options.setCapability("webSocketUrl", true);
+    options.enableBiDi();
 
     driver = new FirefoxDriver(options);
 
@@ -64,7 +64,7 @@ void shouldNotCloseBiDiSessionIfOneWindowIsClosed() {
   void shouldCloseBiDiSessionIfLastWindowIsClosed() {
     FirefoxOptions options = (FirefoxOptions) Browser.FIREFOX.getCapabilities();
     // Enable BiDi
-    options.setCapability("webSocketUrl", true);
+    options.enableBiDi();
 
     driver = new FirefoxDriver(options);
 

java/test/org/openqa/selenium/testing/drivers/Browser.java

@@ -59,7 +59,7 @@ public Capabilities getCapabilities() {
       prefs.put("exited_cleanly", true);
       options.setExperimentalOption("prefs", prefs);
 
-      options.setCapability("webSocketUrl", true);
+      options.enableBiDi();
 
       return options;
     }
@@ -90,7 +90,7 @@ public Capabilities getCapabilities() {
       prefs.put("exited_cleanly", true);
       options.setExperimentalOption("prefs", prefs);
 
-      options.setCapability("webSocketUrl", true);
+      options.enableBiDi();
 
       return options;
     }
@@ -129,7 +129,7 @@ public Capabilities getCapabilities() {
         options.addArguments("-headless");
       }
 
-      options.setCapability("webSocketUrl", true);
+      options.enableBiDi();
 
       return options;
     }