diff --git a/conftest.py b/conftest.py index 2e9e30f..76cbd41 100644 --- a/conftest.py +++ b/conftest.py @@ -16,6 +16,9 @@ from webdriver_manager.chrome import ChromeDriverManager from webdriver_manager.core.utils import ChromeType +from selenium.webdriver.edge.service import Service as EdgeService +from webdriver_manager.microsoft import EdgeChromiumDriverManager + from main import Constants @@ -55,6 +58,8 @@ def web_driver(config) -> WebDriver: driver = webdriver.Chrome(service=ChromeService(ChromeDriverManager().install())) elif config['browser'] == 'chromium': driver = webdriver.Chrome(service=ChromiumService(ChromeDriverManager(chrome_type=ChromeType.CHROMIUM).install())) + elif config['browser'] == 'edge': + driver = webdriver.Edge(service=EdgeService(EdgeChromiumDriverManager().install())) else: raise Exception(f'"{config["browser"]}" is not a supported browser') driver.maximize_window() diff --git a/tests/config.json b/tests/config.json index a524ce5..6b92b4c 100644 --- a/tests/config.json +++ b/tests/config.json @@ -1,5 +1,5 @@ { - "browser": "chrome", + "browser": "edge", "selenium_grid": "false", "selenium_grid_url": "testurl.com", "wait_time": 10