[java] Using se:bidi url when Grid returns it

Author: diemol

java/src/org/openqa/selenium/bidi/BiDiProvider.java

@@ -55,8 +55,14 @@ public HasBiDi getImplementation(Capabilities caps, ExecuteMethod executeMethod)
   }
 
   private Optional<URI> getBiDiUrl(Capabilities caps) {
-    Optional<String> webSocketUrl =
-        Optional.ofNullable((String) caps.getCapability("webSocketUrl"));
+    Object bidiCapability;
+    if (caps.asMap().containsKey("se:bidi")) {
+      // Session is created remotely
+      bidiCapability = caps.getCapability("se:bidi");
+    } else {
+      bidiCapability = caps.getCapability("webSocketUrl");
+    }
+    Optional<String> webSocketUrl = Optional.ofNullable((String) bidiCapability);
 
     return webSocketUrl.map(
         uri -> {