Skip to content

Commit

Permalink
increase test coverage for drf_create_token command
Browse files Browse the repository at this point in the history
  • Loading branch information
Vasyl Dizhak committed Nov 1, 2017
1 parent 93e75ec commit f50824d
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion tests/test_authtoken.py
@@ -1,7 +1,7 @@
import pytest
from django.contrib.admin import site
from django.contrib.auth.models import User
from django.core.management import call_command
from django.core.management import call_command, CommandError
from django.test import TestCase
from django.utils.six import StringIO

Expand Down Expand Up @@ -71,6 +71,11 @@ def test_command_do_not_reset_user_token(self):

assert first_token_key == second_token_key

def test_command_raising_error_for_invalid_user(self):
out = StringIO()
with pytest.raises(CommandError):
call_command('drf_create_token', 'not_existing_user', stdout=out)

def test_command_output(self):
out = StringIO()
call_command('drf_create_token', self.user.username, stdout=out)
Expand Down

0 comments on commit f50824d

Please sign in to comment.