summaryrefslogtreecommitdiff
path: root/tests/builtin_server
diff options
context:
space:
mode:
authordjango-bot <ops@djangoproject.com>2022-02-03 20:24:19 +0100
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2022-02-07 20:37:05 +0100
commit9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch)
treef0506b668a013d0063e5fba3dbf4863b466713ba /tests/builtin_server
parentf68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff)
Refs #33476 -- Reformatted code with Black.
Diffstat (limited to 'tests/builtin_server')
-rw-r--r--tests/builtin_server/tests.py46
-rw-r--r--tests/builtin_server/urls.py2
-rw-r--r--tests/builtin_server/views.py4
3 files changed, 26 insertions, 26 deletions
diff --git a/tests/builtin_server/tests.py b/tests/builtin_server/tests.py
index 7c0a889432..2777db1e13 100644
--- a/tests/builtin_server/tests.py
+++ b/tests/builtin_server/tests.py
@@ -28,20 +28,20 @@ class ServerHandler(simple_server.ServerHandler):
elif not self.headers_sent:
# Before the first output, send the stored headers
- self.bytes_sent = len(data) # make sure we know content-length
+ self.bytes_sent = len(data) # make sure we know content-length
self.send_headers()
else:
self.bytes_sent += len(data)
# XXX check Content-Length and truncate if too many bytes written?
data = BytesIO(data)
- for chunk in iter(lambda: data.read(MAX_SOCKET_CHUNK_SIZE), b''):
+ for chunk in iter(lambda: data.read(MAX_SOCKET_CHUNK_SIZE), b""):
self._write(chunk)
self._flush()
def error_output(self, environ, start_response):
super().error_output(environ, start_response)
- return ['\n'.join(traceback.format_exception(*sys.exc_info()))]
+ return ["\n".join(traceback.format_exception(*sys.exc_info()))]
class DummyHandler:
@@ -61,13 +61,13 @@ class FileWrapperHandler(ServerHandler):
def wsgi_app(environ, start_response):
- start_response('200 OK', [('Content-Type', 'text/plain')])
- return [b'Hello World!']
+ start_response("200 OK", [("Content-Type", "text/plain")])
+ return [b"Hello World!"]
def wsgi_app_file_wrapper(environ, start_response):
- start_response('200 OK', [('Content-Type', 'text/plain')])
- return environ['wsgi.file_wrapper'](BytesIO(b'foo'))
+ start_response("200 OK", [("Content-Type", "text/plain")])
+ return environ["wsgi.file_wrapper"](BytesIO(b"foo"))
class WSGIFileWrapperTests(TestCase):
@@ -80,45 +80,45 @@ class WSGIFileWrapperTests(TestCase):
"""
def test_file_wrapper_uses_sendfile(self):
- env = {'SERVER_PROTOCOL': 'HTTP/1.0'}
+ env = {"SERVER_PROTOCOL": "HTTP/1.0"}
handler = FileWrapperHandler(None, BytesIO(), BytesIO(), env)
handler.run(wsgi_app_file_wrapper)
self.assertTrue(handler._used_sendfile)
- self.assertEqual(handler.stdout.getvalue(), b'')
- self.assertEqual(handler.stderr.getvalue(), b'')
+ self.assertEqual(handler.stdout.getvalue(), b"")
+ self.assertEqual(handler.stderr.getvalue(), b"")
def test_file_wrapper_no_sendfile(self):
- env = {'SERVER_PROTOCOL': 'HTTP/1.0'}
+ env = {"SERVER_PROTOCOL": "HTTP/1.0"}
handler = FileWrapperHandler(None, BytesIO(), BytesIO(), env)
handler.run(wsgi_app)
self.assertFalse(handler._used_sendfile)
- self.assertEqual(handler.stdout.getvalue().splitlines()[-1], b'Hello World!')
- self.assertEqual(handler.stderr.getvalue(), b'')
+ self.assertEqual(handler.stdout.getvalue().splitlines()[-1], b"Hello World!")
+ self.assertEqual(handler.stderr.getvalue(), b"")
- @override_settings(ROOT_URLCONF='builtin_server.urls')
+ @override_settings(ROOT_URLCONF="builtin_server.urls")
def test_file_response_closing(self):
"""
View returning a FileResponse properly closes the file and http
response when file_wrapper is used.
"""
- env = RequestFactory().get('/fileresponse/').environ
+ env = RequestFactory().get("/fileresponse/").environ
handler = FileWrapperHandler(None, BytesIO(), BytesIO(), env)
handler.run(get_internal_wsgi_application())
# Sendfile is used only when file_wrapper has been used.
self.assertTrue(handler._used_sendfile)
# Fetch the original response object.
- self.assertIn('response', FILE_RESPONSE_HOLDER)
- response = FILE_RESPONSE_HOLDER['response']
+ self.assertIn("response", FILE_RESPONSE_HOLDER)
+ response = FILE_RESPONSE_HOLDER["response"]
# The response and file buffers are closed.
self.assertIs(response.closed, True)
- buf1, buf2 = FILE_RESPONSE_HOLDER['buffers']
+ buf1, buf2 = FILE_RESPONSE_HOLDER["buffers"]
self.assertIs(buf1.closed, True)
self.assertIs(buf2.closed, True)
FILE_RESPONSE_HOLDER.clear()
- @override_settings(ROOT_URLCONF='builtin_server.urls')
+ @override_settings(ROOT_URLCONF="builtin_server.urls")
def test_file_response_call_request_finished(self):
- env = RequestFactory().get('/fileresponse/').environ
+ env = RequestFactory().get("/fileresponse/").environ
handler = FileWrapperHandler(None, BytesIO(), BytesIO(), env)
with mock.MagicMock() as signal_handler:
request_finished.connect(signal_handler)
@@ -149,9 +149,9 @@ class WriteChunkCounterHandler(ServerHandler):
def send_big_data_app(environ, start_response):
- start_response('200 OK', [('Content-Type', 'text/plain')])
+ start_response("200 OK", [("Content-Type", "text/plain")])
# Return a blob of data that is 1.5 times the maximum chunk size.
- return [b'x' * (MAX_SOCKET_CHUNK_SIZE + MAX_SOCKET_CHUNK_SIZE // 2)]
+ return [b"x" * (MAX_SOCKET_CHUNK_SIZE + MAX_SOCKET_CHUNK_SIZE // 2)]
class ServerHandlerChunksProperly(TestCase):
@@ -164,7 +164,7 @@ class ServerHandlerChunksProperly(TestCase):
"""
def test_chunked_data(self):
- env = {'SERVER_PROTOCOL': 'HTTP/1.0'}
+ env = {"SERVER_PROTOCOL": "HTTP/1.0"}
handler = WriteChunkCounterHandler(None, BytesIO(), BytesIO(), env)
handler.run(send_big_data_app)
self.assertEqual(handler.write_chunk_counter, 2)
diff --git a/tests/builtin_server/urls.py b/tests/builtin_server/urls.py
index c26366f1e6..e7654359e0 100644
--- a/tests/builtin_server/urls.py
+++ b/tests/builtin_server/urls.py
@@ -3,5 +3,5 @@ from django.urls import path
from . import views
urlpatterns = [
- path('fileresponse/', views.file_response),
+ path("fileresponse/", views.file_response),
]
diff --git a/tests/builtin_server/views.py b/tests/builtin_server/views.py
index be7c7e94ab..fdadb04ccb 100644
--- a/tests/builtin_server/views.py
+++ b/tests/builtin_server/views.py
@@ -10,6 +10,6 @@ def file_response(request):
f2 = BytesIO(b"test2")
response = FileResponse(f1)
response._resource_closers.append(f2.close)
- FILE_RESPONSE_HOLDER['response'] = response
- FILE_RESPONSE_HOLDER['buffers'] = (f1, f2)
+ FILE_RESPONSE_HOLDER["response"] = response
+ FILE_RESPONSE_HOLDER["buffers"] = (f1, f2)
return response