[java] Adding --allow-origins when starting GeckoDriver

diemol · diemol · commit 46cef40264f0 · 2022-05-07T15:22:23.000+02:00
Needed for Firefox 100 and above. GeckoDriver 0.31.0 or above
is needed for this.
diff --git a/java/src/org/openqa/selenium/firefox/GeckoDriverService.java b/java/src/org/openqa/selenium/firefox/GeckoDriverService.java
@@ -17,10 +17,6 @@
 
 package org.openqa.selenium.firefox;
 
-import static java.util.Collections.unmodifiableList;
-import static java.util.concurrent.TimeUnit.MILLISECONDS;
-import static org.openqa.selenium.remote.Browser.FIREFOX;
-
 import com.google.auto.service.AutoService;
 import com.google.common.io.ByteStreams;
 
@@ -39,6 +35,10 @@
 import java.util.Map;
 import java.util.Optional;
 
+import static java.util.Collections.unmodifiableList;
+import static java.util.concurrent.TimeUnit.MILLISECONDS;
+import static org.openqa.selenium.remote.Browser.FIREFOX;
+
 /**
  * Manages the life and death of an GeckoDriver aka 'wires'.
  */
@@ -187,15 +187,18 @@ protected File findDefaultExecutable() {
     @Override
     protected List<String> createArgs() {
       List<String> args = new ArrayList<>();
+      int wsPort = PortProber.findFreePort();
       args.add(String.format("--port=%d", getPort()));
-      args.add(String.format("--websocket-port=%d", PortProber.findFreePort()));
+      args.add(String.format("--websocket-port=%d", wsPort));
+      args.add(String.format("--allow-origins=http://localhost:%d", wsPort));
       if (firefoxBinary != null) {
         args.add("-b");
         args.add(firefoxBinary.getPath());
       } else {
         // Read system property for Firefox binary and use those if they are set
-        Optional<Executable> executable = Optional.ofNullable(FirefoxBinary.locateFirefoxBinaryFromSystemProperty());
-        executable.ifPresent( e -> {
+        Optional<Executable> executable =
+          Optional.ofNullable(FirefoxBinary.locateFirefoxBinaryFromSystemProperty());
+        executable.ifPresent(e -> {
           args.add("-b");
           args.add(e.getPath());
         });
