75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
# Copyright The IETF Trust 2014-2021, All Rights Reserved
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from django.urls import reverse as urlreverse
|
|
from unittest import skipIf
|
|
|
|
skip_selenium = False
|
|
skip_message = ""
|
|
try:
|
|
from selenium import webdriver
|
|
except ImportError as e:
|
|
skip_selenium = True
|
|
skip_message = "Skipping selenium tests: %s" % e
|
|
|
|
|
|
from ietf.utils.pipe import pipe
|
|
from ietf.utils.test_runner import IetfLiveServerTestCase
|
|
from ietf import settings
|
|
|
|
|
|
executable_name = 'chromedriver'
|
|
code, out, err = pipe('{} --version'.format(executable_name))
|
|
if code != 0:
|
|
skip_selenium = True
|
|
skip_message = "Skipping selenium tests: '{}' executable not found.".format(executable_name)
|
|
if skip_selenium:
|
|
print(" "+skip_message)
|
|
|
|
def start_web_driver():
|
|
options = webdriver.ChromeOptions()
|
|
options.add_argument("headless")
|
|
options.add_argument("disable-extensions")
|
|
options.add_argument("disable-gpu") # headless needs this
|
|
options.add_argument("no-sandbox") # docker needs this
|
|
return webdriver.Chrome(options=options, service_log_path=settings.TEST_GHOSTDRIVER_LOG_PATH)
|
|
|
|
|
|
def selenium_enabled():
|
|
"""Are Selenium tests enabled?"""
|
|
return not skip_selenium
|
|
|
|
|
|
def ifSeleniumEnabled(func):
|
|
"""Only run test if Selenium testing is enabled"""
|
|
return skipIf(skip_selenium, skip_message)(func)
|
|
|
|
|
|
class IetfSeleniumTestCase(IetfLiveServerTestCase):
|
|
login_view = 'ietf.ietfauth.views.login'
|
|
|
|
def setUp(self):
|
|
super(IetfSeleniumTestCase, self).setUp()
|
|
self.driver = start_web_driver()
|
|
self.driver.set_window_size(1024,768)
|
|
|
|
def tearDown(self):
|
|
super(IetfSeleniumTestCase, self).tearDown()
|
|
self.driver.close()
|
|
|
|
def absreverse(self,*args,**kwargs):
|
|
return '%s%s'%(self.live_server_url, urlreverse(*args, **kwargs))
|
|
|
|
def debug_snapshot(self,filename='debug_this.png'):
|
|
self.driver.execute_script("document.body.bgColor = 'white';")
|
|
self.driver.save_screenshot(filename)
|
|
|
|
def login(self, username='plain'):
|
|
url = self.absreverse(self.login_view)
|
|
password = '%s+password' % username
|
|
self.driver.get(url)
|
|
self.driver.find_element_by_name('username').send_keys(username)
|
|
self.driver.find_element_by_name('password').send_keys(password)
|
|
self.driver.find_element_by_xpath('//button[@type="submit"]').click()
|
|
|