mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-09-21 03:19:00 +03:00
81 lines
3.0 KiB
Python
81 lines
3.0 KiB
Python
import pytest
|
|
from django.contrib.admin import site
|
|
from django.contrib.auth.models import User
|
|
from django.core.management import call_command
|
|
from django.test import TestCase
|
|
from django.utils.six import StringIO
|
|
|
|
from rest_framework.authtoken.admin import TokenAdmin
|
|
from rest_framework.authtoken.management.commands.drf_create_token import \
|
|
Command as AuthTokenCommand
|
|
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({})
|
|
|
|
def test_whitespace_in_password(self):
|
|
data = {'username': self.user.username, 'password': 'test pass '}
|
|
self.user.set_password(data['password'])
|
|
self.user.save()
|
|
assert AuthTokenSerializer(data=data).is_valid()
|
|
|
|
|
|
class AuthTokenCommandTests(TestCase):
|
|
|
|
def setUp(self):
|
|
self.site = site
|
|
self.user = User.objects.create_user(username='test_user')
|
|
|
|
def test_command_create_user_token(self):
|
|
token = AuthTokenCommand().create_user_token(self.user.username, False)
|
|
assert token is not None
|
|
token_saved = Token.objects.first()
|
|
assert token.key == token_saved.key
|
|
|
|
def test_command_create_user_token_invalid_user(self):
|
|
with pytest.raises(User.DoesNotExist):
|
|
AuthTokenCommand().create_user_token('not_existing_user', False)
|
|
|
|
def test_command_reset_user_token(self):
|
|
AuthTokenCommand().create_user_token(self.user.username, False)
|
|
first_token_key = Token.objects.first().key
|
|
AuthTokenCommand().create_user_token(self.user.username, True)
|
|
second_token_key = Token.objects.first().key
|
|
|
|
assert first_token_key != second_token_key
|
|
|
|
def test_command_do_not_reset_user_token(self):
|
|
AuthTokenCommand().create_user_token(self.user.username, False)
|
|
first_token_key = Token.objects.first().key
|
|
AuthTokenCommand().create_user_token(self.user.username, False)
|
|
second_token_key = Token.objects.first().key
|
|
|
|
assert first_token_key == second_token_key
|
|
|
|
def test_command_output(self):
|
|
out = StringIO()
|
|
call_command('drf_create_token', self.user.username, stdout=out)
|
|
token_saved = Token.objects.first()
|
|
self.assertIn('Generated token', out.getvalue())
|
|
self.assertIn(self.user.username, out.getvalue())
|
|
self.assertIn(token_saved.key, out.getvalue())
|