From cd9045f28de6fe3e9e77e05063c503314640f323 Mon Sep 17 00:00:00 2001 From: Elena ``of Valhalla'' Grandi Date: Fri, 5 Jan 2024 11:10:31 +0100 Subject: [PATCH] Fallback to Chrom(e|ium) if the Firefox driver for selenium isn't available --- src/bepasty/tests/test_website.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/src/bepasty/tests/test_website.py b/src/bepasty/tests/test_website.py index 49508e07..4e8af1cd 100644 --- a/src/bepasty/tests/test_website.py +++ b/src/bepasty/tests/test_website.py @@ -1,6 +1,10 @@ from selenium.webdriver import Firefox +from selenium.webdriver.chrome.service import Service as ChromeService +from selenium.webdriver import Chrome from selenium.webdriver.common.keys import Keys -from selenium.common.exceptions import NoSuchElementException +from selenium.common.exceptions import ( + NoSuchDriverException, NoSuchElementException +) import pytest @@ -17,7 +21,11 @@ def setup_class(self): """ Setup: Open a mozilla browser, login """ - self.browser = Firefox() + try: + self.browser = Firefox() + except NoSuchDriverException: + service = ChromeService(executable_path="/usr/bin/chromedriver") + self.browser = Chrome(service=service) self.browser.get('http://localhost:5000/') token = self.browser.find_element_by_name("token") password = "foo"