mirror of
https://github.com/django/django.git
synced 2024-12-23 01:25:58 +00:00
d907371ef9
It's a transitional setting helpful in migrating multiple instance of the same project to Django 3.1+. Thanks Markus Holtermann for the report and review, Florian Apolloner for the implementation idea and review, and Carlton Gibson for the review.
62 lines
2.7 KiB
Python
62 lines
2.7 KiB
Python
from django.contrib.auth import HASH_SESSION_KEY
|
|
from django.contrib.auth.middleware import AuthenticationMiddleware
|
|
from django.contrib.auth.models import User
|
|
from django.http import HttpRequest, HttpResponse
|
|
from django.test import TestCase, override_settings
|
|
from django.test.utils import ignore_warnings
|
|
from django.utils.deprecation import RemovedInDjango40Warning
|
|
|
|
|
|
class TestAuthenticationMiddleware(TestCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
cls.user = User.objects.create_user('test_user', 'test@example.com', 'test_password')
|
|
|
|
def setUp(self):
|
|
self.middleware = AuthenticationMiddleware(lambda req: HttpResponse())
|
|
self.client.force_login(self.user)
|
|
self.request = HttpRequest()
|
|
self.request.session = self.client.session
|
|
|
|
def test_no_password_change_doesnt_invalidate_session(self):
|
|
self.request.session = self.client.session
|
|
self.middleware(self.request)
|
|
self.assertIsNotNone(self.request.user)
|
|
self.assertFalse(self.request.user.is_anonymous)
|
|
|
|
def test_no_password_change_does_not_invalidate_legacy_session(self):
|
|
# RemovedInDjango40Warning: pre-Django 3.1 hashes will be invalid.
|
|
session = self.client.session
|
|
session[HASH_SESSION_KEY] = self.user._legacy_get_session_auth_hash()
|
|
session.save()
|
|
self.request.session = session
|
|
self.middleware(self.request)
|
|
self.assertIsNotNone(self.request.user)
|
|
self.assertFalse(self.request.user.is_anonymous)
|
|
|
|
@ignore_warnings(category=RemovedInDjango40Warning)
|
|
def test_session_default_hashing_algorithm(self):
|
|
hash_session = self.client.session[HASH_SESSION_KEY]
|
|
with override_settings(DEFAULT_HASHING_ALGORITHM='sha1'):
|
|
self.assertNotEqual(hash_session, self.user.get_session_auth_hash())
|
|
|
|
def test_changed_password_invalidates_session(self):
|
|
# After password change, user should be anonymous
|
|
self.user.set_password('new_password')
|
|
self.user.save()
|
|
self.middleware(self.request)
|
|
self.assertIsNotNone(self.request.user)
|
|
self.assertTrue(self.request.user.is_anonymous)
|
|
# session should be flushed
|
|
self.assertIsNone(self.request.session.session_key)
|
|
|
|
def test_no_session(self):
|
|
msg = (
|
|
"The Django authentication middleware requires session middleware "
|
|
"to be installed. Edit your MIDDLEWARE setting to insert "
|
|
"'django.contrib.sessions.middleware.SessionMiddleware' before "
|
|
"'django.contrib.auth.middleware.AuthenticationMiddleware'."
|
|
)
|
|
with self.assertRaisesMessage(AssertionError, msg):
|
|
self.middleware(HttpRequest())
|