encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-08-04 12:30:11 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix authtoken managment command username param

Browse Source
This commit is contained in:
Sergei Azarkin 2017-09-12 14:18:25 +03:00
parent a560a49d56
commit f98641ef97
2 changed files with 21 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
rest_framework/authtoken/management/commands/drf_create_token.py
View File

@ -31,15 +31,16 @@ class Command(BaseCommand):
) )
def handle(self, *args, **options): def handle(self, *args, **options):
username = options['username'] usernames = options['username']
reset_token = options['reset_token'] reset_token = options['reset_token']
try: for username in usernames:
token = self.create_user_token(username, reset_token) try:
except UserModel.DoesNotExist: token = self.create_user_token(username, reset_token)
raise CommandError( except UserModel.DoesNotExist:
'Cannot create the Token: user {0} does not exist'.format( raise CommandError(
username) 'Cannot create the Token: user {0} does not exist'.format(
) username)
self.stdout.write( )
'Generated token {0} for user {1}'.format(token.key, username)) self.stdout.write(
'Generated token {0} for user {1}'.format(token.key, username))

10
tests/test_authtoken.py
View File

@ -1,7 +1,9 @@
import pytest import pytest
from django.contrib.admin import site from django.contrib.admin import site
from django.contrib.auth.models import User from django.contrib.auth.models import User
from django.core.management import call_command
from django.test import TestCase from django.test import TestCase
from django.utils.six import StringIO
from rest_framework.authtoken.admin import TokenAdmin from rest_framework.authtoken.admin import TokenAdmin
from rest_framework.authtoken.management.commands.drf_create_token import \ from rest_framework.authtoken.management.commands.drf_create_token import \
@ -68,3 +70,11 @@ class AuthTokenCommandTests(TestCase):
second_token_key = Token.objects.first().key second_token_key = Token.objects.first().key
assert first_token_key == second_token_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())