Migrated user_commands doctests. Thanks to Eric Florenzano.
git-svn-id: http://code.djangoproject.com/svn/django/trunk@13823 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
parent
783656cf5b
commit
050aa57990
|
@ -11,4 +11,4 @@ class Command(BaseCommand):
|
||||||
]
|
]
|
||||||
|
|
||||||
def handle(self, *args, **options):
|
def handle(self, *args, **options):
|
||||||
print "I don't feel like dancing %s." % options["style"]
|
self.stdout.write("I don't feel like dancing %s." % options["style"])
|
||||||
|
|
|
@ -12,22 +12,3 @@ For more details on how to define your own ``manage.py`` commands, look at the
|
||||||
``django.core.management.commands`` directory. This directory contains the
|
``django.core.management.commands`` directory. This directory contains the
|
||||||
definitions for the base Django ``manage.py`` commands.
|
definitions for the base Django ``manage.py`` commands.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
__test__ = {'API_TESTS': """
|
|
||||||
>>> from django.core import management
|
|
||||||
|
|
||||||
# Invoke a simple user-defined command
|
|
||||||
>>> management.call_command('dance', style="Jive")
|
|
||||||
I don't feel like dancing Jive.
|
|
||||||
|
|
||||||
# Invoke a command that doesn't exist
|
|
||||||
>>> management.call_command('explode')
|
|
||||||
Traceback (most recent call last):
|
|
||||||
...
|
|
||||||
CommandError: Unknown command: 'explode'
|
|
||||||
|
|
||||||
# Invoke a command with default option `style`
|
|
||||||
>>> management.call_command('dance')
|
|
||||||
I don't feel like dancing Rock'n'Roll.
|
|
||||||
|
|
||||||
"""}
|
|
||||||
|
|
|
@ -0,0 +1,21 @@
|
||||||
|
from StringIO import StringIO
|
||||||
|
|
||||||
|
from django.test import TestCase
|
||||||
|
from django.core import management
|
||||||
|
from django.core.management.base import CommandError
|
||||||
|
|
||||||
|
class CommandTests(TestCase):
|
||||||
|
def test_command(self):
|
||||||
|
out = StringIO()
|
||||||
|
management.call_command('dance', stdout=out)
|
||||||
|
self.assertEquals(out.getvalue(),
|
||||||
|
"I don't feel like dancing Rock'n'Roll.")
|
||||||
|
|
||||||
|
def test_command_style(self):
|
||||||
|
out = StringIO()
|
||||||
|
management.call_command('dance', style='Jive', stdout=out)
|
||||||
|
self.assertEquals(out.getvalue(),
|
||||||
|
"I don't feel like dancing Jive.")
|
||||||
|
|
||||||
|
def test_explode(self):
|
||||||
|
self.assertRaises(CommandError, management.call_command, ('explode',))
|
Loading…
Reference in New Issue