Kaydet (Commit) 050aa579 authored tarafından Russell Keith-Magee's avatar Russell Keith-Magee

Migrated user_commands doctests. Thanks to Eric Florenzano.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@13823 bcc190cf-cafb-0310-a4f2-bffc1f526a37
üst 783656cf
......@@ -11,4 +11,4 @@ class Command(BaseCommand):
]
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
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.
"""}
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',))
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment