Apply suggestions from code review

Co-authored-by: Christopher Cave-Ayland <[email protected]>
ImperialCollegeLondon · Sep 19, 2024 · a2382e4 · a2382e4
1 parent 18e0f93
commit a2382e4
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/test_views.py b/tests/test_views.py
@@ -18,9 +18,7 @@ def test_login_redirect(self, client):
         response = client.get(self.endpoint)
         assert response.status_code == HTTPStatus.FOUND
 
-        # AssertRedirects try to match the exact URL, but the URL may contain a query
-        # string. So, we just check the start of the URL.
-        assert response.url.startswith(reverse("main:login"))
+        assertRedirects(response, reverse("main:login") + f"?next={self.endpoint}")
 
 
 class ProcessActionsTest(LoginRequiredTest):