mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-07-19 11:15:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Fixed #29295 -- Fixed management command crash when using subparsers.

Browse source 
Thanks Tim Graham for the fix.
This commit is contained in:
Hasan Ramezani 2018-04-21 23:04:00 +04:30 • committed by Tim Graham
parent 21420096c4
commit dd68b51e1d
4 changed files with 32 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/user_commands/management/commands/subparser.py Normal file
View file

@ -0,0 +1,12 @@
from django.core.management.base import BaseCommand
class Command(BaseCommand):
def add_arguments(self, parser):
subparsers = parser.add_subparsers()
parser_foo = subparsers.add_parser('foo')
parser_foo.add_argument('bar', type=int)
def handle(self, *args, **options):
self.stdout.write(','.join(options))

10
tests/user_commands/tests.py
View file

@ -206,6 +206,16 @@ class CommandTests(SimpleTestCase):
self.assertIn('need_me', out.getvalue())
self.assertIn('needme2', out.getvalue())
def test_subparser(self):
out = StringIO()
management.call_command('subparser', 'foo', 12, stdout=out)
self.assertIn('bar', out.getvalue())
def test_subparser_invalid_option(self):
msg = "Error: invalid choice: 'test' (choose from 'foo')"
with self.assertRaisesMessage(CommandError, msg):
management.call_command('subparser', 'test', 12)
class CommandRunTests(AdminScriptTestCase):
"""