Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Browse files

Migrated user_commands doctests. Thanks to Eric Florenzano.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@13823 bcc190cf-cafb-0310-a4f2-bffc1f526a37
  • Loading branch information...
commit 050aa57990a5ce7ad92c089705d328f415df097b 1 parent 783656c
@freakboy3742 freakboy3742 authored
View
2  tests/modeltests/user_commands/management/commands/dance.py
@@ -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"])
View
19 tests/modeltests/user_commands/models.py
@@ -12,22 +12,3 @@
``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.
-
-"""}
View
21 tests/modeltests/user_commands/tests.py
@@ -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',))
Please sign in to comment.
Something went wrong with that request. Please try again.