setup-selenium-testing 0.1.4__tar.gz → 0.1.6__tar.gz

{setup_selenium_testing-0.1.4 → setup_selenium_testing-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: setup-selenium-testing
-Version: 0.1.4
+Version: 0.1.6
 Summary: Setup Selenium for automation testing
 Home-page: https://github.com/bandophahita/setup_selenium
 License: MIT

{setup_selenium_testing-0.1.4 → setup_selenium_testing-0.1.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "setup-selenium-testing"
-version = "0.1.4"
+version = "0.1.6"
 description = "Setup Selenium for automation testing"
 authors = ["Marcel Wilson <trenchrats@gmail.com>"]
 license = "MIT"

{setup_selenium_testing-0.1.4 → setup_selenium_testing-0.1.6}/setup_selenium/setup_selenium.py

@@ -8,7 +8,7 @@ import os as os
 from enum import Enum
 from typing import TYPE_CHECKING, Optional, Union
 
-from selenium import webdriver
+from selenium import __version__, webdriver
 from selenium.common.exceptions import NoSuchWindowException, WebDriverException
 from selenium.webdriver.chrome.service import Service as ChromeService
 from selenium.webdriver.common.selenium_manager import SeleniumManager
@@ -28,6 +28,11 @@ if TYPE_CHECKING:
         webdriver.FirefoxOptions, webdriver.ChromeOptions, webdriver.EdgeOptions
     ]
 
+NEW_SELENIUM = False
+if Version(__version__) >= Version("4.20.0"):
+    NEW_SELENIUM = True
+
+
 __all__ = ["SetupSelenium"]
 
 
@@ -171,7 +176,11 @@ class SetupSelenium:
         driver_version = driver_version or None
 
         sm = SeleniumManager()
-        args = [f"{sm.get_binary()}", "--browser", browser]
+
+        if NEW_SELENIUM:
+            args = [f"{sm._get_binary()}", "--browser", browser]
+        else:
+            args = [f"{sm.get_binary()}", "--browser", browser]  # type: ignore[attr-defined]
 
         if browser_version:
             args.append("--browser-version")
@@ -187,7 +196,12 @@ class SetupSelenium:
             args.append("--browser-path")
             args.append(browser_path)
 
-        output = sm.run(args)
+        if NEW_SELENIUM:
+            args.append("--output")
+            args.append("json")
+            output = sm._run(args)
+        else:
+            output = sm.run(args)  # type: ignore[attr-defined]
         driver_path = output["driver_path"]
         browser_path = output["browser_path"]