summaryrefslogtreecommitdiff
path: root/tests/auth_tests/test_middleware.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/auth_tests/test_middleware.py')
-rw-r--r--tests/auth_tests/test_middleware.py18
1 files changed, 18 insertions, 0 deletions
diff --git a/tests/auth_tests/test_middleware.py b/tests/auth_tests/test_middleware.py
index e7e5d3b4dc..5e106d40f7 100644
--- a/tests/auth_tests/test_middleware.py
+++ b/tests/auth_tests/test_middleware.py
@@ -206,3 +206,21 @@ class TestLoginRequiredMiddleware(TestCase):
def test_get_redirect_field_name_default(self):
redirect_field_name = self.middleware.get_redirect_field_name(lambda: None)
self.assertEqual(redirect_field_name, REDIRECT_FIELD_NAME)
+
+ def test_public_view_logged_in_performance(self):
+ """
+ Public views don't trigger fetching the user from the database.
+ """
+ self.client.force_login(self.user)
+ with self.assertNumQueries(0):
+ response = self.client.get("/public_view/")
+ self.assertEqual(response.status_code, 200)
+
+ def test_protected_view_logged_in_performance(self):
+ """
+ Protected views do trigger fetching the user from the database.
+ """
+ self.client.force_login(self.user)
+ with self.assertNumQueries(2): # session and user
+ response = self.client.get("/protected_view/")
+ self.assertEqual(response.status_code, 200)