52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
|
from unittest import TestCase
|
||
|
|
from StringIO import StringIO
|
||
|
|
from django.core.servers.basehttp import ServerHandler
|
||
|
|
|
||
|
|
#
|
||
|
|
# Tests for #9659: wsgi.file_wrapper in the builtin server.
|
||
|
|
# We need to mock a couple of of handlers and keep track of what
|
||
|
|
# gets called when using a couple kinds of WSGI apps.
|
||
|
|
#
|
||
|
|
|
||
|
|
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 wsgi_app(environ, start_response):
|
||
|
|
start_response('200 OK', [('Content-Type', 'text/plain')])
|
||
|
|
return ['Hello World!']
|
||
|
|
|
||
|
|
def wsgi_app_file_wrapper(environ, start_response):
|
||
|
|
start_response('200 OK', [('Content-Type', 'text/plain')])
|
||
|
|
return environ['wsgi.file_wrapper'](StringIO('foo'))
|
||
|
|
|
||
|
|
class WSGIFileWrapperTests(TestCase):
|
||
|
|
"""
|
||
|
|
Test that the wsgi.file_wrapper works for the builting server.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def test_file_wrapper_uses_sendfile(self):
|
||
|
|
env = {'SERVER_PROTOCOL': 'HTTP/1.0'}
|
||
|
|
err = StringIO()
|
||
|
|
handler = FileWrapperHandler(None, StringIO(), err, env)
|
||
|
|
handler.run(wsgi_app_file_wrapper)
|
||
|
|
self.assert_(handler._used_sendfile)
|
||
|
|
|
||
|
|
def test_file_wrapper_no_sendfile(self):
|
||
|
|
env = {'SERVER_PROTOCOL': 'HTTP/1.0'}
|
||
|
|
err = StringIO()
|
||
|
|
handler = FileWrapperHandler(None, StringIO(), err, env)
|
||
|
|
handler.run(wsgi_app)
|
||
|
|
self.failIf(handler._used_sendfile)
|
||
|
|
self.assertEqual(handler.stdout.getvalue().splitlines()[-1],'Hello World!')
|