Because BaseCommand catches all CommandError exceptions and turns them into nicely formatted strings before calling sys.exit(1), it can be tricky to properly test failure.
Normally, this is great, but it makes testing that a command fails when we expect to a little harder. That's where this snippet comes in. It redirects sys.stderr to an instance StringIO where we can monitor what's output and catches the BaseException raised by sys.exit(1). Form here, it's trivial to test that a management command fails exactly as you'd expect it to.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 | import cStrinIO
import sys
from django.test import TestCase
class MyCommandTestCase(TestCase):
def test_mycommand_failure(self):
old_stderr = sys.stderr
sys.stderr = cStringIO.StringIO()
self.assertRaises(BaseException, management.call_command, 'mycommand')
self.assertTrue('Error: Some error condition here' in sys.stderr.getvalue())
sys.stderr = old_stderr
|
More like this
- Template tag - list punctuation for a list of items by shapiromatron 10 months, 2 weeks ago
- JSONRequestMiddleware adds a .json() method to your HttpRequests by cdcarter 10 months, 2 weeks ago
- Serializer factory with Django Rest Framework by julio 1 year, 5 months ago
- Image compression before saving the new model / work with JPG, PNG by Schleidens 1 year, 6 months ago
- Help text hyperlinks by sa2812 1 year, 6 months ago
Comments
Please login first before commenting.