diff --git a/tests/LandingPage/test_views.py b/tests/LandingPage/test_views.py index 998dbdc..d2322cd 100644 --- a/tests/LandingPage/test_views.py +++ b/tests/LandingPage/test_views.py @@ -40,6 +40,9 @@ class TestLogin(TestCase): self.assertEqual(user.email,'johnsmith@gmail.com') self.assertEqual(user.display_name, 'Mr. Smith') + # Check that the user has been logged in + self.assertEqual(client.get('/').context['user'], user) + def test_reject_bad_state(self): with responses.RequestsMock() as rm: client = Client() @@ -73,6 +76,9 @@ class TestLogin(TestCase): self.assertEqual(user.email,'johnsmith@gmail.com') self.assertEqual(user.display_name, 'Mr. Smith') + # Check that the user has been logged in + self.assertEqual(client.get('/').context['user'], user) + def test_states_unique(self): with responses.RequestsMock() as rm: client1 = Client()