Skip to content

Commit 514b5a6

Browse files
archy-rock3t-cloudtomchristie
authored andcommitted
Add couple of test for AuthToken (encode#4816)
* Add couple of test for AuthToken * PEP8 cleanup
1 parent 5efbe17 commit 514b5a6

File tree

1 file changed

+29
-0
lines changed

1 file changed

+29
-0
lines changed

tests/test_authtoken.py

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
1+
import pytest
2+
from django.contrib.admin import site
3+
from django.contrib.auth.models import User
4+
from django.test import TestCase
5+
6+
from rest_framework.authtoken.admin import TokenAdmin
7+
from rest_framework.authtoken.models import Token
8+
from rest_framework.authtoken.serializers import AuthTokenSerializer
9+
from rest_framework.exceptions import ValidationError
10+
11+
12+
class AuthTokenTests(TestCase):
13+
14+
def setUp(self):
15+
self.site = site
16+
self.user = User.objects.create_user(username='test_user')
17+
self.token = Token.objects.create(key='test token', user=self.user)
18+
19+
def test_model_admin_displayed_fields(self):
20+
mock_request = object()
21+
token_admin = TokenAdmin(self.token, self.site)
22+
assert token_admin.get_fields(mock_request) == ('user',)
23+
24+
def test_token_string_representation(self):
25+
assert str(self.token) == 'test token'
26+
27+
def test_validate_raise_error_if_no_credentials_provided(self):
28+
with pytest.raises(ValidationError):
29+
AuthTokenSerializer().validate({})

0 commit comments

Comments
 (0)