setup-selenium-testing 0.1.2__tar.gz → 0.1.3__tar.gz

{setup_selenium_testing-0.1.2 → setup_selenium_testing-0.1.3}/PKG-INFO

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

{setup_selenium_testing-0.1.2 → setup_selenium_testing-0.1.3}/pyproject.toml

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

{setup_selenium_testing-0.1.2 → setup_selenium_testing-0.1.3}/setup_selenium/selenium_module.py

@@ -1,11 +1,12 @@
 """Setup selenium for testing"""
+
 from __future__ import annotations
 
 import errno
 import logging
 import os as os
 from enum import Enum
-from typing import TYPE_CHECKING, Union
+from typing import TYPE_CHECKING, Optional, Union
 
 from selenium import webdriver
 from selenium.common.exceptions import NoSuchWindowException, WebDriverException
@@ -23,6 +24,9 @@ if TYPE_CHECKING:
     from selenium.webdriver.common.options import ArgOptions
 
     T_WebDriver: TypeAlias = Union[Firefox, Chrome, Edge]
+    T_DrvOpts: TypeAlias = Union[
+        webdriver.FirefoxOptions, webdriver.ChromeOptions, webdriver.EdgeOptions
+    ]
 
 __all__ = ["SetupSelenium"]
 
@@ -202,20 +206,24 @@ class SetupSelenium:
         log_dir: str = "./logs",
         binary: str | None = None,
         driver_path: str | None = None,
+        options: Optional[T_DrvOpts] = None,
     ) -> T_WebDriver:
         """Instantiates the browser driver"""
         browser = browser.lower()
         driver: T_WebDriver
         if browser == Browser.FIREFOX:
+            assert options is None or isinstance(options, webdriver.FirefoxOptions)
             driver = SetupSelenium.firefox(
                 headless=headless,
                 enable_log_driver=enable_log_driver,
                 log_dir=log_dir,
                 binary=binary,
                 driver_path=driver_path,
+                options=options,
             )
 
         elif browser == Browser.CHROME:
+            assert options is None or isinstance(options, webdriver.ChromeOptions)
             driver = SetupSelenium.chrome(
                 headless=headless,
                 enable_log_performance=enable_log_performance,
@@ -224,9 +232,11 @@ class SetupSelenium:
                 log_dir=log_dir,
                 binary=binary,
                 driver_path=driver_path,
+                options=options,
             )
 
         elif browser == Browser.EDGE:
+            assert options is None or isinstance(options, webdriver.EdgeOptions)
             driver = SetupSelenium.edge(
                 headless=headless,
                 enable_log_performance=enable_log_performance,
@@ -235,6 +245,7 @@ class SetupSelenium:
                 log_dir=log_dir,
                 binary=binary,
                 driver_path=driver_path,
+                options=options,
             )
 
         else:
@@ -263,7 +274,7 @@ class SetupSelenium:
         log_dir: str = "./logs",
         driver_path: str | None = None,
         binary: str | None = None,
-        options: webdriver.FirefoxOptions = None,
+        options: webdriver.FirefoxOptions | None = None,
     ) -> webdriver.Firefox:
         """Instantiates firefox geockodriver"""
         options = options or SetupSelenium.firefox_options()
@@ -336,7 +347,7 @@ class SetupSelenium:
         log_dir: str = "./logs",
         driver_path: str | None = None,
         binary: str | None = None,
-        options: webdriver.ChromeOptions = None,
+        options: webdriver.ChromeOptions | None = None,
     ) -> webdriver.Chrome:
         """Instantiates chromedriver"""
         options = options or SetupSelenium.chrome_options()
@@ -379,12 +390,12 @@ class SetupSelenium:
             service = ChromeService(
                 executable_path=driver_path,
                 service_args=args,
-                log_output=logpath,
+                log_output=logpath,  # type: ignore[arg-type]
             )
         else:
             service = ChromeService(
                 service_args=args,
-                log_output=logpath,
+                log_output=logpath,  # type: ignore[arg-type]
             )
 
         driver = webdriver.Chrome(service=service, options=options)
@@ -469,7 +480,7 @@ class SetupSelenium:
         log_dir: str = "./logs",
         driver_path: str | None = None,
         binary: str | None = None,
-        options: webdriver.EdgeOptions = None,
+        options: webdriver.EdgeOptions | None = None,
     ) -> webdriver.Edge:
         """Instantiates edgedriver"""
         options = options or SetupSelenium.edge_options()
@@ -513,12 +524,12 @@ class SetupSelenium:
             service = EdgeService(
                 executable_path=driver_path,
                 service_args=args,
-                log_output=logpath,
+                log_output=logpath,  # type: ignore[arg-type]
             )
         else:
             service = EdgeService(
                 service_args=args,
-                log_output=logpath,
+                log_output=logpath,  # type: ignore[arg-type]
             )
         driver = webdriver.Edge(service=service, options=options)