mirror of
				https://github.com/encode/django-rest-framework.git
				synced 2025-10-31 07:57:55 +03:00 
			
		
		
		
	Add couple of test for AuthToken (#4816)
* Add couple of test for AuthToken * PEP8 cleanup
This commit is contained in:
		
							parent
							
								
									5efbe17a86
								
							
						
					
					
						commit
						514b5a6dd1
					
				
							
								
								
									
										29
									
								
								tests/test_authtoken.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										29
									
								
								tests/test_authtoken.py
									
									
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,29 @@ | |||
| import pytest | ||||
| from django.contrib.admin import site | ||||
| from django.contrib.auth.models import User | ||||
| from django.test import TestCase | ||||
| 
 | ||||
| from rest_framework.authtoken.admin import TokenAdmin | ||||
| from rest_framework.authtoken.models import Token | ||||
| from rest_framework.authtoken.serializers import AuthTokenSerializer | ||||
| from rest_framework.exceptions import ValidationError | ||||
| 
 | ||||
| 
 | ||||
| class AuthTokenTests(TestCase): | ||||
| 
 | ||||
|     def setUp(self): | ||||
|         self.site = site | ||||
|         self.user = User.objects.create_user(username='test_user') | ||||
|         self.token = Token.objects.create(key='test token', user=self.user) | ||||
| 
 | ||||
|     def test_model_admin_displayed_fields(self): | ||||
|         mock_request = object() | ||||
|         token_admin = TokenAdmin(self.token, self.site) | ||||
|         assert token_admin.get_fields(mock_request) == ('user',) | ||||
| 
 | ||||
|     def test_token_string_representation(self): | ||||
|         assert str(self.token) == 'test token' | ||||
| 
 | ||||
|     def test_validate_raise_error_if_no_credentials_provided(self): | ||||
|         with pytest.raises(ValidationError): | ||||
|             AuthTokenSerializer().validate({}) | ||||
		Loading…
	
		Reference in New Issue
	
	Block a user