diff options
Diffstat (limited to 'parts/django/tests/modeltests/user_commands/tests.py')
-rw-r--r-- | parts/django/tests/modeltests/user_commands/tests.py | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/parts/django/tests/modeltests/user_commands/tests.py b/parts/django/tests/modeltests/user_commands/tests.py new file mode 100644 index 0000000..84aa7a5 --- /dev/null +++ b/parts/django/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',))
\ No newline at end of file |