Fix authtoken managment command (#5415)

sazarkin · tomchristie · commit 9aaea2586bf5 · 2017-09-12T14:03:29.000+01:00
* Fix authtoken managment command username param
diff --git a/rest_framework/authtoken/management/commands/drf_create_token.py b/rest_framework/authtoken/management/commands/drf_create_token.py
@@ -19,7 +19,7 @@ def create_user_token(self, username, reset_token):
         return token[0]
 
     def add_arguments(self, parser):
-        parser.add_argument('username', type=str, nargs='+')
+        parser.add_argument('username', type=str)
 
         parser.add_argument(
             '-r',
diff --git a/tests/test_authtoken.py b/tests/test_authtoken.py
@@ -1,7 +1,9 @@
 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 \
@@ -68,3 +70,11 @@ def test_command_do_not_reset_user_token(self):
         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())
