File tree Expand file tree Collapse file tree 1 file changed +29
-0
lines changed Expand file tree Collapse file tree 1 file changed +29
-0
lines changed Original file line number Diff line number Diff line change
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 ({})
You can’t perform that action at this time.
0 commit comments