Django snippets: Testing for failure in management commands

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

Forcing unit test runner to abort after failed test by simonbun 4 years, 8 months ago
Variable._resolve_lookup monkeypatch by showell 2 years, 2 months ago
Django and Twill by spookylukey 3 years, 10 months ago
"Zoom in" on rendered HTML that the test client returns by peterbe 2 years, 9 months ago
Stop tests at the first failure by akaihola 4 years, 1 month ago

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.

Author:: David
Posted:: August 17, 2010
Language:: Python
Django Version:: 1.2
Tags:
Score:: 1 (after 1 ratings)

Django snippets

Testing for failure in management commands

More like this

Tools

Comments

Snippets

About