Add option to override the driver version used

Author: mdmintz

seleniumbase/behave/behave_sb.py

@@ -40,6 +40,7 @@
 -D extension-zip=ZIP  (Load a Chrome Extension .zip|.crx, comma-separated.)
 -D extension-dir=DIR  (Load a Chrome Extension directory, comma-separated.)
 -D binary-location=PATH  (Set path of the Chromium browser binary to use.)
+-D driver-version=VER  (Set the chromedriver or uc_driver version to use.)
 -D sjw  (Skip JS Waits for readyState to be "complete" or Angular to load.)
 -D pls=PLS  (Set pageLoadStrategy on Chrome: "normal", "eager", or "none".)
 -D headless  (Run tests in headless mode. The default arg on Linux OS.)
@@ -164,6 +165,7 @@ def get_configured_sb(context):
     sb.extension_zip = None
     sb.extension_dir = None
     sb.binary_location = None
+    sb.driver_version = None
     sb.page_load_strategy = None
     sb.database_env = "test"
     sb.log_path = "latest_logs" + os.sep
@@ -459,6 +461,13 @@ def get_configured_sb(context):
                 binary_location = sb.binary_location  # revert to default
             sb.binary_location = binary_location
             continue
+        # Handle: -D driver-version=VER / driver_version=VER
+        if low_key in ["driver-version", "driver_version"]:
+            driver_version = userdata[key]
+            if driver_version == "true":
+                driver_version = sb.driver_version  # revert to default
+            sb.driver_version = driver_version
+            continue
         # Handle: -D pls=PLS / page-load-strategy=PLS / page_load_strategy=PLS
         if low_key in ["pls", "page-load-strategy", "page_load_strategy"]:
             page_load_strategy = userdata[key].lower()

seleniumbase/core/browser_launcher.py

@@ -153,15 +153,25 @@ def get_uc_driver_version():
     return uc_driver_version
 
 
-def find_chromedriver_version_to_use(use_version):
-    # Because https://chromedriver.chromium.org/downloads stops at 114
-    final_chromedriver = "114"
+def find_chromedriver_version_to_use(use_version, driver_version):
+    # Note: https://chromedriver.chromium.org/downloads stops at 114.
+    # Future drivers are part of the Chrome-for-Testing collection.
     if (
-        use_version
-        and use_version.isdigit()
-        and int(use_version) > int(final_chromedriver)
+        driver_version
+        and str(driver_version).split(".")[0].isdigit()
+        and int(str(driver_version).split(".")[0]) >= 72
     ):
-        use_version = final_chromedriver
+        use_version = str(driver_version)
+    return use_version
+
+
+def find_edgedriver_version_to_use(use_version, driver_version):
+    if (
+        driver_version
+        and str(driver_version).split(".")[0].isdigit()
+        and int(str(driver_version).split(".")[0]) >= 80
+    ):
+        use_version = str(driver_version)
     return use_version
 
 
