From dee572c40f92f424844ec956bed2c33df15b89e2 Mon Sep 17 00:00:00 2001
From: Sam <sam@samr1.net>
Date: Sat, 6 Nov 2021 23:16:19 +0100
Subject: [PATCH] fix e2e tests

---
 e2e/test_login.py        | 10 +++++-----
 e2e/test_registration.py |  4 ++++
 2 files changed, 9 insertions(+), 5 deletions(-)

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)