Ticket #14523: 14523-review.diff

django/core/handlers/base.py

@@ -120 +120 @@
                         view_name = callback.__class__.__name__ + '.__call__' # If it's a class
                     raise ValueError("The view %s.%s didn't return an HttpResponse object." % (callback.__module__, view_name))
 
+                # Apply response middleware
+                for middleware_method in self._response_middleware:
+                    response = middleware_method(request, response)
+                response = self.apply_response_fixes(request, response)
+
                 return response
             except http.Http404, e:
                 logger.warning('Not Found: %s' % request.path,

django/core/handlers/modpython.py

@@ -189 +189 @@
                 response = http.HttpResponseBadRequest()
             else:
                 response = self.get_response(request)
-
-                # Apply response middleware
-                for middleware_method in self._response_middleware:
-                    response = middleware_method(request, response)
-                response = self.apply_response_fixes(request, response)
         finally:
             signals.request_finished.send(sender=self.__class__)
 

django/core/handlers/wsgi.py

@@ -254 +254 @@
                 response = http.HttpResponseBadRequest()
             else:
                 response = self.get_response(request)
-
-                # Apply response middleware
-                for middleware_method in self._response_middleware:
-                    response = middleware_method(request, response)
-                response = self.apply_response_fixes(request, response)
         finally:
             signals.request_finished.send(sender=self.__class__)
 

tests/regressiontests/middleware_exceptions/tests.py

@@ -6 +6 @@
 class TestException(Exception):
     pass
 
-class TestMiddleware(object):
+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 = []
@@ -23 +27 @@
     def _on_request_exception(self, sender, request, **kwargs):
         self.exceptions.append(sys.exc_info())
 
-    def test_process_request(self):
-        self.client.handler._request_middleware.insert(0, TestMiddleware().process_request)
+    def _assert_exception_handled(self):
         try:
-            response = self.client.get('/')
+            response = self.client.get('/middleware_exceptions/')
         except TestException, e:
             # Test client indefinitely re-raises any exceptions being raised
             # during request handling. Hence actual testing that exception was
@@ -38 +41 @@
         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()