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