Ticket #5525: distrib_server_response_code_message.patch

twisted/web/server.py

@@ -286 +286 @@
         """
         self.setETag(tag)
 
-    def view_setResponseCode(self, issuer, code):
+    def view_setResponseCode(self, issuer, code, message = None):
         """Remote version of setResponseCode; same interface.
         """
-        self.setResponseCode(code)
+        self.setResponseCode(code, message)
 
     def view_registerProducer(self, issuer, producer, streaming):
         """Remote version of registerProducer; same interface.

twisted/web/test/test_distrib.py

@@ -145 +145 @@
         request.addCallback(cbRequested)
         return request
 
+    def test_requestResponseCode(self):
+        """
+        The response code can be set by the request object passed to a
+        distributed resource's C{render} method.
+        """
+        class SetResponseCode(resource.Resource):
+            def render(self, request):
+                request.setResponseCode(200)
+                return ""
+
+        request = self._requestTest(SetResponseCode())
+        def cbRequested(result):
+            self.assertEqual(result, "")
+        request.addCallback(cbRequested)
+        return request
+
+    def test_requestResponseCode_message(self):
+        """
+        The response code and message can be set by the request object passed to a
+        distributed resource's C{render} method.
+        """
+        class SetResponseCode(resource.Resource):
+            def render(self, request):
+                request.setResponseCode(200, "OK")
+                return ""
+
+        request = self._requestTest(SetResponseCode())
+        def cbRequested(result):
+            self.assertEqual(result, "")
+        request.addCallback(cbRequested)
+        return request
 
     def test_largeWrite(self):
         """