Skip to content

Commit 83ce800

Browse files
committed
Allow running e2e tests with Chromium
1 parent 27234b2 commit 83ce800

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

two_factor/plugins/webauthn/tests/test_e2e.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222

2323
try:
2424
from webdriver_manager.chrome import ChromeDriverManager
25+
from webdriver_manager.utils import ChromeType
2526
except ImportError:
2627
ChromeDriverManager = None
2728

@@ -39,7 +40,11 @@ def setUp(self):
3940

4041
options = webdriver.ChromeOptions()
4142
options.add_argument('headless')
42-
self.webdriver = webdriver.Chrome(ChromeDriverManager().install(), options=options)
43+
try:
44+
driver = ChromeDriverManager()
45+
except ValueError:
46+
driver = ChromeDriverManager(chrome_type=ChromeType.CHROMIUM)
47+
self.webdriver = webdriver.Chrome(driver.install(), options=options)
4348

4449
super().setUp()
4550

0 commit comments

Comments
 (0)