blob: 6da20ac5fa452dfcd86d22c3eee7299b156b1801 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
from django.contrib import auth
from django.contrib.auth.models import User
from django.http import HttpRequest
from django.test import TestCase
class TestLogin(TestCase):
@classmethod
def setUpTestData(cls):
cls.user = User.objects.create_user(username="testuser", password="password")
def setUp(self):
self.request = HttpRequest()
self.request.session = self.client.session
def test_user_login(self):
auth.login(self.request, self.user)
self.assertEqual(self.request.session[auth.SESSION_KEY], str(self.user.pk))
def test_inactive_user(self):
self.user.is_active = False
self.user.save(update_fields=["is_active"])
auth.login(self.request, self.user)
self.assertEqual(self.request.session[auth.SESSION_KEY], str(self.user.pk))
def test_without_user(self):
with self.assertRaisesMessage(
AttributeError,
"'NoneType' object has no attribute 'get_session_auth_hash'",
):
auth.login(self.request, None)
|