diff --git a/e2e/test_login.py b/e2e/test_login.py index d0e14519..d49d8656 100644 --- a/e2e/test_login.py +++ b/e2e/test_login.py @@ -24,11 +24,11 @@ class TestLogin: assert button.get_attribute('type') == 'submit' assert 'Log in' in button.text - forgot_password_link = selenium.find_element_by_class_name( - 'password-forgotten' - ) - assert forgot_password_link.tag_name == 'a' - assert 'Forgot password?' in forgot_password_link.text + links = selenium.find_elements_by_class_name('links') + assert links[0].tag_name == 'a' + assert 'Register' in links[0].text + assert links[1].tag_name == 'a' + assert 'Forgot password?' in links[1].text def test_user_can_log_in(self, selenium): user = { diff --git a/e2e/test_registration.py b/e2e/test_registration.py index ca756f47..9db4ff86 100644 --- a/e2e/test_registration.py +++ b/e2e/test_registration.py @@ -29,6 +29,10 @@ class TestRegistration: assert button.get_attribute('type') == 'submit' assert 'Register' in button.text + link = selenium.find_element_by_class_name('links') + assert link.tag_name == 'a' + assert 'Login' in link.text + def test_user_can_register(self, selenium): user = register_valid_user(selenium)