[py] Force free ports for cdp in geckodriver

SeleniumHQ · Sep 28, 2021 · 68cd181 · 68cd181
1 parent 9e8c7b3
commit 68cd181
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/py/selenium/webdriver/firefox/service.py b/py/selenium/webdriver/firefox/service.py
@@ -17,7 +17,7 @@
 
 from typing import List
 
-from selenium.webdriver.common import service
+from selenium.webdriver.common import (service, utils)
 
 
 class Service(service.Service):
@@ -48,9 +48,15 @@ def __init__(self, executable_path: str, port: int = 0, service_args: List[str]
         service.Service.__init__(
             self, executable_path, port=port, log_file=log_file, env=env)
         self.service_args = service_args or []
+        # Set a port for CDP
+        self.service_args.append("--websocket-port")
+        self.service_args.append("%d" % utils.free_port())
+        # Set the webdriver port
+        self.service_args.append("--port")
+        self.service_args.append("%d" % self.port)
 
     def command_line_args(self) -> List[str]:
-        return ["--port", "%d" % self.port] + self.service_args
+        return self.service_args
 
     def send_remote_shutdown_command(self):
         pass