Ticket #9659: 9659-fixed-filewrapper.2.patch

regressiontests/basehttp_server/tests.py

@@ +1 @@
+from unittest import TestCase
+from StringIO import StringIO
+
+from django.core.servers.basehttp import ServerHandler
+
+
+class DummyHandler(object):
+
+    def log_request(*args, **kwargs):
+        pass
+
+
+class FileWrapperHandler(ServerHandler):
+
+    def __init__(self, *args, **kwargs):
+        ServerHandler.__init__(self, *args, **kwargs)
+        self.request_handler = DummyHandler()
+        self._used_sendfile = False
+
+    def sendfile(self):
+        self._used_sendfile = True
+        return True
+
+
+def hello_file(environ, start_response):
+    start_response('200 OK', [('Content-Type', 'text/plain')])
+    return environ['wsgi.file_wrapper'](StringIO('foo'))
+
+
+def hello_world(environ, start_response):
+    start_response('200 OK', [('Content-Type', 'text/plain')])
+    return ['Hello World!']
+
+
+class Bug9659Test(TestCase):
+    """
+    Test that the file wrapper works.
+    """
+
+    def test_bug_9659(self):
+        env = {'SERVER_PROTOCOL': 'HTTP/1.0'}
+        err = StringIO()
+        handler = FileWrapperHandler(None, StringIO(), err, env)
+        handler.run(hello_file)
+        self.assert_(handler._used_sendfile)
+
+        handler = FileWrapperHandler(None, StringIO(), err, env)
+        handler.run(hello_world)
+        self.assert_(not handler._used_sendfile)
+        self.assertEqual(handler.stdout.getvalue().splitlines()[-1],
+                         'Hello World!')