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')