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