diff --git a/src/tests/base/__init__.py b/src/tests/base/__init__.py index dfc720572f..7f3b6f7b46 100644 --- a/src/tests/base/__init__.py +++ b/src/tests/base/__init__.py @@ -18,7 +18,13 @@ class BrowserTest(StaticLiveServerTestCase): settings.DEBUG = ('--debug' in sys.argv) def setUp(self): - self.driver = getattr(webdriver, BROWSER)() + if hasattr(webdriver, BROWSER): + self.driver = getattr(webdriver, BROWSER)() + else: + self.driver = webdriver.Remote( + desired_capabilities=webdriver.DesiredCapabilities.CHROME, + command_executor=BROWSER + ) self.driver.set_window_size(1920, 1080) self.driver.implicitly_wait(10)