class AdminFunctionalTest(FunctionalTest):
"""
Base class for a functional test case for admin pages.
"""
url = '/admin/' # override in derived classes
username = '******'
password = '******'
def _create_superuser(self):
"""
Helper method to create a superuser for test cases.
"""
user_model = get_user_model()
return user_model.objects.create_superuser(
email=self.username,
password=self.password
)
def setUp(self):
self._create_superuser()
self.open(self.url)
self.login()
def login(self):
self.page = LoginPage(self.driver)
self.page.login(self.username, self.password)
def open(self, url, retries=3):
while True:
try:
self.driver.get(self.live_server_url + url)
break
except socket.timeout:
self.driver.quit()
self.driver = get_web_driver(self.__class__.__name__)
self.driver.implicitly_wait(10)
self.driver.maximize_window()
self.login()
retries -= 1
if retries <= 0:
raise
class AdminFunctionalTest(FunctionalTest):
"""
Base class for a functional test case for admin pages.
"""
url = '/admin/' # override in derived classes
username = '******'
password = '******'
def _create_superuser(self):
"""
Helper method to create a superuser for test cases.
"""
user_model = get_user_model()
return user_model.objects.create_superuser(email=self.username,
password=self.password)
def setUp(self):
self._create_superuser()
self.driver.get(self.live_server_url + self.url)
self.page = LoginPage(self.driver)
self.page.login(self.username, self.password)
def login(self):
self.page = LoginPage(self.driver)
self.page.login(self.username, self.password)
def setUp(self):
self._create_superuser()
self.driver.get(self.live_server_url + self.url)
self.page = LoginPage(self.driver)
self.page.login(self.username, self.password)
def e_fill_in_forbidden_user(self):
print("Login attempt with locked out user")
page = LoginPage(self.driver)
page.login(username="******", password="******")
def e_fill_in_invalid_credentials(self):
print("Login attempt with invalid credentials")
page = LoginPage(self.driver)
page.login(username="******", password="******")
def e_fill_in_empty_username(self):
print("Login attempt with empty username")
page = LoginPage(self.driver)
page.login(username="", password="******")
def e_fill_in_empty_password(self):
print("Login attempt with empty password")
page = LoginPage(self.driver)
page.login(username="******", password="")
def v_login_page(self):
page = LoginPage(self.driver)
self.assertTrue(page.is_login_button_present,
"Login button should be present.")
def e_load_login_page(self):
print("Load the login page of site from: {}".format(BASE_URL))
page = LoginPage(self.driver, BASE_URL)
page.open()
self.driver.set_page_load_timeout(20)
def e_fill_in_standard_user(self):
print("Login as standard user")
page = LoginPage(self.driver)
page.login(username="******", password="******")