From 8cdd7a89ad6115b80ae57ed6cbb0d41abce09816 Mon Sep 17 00:00:00 2001 From: Raphael Michel Date: Sun, 12 Apr 2015 19:29:33 +0200 Subject: [PATCH] Improve handling of remote test drivers --- src/tests/base/__init__.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) 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)