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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
from typing import Optional
import django.contrib.auth as django_auth
import pytest
import strawberry
from django.conf import settings
from django.core.exceptions import ObjectDoesNotExist
import strawberry_django
from strawberry_django import auth
from tests import utils
UserModel = django_auth.get_user_model()
@strawberry_django.type(UserModel)
class User:
username: strawberry.auto
email: strawberry.auto
@strawberry_django.input(UserModel)
class UserInput:
username: strawberry.auto
password: strawberry.auto
email: strawberry.auto
@strawberry.type
class Mutation:
login: Optional[User] = auth.login() # type: ignore
logout = auth.logout()
register: User = auth.register(UserInput)
@pytest.fixture
def mutation(db):
return utils.generate_query(mutation=Mutation)
def test_login(mutation, user, context):
result = mutation(
'{ login(username: "user", password: "password") { username } }',
context_value=context,
)
assert not result.errors
assert result.data["login"] == {"username": "user"}
assert context.request.user == user
def test_login_with_wrong_password(mutation, user, context):
result = mutation(
'{ login(username: "user", password: "wrong") { username } }',
context_value=context,
)
assert result.errors
assert result.data["login"] is None
assert context.request.user.is_anonymous
def test_logout(mutation, user, context):
django_auth.login(
context.request,
user,
backend=settings.AUTHENTICATION_BACKENDS[0],
)
result = mutation("{ logout }", context_value=context)
assert not result.errors
assert result.data["logout"] is True
assert context.request.user.is_anonymous
def test_logout_without_logged_in(mutation, user, context):
result = mutation("{ logout }", context_value=context)
assert not result.errors
assert result.data["logout"] is False
def test_register_new_user(mutation, user, context):
result = mutation(
'{ register(data: {username: "new_user",'
' password: "test_password"}) { username } }',
context_value=context,
)
assert not result.errors
assert result.data["register"] == {"username": "new_user"}
user = UserModel.objects.get(username="new_user")
assert user.pk
assert user.check_password("test_password")
def test_register_with_invalid_password(mutation, user, context):
result = mutation(
'{ register(data: {username: "invalid_user", password: "a"}) { username } }',
context_value=context,
)
assert len(result.errors) == 1
assert "too short" in result.errors[0].message
with pytest.raises(ObjectDoesNotExist):
assert UserModel.objects.get(username="invalid_user")
|
