def get_selenium(**kwargs): driver = getattr(settings, 'TEST_SELENIUM_DRIVER', 'firefox') if driver in ('chrome', 'chrome_headless'): from selenium.webdriver.chrome.options import Options from selenium.webdriver.chrome.webdriver import WebDriver as ChromeDriver options = Options() if driver == 'chrome_headless': options.add_argument('headless') options.add_argument('disable-gpu') for key, val in kwargs.items(): if val is not None: options.add_argument('{key}={val}'.format(key=key, val=val)) else: options.add_argument('{key}'.format(key=key)) driver_path = os.environ.get('CHROME_DRIVER_PATH', None) if driver_path is not None: return ChromeDriver(driver_path, options=options) return ChromeDriver(options=options) elif driver == 'phantomjs': from selenium.webdriver import PhantomJS return PhantomJS() else: from selenium.webdriver.firefox.webdriver import WebDriver as FirefoxDriver return FirefoxDriver()
def _create_driver(self, local, grid_url): self._update_capabilities_with_options() if not local: self._driver = RemoteDriver( command_executor=grid_url, desired_capabilities=self._capabilities) else: self._driver = FirefoxDriver( desired_capabilities=self._capabilities)
def get_selenium_driver(driver_name): if driver_name == 'chrome': from selenium.webdriver.chrome.webdriver import WebDriver as ChromeDriver return ChromeDriver() elif driver_name == 'phantomjs': from selenium.webdriver import PhantomJS return PhantomJS() elif driver_name == 'firefox': from selenium.webdriver.firefox.webdriver import WebDriver as FirefoxDriver return FirefoxDriver() raise ValueError('Unkown driver name')
def get_selenium(): driver = getattr(settings, 'TEST_SELENIUM_DRIVER', 'firefox') if driver == 'chrome': from selenium.webdriver.chrome.webdriver import WebDriver as ChromeDriver return ChromeDriver() elif driver == 'phantomjs': from selenium.webdriver import PhantomJS return PhantomJS() else: from selenium.webdriver.firefox.webdriver import WebDriver as FirefoxDriver return FirefoxDriver()
def setUp(self): self.mymetadata = TestMetadata() self.mymetadata.view_name = 'template_test' self.mymetadata.title = TitleTag(tag_value='Title Test') self.mymetadata.author = MetaTag(meta_name='author', meta_content='Test Author') self.mymetadata.description = MetaTag(meta_name='description', meta_content='Test Description') self.mymetadata.keywords = KeywordsTag(meta_name='keywords', meta_content='test, author') # Attr name and class attribute name are not equals, let's do it by setattr setattr(self.mymetadata, 'og:title', MetaTag(meta_name='og:title')) setattr(self.mymetadata, 'twitter:title', MetaTag(meta_name='twitter:title')) setattr(self.mymetadata, 'og:description', MetaTag(meta_name='og:description')) setattr(self.mymetadata, 'twitter:description', MetaTag(meta_name='twitter:description')) setattr( self.mymetadata, 'og:url', MetaTag(meta_name='og:url', meta_content='http://infantium.com')) setattr(self.mymetadata, 'twitter:url', MetaTag(meta_name='twitter:url')) setattr(self.mymetadata, 'og:type', MetaTag(meta_name='og:type', meta_content='type')) setattr(self.mymetadata, 'twitter:card', MetaTag(meta_name='twitter:card', meta_content='card')) setattr( self.mymetadata, 'og:image', ImageMetaTag(meta_name='og:image', meta_content='img/partners/animation2/base.png')) setattr(self.mymetadata, 'twitter:image', ImageMetaTag(meta_name='twitter:image')) self.mymetadata.save() self.firefox = FirefoxDriver()