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

Fix authtoken managment command (#5415)

Browse Source 
* Fix authtoken managment command username param
This commit is contained in:
Sergei Azarkin 2017-09-12 16:03:29 +03:00 committed by Tom Christie
parent 7037ce88e9
commit 9aaea2586b
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -19,7 +19,7 @@ class Command(BaseCommand):
return token[0] return token[0]
def add_arguments(self, parser): def add_arguments(self, parser):
parser.add_argument('username', type=str, nargs='+') parser.add_argument('username', type=str)
parser.add_argument( parser.add_argument(
'-r', '-r',

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())