@@ -526,6 +536,7 @@ def _set_chrome_options(
     extension_zip,
     extension_dir,
     binary_location,
+    driver_version,
     page_load_strategy,
     use_wire,
     external_pdf,
@@ -1088,6 +1099,7 @@ def get_driver(
     extension_zip=None,
     extension_dir=None,
     binary_location=None,
+    driver_version=None,
     page_load_strategy=None,
     use_wire=False,
     external_pdf=False,
@@ -1295,6 +1307,7 @@ def get_driver(
             extension_zip,
             extension_dir,
             binary_location,
+            driver_version,
             page_load_strategy,
             use_wire,
             external_pdf,
@@ -1347,6 +1360,7 @@ def get_driver(
             extension_zip,
             extension_dir,
             binary_location,
+            driver_version,
             page_load_strategy,
             use_wire,
             external_pdf,
@@ -1403,6 +1417,7 @@ def get_remote_driver(
     extension_zip,
     extension_dir,
     binary_location,
+    driver_version,
     page_load_strategy,
     use_wire,
     external_pdf,
@@ -1522,6 +1537,7 @@ def get_remote_driver(
             extension_zip,
             extension_dir,
             binary_location,
+            driver_version,
             page_load_strategy,
             use_wire,
             external_pdf,
@@ -1712,6 +1728,7 @@ def get_remote_driver(
             extension_zip,
             extension_dir,
             binary_location,
+            driver_version,
             page_load_strategy,
             use_wire,
             external_pdf,
@@ -1834,6 +1851,7 @@ def get_remote_driver(
             extension_zip,
             extension_dir,
             binary_location,
+            driver_version,
             page_load_strategy,
             use_wire,
             external_pdf,
@@ -1956,6 +1974,7 @@ def get_local_driver(
     extension_zip,
     extension_dir,
     binary_location,
+    driver_version,
     page_load_strategy,
     use_wire,
     external_pdf,
@@ -2229,7 +2248,10 @@ def get_local_driver(
             major_edge_version = None
         if major_edge_version:
             use_version = major_edge_version
-        driver_version = None
+        use_version = find_edgedriver_version_to_use(
+            use_version, driver_version
+        )
+        edge_driver_version = None
         edgedriver_upgrade_needed = False
         if os.path.exists(LOCAL_EDGEDRIVER):
             try:
@@ -2254,16 +2276,16 @@ def get_local_driver(
                 else:
                     output = 0
                 if int(output) >= 2:
-                    driver_version = output
+                    edge_driver_version = output
             except Exception:
                 pass
         local_edgedriver_exists = False
         if LOCAL_EDGEDRIVER and os.path.exists(LOCAL_EDGEDRIVER):
             local_edgedriver_exists = True
             if (
                 use_version != "latest"
-                and driver_version
-                and use_version != driver_version
+                and edge_driver_version
+                and use_version != edge_driver_version
             ):
                 edgedriver_upgrade_needed = True
             else:
@@ -2761,6 +2783,7 @@ def get_local_driver(
                 extension_zip,
                 extension_dir,
                 binary_location,
+                driver_version,
                 page_load_strategy,
                 use_wire,
                 external_pdf,
@@ -2817,6 +2840,7 @@ def get_local_driver(
                 extension_zip,
                 extension_dir,
                 binary_location,
+                driver_version,
                 page_load_strategy,
                 use_wire,
                 external_pdf,
@@ -2860,7 +2884,7 @@ def get_local_driver(
                     major_chrome_version = None
             if major_chrome_version:
                 use_version = major_chrome_version
-            driver_version = None
+            ch_driver_version = None
             path_chromedriver = chromedriver_on_path()
             if os.path.exists(LOCAL_CHROMEDRIVER):
                 try:
@@ -2873,7 +2897,7 @@ def get_local_driver(
                         output = output.decode("utf-8")
                     output = output.split(" ")[1].split(".")[0]
                     if int(output) >= 2:
-                        driver_version = output
+                        ch_driver_version = output
                 except Exception:
                     pass
             elif path_chromedriver:
@@ -2887,7 +2911,7 @@ def get_local_driver(
                         output = output.decode("utf-8")
                     output = output.split(" ")[1].split(".")[0]
                     if int(output) >= 2:
-                        driver_version = output
+                        ch_driver_version = output
                 except Exception:
                     pass
             if headless2:
@@ -2922,12 +2946,14 @@ def get_local_driver(
                     from seleniumbase import config as sb_config
 
                     sb_config.multi_proxy = True
-            use_version = find_chromedriver_version_to_use(use_version)
+            use_version = find_chromedriver_version_to_use(
+                use_version, driver_version
+            )
             if (
                 LOCAL_CHROMEDRIVER
                 and os.path.exists(LOCAL_CHROMEDRIVER)
                 and (
-                    use_version == driver_version
+                    use_version == ch_driver_version
                     or use_version == "latest"
                 )
                 and (
@@ -2951,11 +2977,11 @@ def get_local_driver(
                 not path_chromedriver
                 or (
                     use_version != "latest"
-                    and driver_version
-                    and use_version != driver_version
+                    and ch_driver_version
+                    and use_version != ch_driver_version
                     and (
                         selenium4_or_newer
-                        or driver_version != "72"
+                        or ch_driver_version != "72"
                     )
                 )
                 or (
@@ -3000,9 +3026,9 @@ def get_local_driver(
                             if (
                                 not path_chromedriver
                                 or (
-                                    driver_version
+                                    ch_driver_version
                                     and (
-                                        int(driver_version)
+                                        int(ch_driver_version)
                                         < int(d_latest_major)
                                     )
                                 )
@@ -3056,9 +3082,9 @@ def get_local_driver(
                                     if (
                                         not path_chromedriver
                                         or (
-                                            driver_version
+                                            ch_driver_version
                                             and (
-                                                int(driver_version)
+                                                int(ch_driver_version)
                                                 < int(d_latest_major)
                                             )
                                         )
@@ -3077,7 +3103,9 @@ def get_local_driver(
                 )
                 with uc_lock:  # Avoid multithreaded issues
                     uc_driver_version = get_uc_driver_version()
-                    use_version = find_chromedriver_version_to_use(use_version)
+                    use_version = find_chromedriver_version_to_use(
+                        use_version, driver_version
+                    )
                     if (
                         (
                             uc_driver_version != use_version
@@ -3252,7 +3280,9 @@ def get_local_driver(
                             and int(major_chrome_version) >= 86
                         ):
                             mcv = major_chrome_version
-                            mcv = find_chromedriver_version_to_use(mcv)
+                            mcv = find_chromedriver_version_to_use(
+                                mcv, driver_version
+                            )
                     headless = True
                     headless_options = _set_chrome_options(
                         browser_name,
@@ -3294,6 +3324,7 @@ def get_local_driver(
                         extension_zip,
                         extension_dir,
                         binary_location,
+                        driver_version,
                         page_load_strategy,
                         use_wire,
                         external_pdf,

seleniumbase/fixtures/base_case.py

@@ -3628,6 +3628,7 @@ def get_new_driver(
         extension_zip=None,
         extension_dir=None,
         binary_location=None,
+        driver_version=None,
         page_load_strategy=None,
         use_wire=None,
         external_pdf=None,
@@ -3683,6 +3684,7 @@ def get_new_driver(
         extension_zip - A Chrome Extension ZIP file to use (Chrome-only)
         extension_dir - A Chrome Extension folder to use (Chrome-only)
         binary_location - the path of the browser binary to use (Chromium)
+        driver_version - the chromedriver or uc_driver version to force
         page_load_strategy - the option to change pageLoadStrategy (Chrome)
         use_wire - Use selenium-wire webdriver instead of the selenium one
         external_pdf - "plugins.always_open_pdf_externally": True. (Chrome)
@@ -3807,6 +3809,8 @@ def get_new_driver(
             extension_dir = self.extension_dir
         if binary_location is None:
             binary_location = self.binary_location
+        if driver_version is None:
+            driver_version = self.driver_version
         if page_load_strategy is None:
             page_load_strategy = self.page_load_strategy
         if use_wire is None:
@@ -3886,6 +3890,7 @@ def get_new_driver(
             extension_zip=extension_zip,
             extension_dir=extension_dir,
             binary_location=binary_location,
+            driver_version=driver_version,
             page_load_strategy=page_load_strategy,
             use_wire=use_wire,
             external_pdf=external_pdf,
@@ -13882,6 +13887,7 @@ def setUp(self, masterqa_mode=False):
             self.extension_zip = sb_config.extension_zip
             self.extension_dir = sb_config.extension_dir
             self.binary_location = sb_config.binary_location
+            self.driver_version = sb_config.driver_version
             self.page_load_strategy = sb_config.page_load_strategy
             self.use_wire = sb_config.use_wire
             self.external_pdf = sb_config.external_pdf
@@ -14202,6 +14208,7 @@ def setUp(self, masterqa_mode=False):
                 extension_zip=self.extension_zip,
                 extension_dir=self.extension_dir,
                 binary_location=self.binary_location,
+                driver_version=self.driver_version,
                 page_load_strategy=self.page_load_strategy,
                 use_wire=self.use_wire,
                 external_pdf=self.external_pdf,

seleniumbase/plugins/driver_manager.py

@@ -96,6 +96,7 @@ def Driver(
     extension_zip=None,  # Load a Chrome Extension .zip|.crx, comma-separated.)
     extension_dir=None,  # Load a Chrome Extension directory, comma-separated.)
     binary_location=None,  # Set path of the Chromium browser binary to use.
+    driver_version=None,  # Set the chromedriver or uc_driver version to use.
     page_load_strategy=None,  # Set Chrome PLS to "normal", "eager", or "none".
     use_wire=None,  # Use selenium-wire's webdriver over selenium webdriver.
     external_pdf=None,  # Set Chrome "plugins.always_open_pdf_externally":True.
@@ -395,6 +396,16 @@ def Driver(
             ad_block_on = True
         else:
             ad_block_on = False
+    if driver_version is None:
+        arg_join = " ".join(sys_argv)
+        if "--driver-version=" in arg_join:
+            driver_version = (
+                arg_join.split("--driver-version=")[1].split(" ")[0]
+            )
+        elif "--driver_version=" in arg_join:
+            driver_version = (
+                arg_join.split("--driver_version=")[1].split(" ")[0]
+            )
     browser_name = browser
 
     # Launch a web browser
@@ -444,6 +455,7 @@ def Driver(
         extension_zip=extension_zip,
         extension_dir=extension_dir,
         binary_location=binary_location,
+        driver_version=driver_version,
         page_load_strategy=page_load_strategy,
         use_wire=use_wire,
         external_pdf=external_pdf,