django/tests/regressiontests/middleware_exceptions/tests.py

import sys

from django.test import TestCase
from django.core.signals import got_request_exception

class TestException(Exception):
    pass

class TestRequestMiddleware(object):
    def process_request(self, request):
        raise TestException('Test Exception')

class TestResponseMiddleware(object):
    def process_response(self, request, response):
        raise TestException('Test Exception')

class MiddlewareExceptionTest(TestCase):
    def setUp(self):
        self.exceptions = []
        got_request_exception.connect(self._on_request_exception)
        self.client.handler.load_middleware()

    def tearDown(self):
        got_request_exception.disconnect(self._on_request_exception)
        self.exceptions = []

    def _on_request_exception(self, sender, request, **kwargs):
        self.exceptions.append(sys.exc_info())

    def _assert_exception_handled(self):
        try:
            response = self.client.get('/middleware_exceptions/')
        except TestException, e:
            # Test client intentionally re-raises any exceptions being raised
            # during request handling. Hence actual testing that exception was
            # properly handled is done by relying on got_request_exception
            # signal being sent.
            pass
        except Exception, e:
            self.fail("Unexpected exception: %s" % e)
        self.assertEquals(len(self.exceptions), 1)
        exception, value, tb = self.exceptions[0]
        self.assertEquals(value.args, ('Test Exception', ))

    def test_process_request(self):
        self.client.handler._request_middleware.insert(0, TestRequestMiddleware().process_request)
        self._assert_exception_handled()

    def test_process_response(self):
        self.client.handler._response_middleware.insert(0, TestResponseMiddleware().process_response)
        self._assert_exception_handled()