0
0
mirror of https://github.com/django/django.git synced 2024-11-29 22:56:46 +01:00
django/tests/modeltests/user_commands/tests.py
2011-10-13 18:04:12 +00:00

22 lines
707 B
Python

from StringIO import StringIO
from django.core import management
from django.core.management.base import CommandError
from django.test import TestCase
class CommandTests(TestCase):
def test_command(self):
out = StringIO()
management.call_command('dance', stdout=out)
self.assertEqual(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.assertEqual(out.getvalue(),
"I don't feel like dancing Jive.")
def test_explode(self):
self.assertRaises(CommandError, management.call_command, ('explode',))