def get_driver(self):
# Start a new browser and return the WebDriver
browser_name = self.config.get('selenium', 'browser')
if browser_name == 'firefox':
from selenium.webdriver import Firefox
browser_binary = FirefoxBinary()
driver = Firefox(firefox_binary=browser_binary)
driver._is_remote = False # Workaround for http://stackoverflow.com/a/42770761/489916
return driver
if browser_name == 'chrome':
from selenium.webdriver import Chrome
return Chrome()
if browser_name == 'phantomjs':
from selenium.webdriver import PhantomJS
return PhantomJS()
# @TODO: Add chrome
raise RuntimeError('Unsupported/unknown browser')
