diff options
| author | django-bot <ops@djangoproject.com> | 2022-02-03 20:24:19 +0100 |
|---|---|---|
| committer | Mariusz Felisiak <felisiak.mariusz@gmail.com> | 2022-02-07 20:37:05 +0100 |
| commit | 9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch) | |
| tree | f0506b668a013d0063e5fba3dbf4863b466713ba /tests/builtin_server | |
| parent | f68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff) | |
Refs #33476 -- Reformatted code with Black.
Diffstat (limited to 'tests/builtin_server')
| -rw-r--r-- | tests/builtin_server/tests.py | 46 | ||||
| -rw-r--r-- | tests/builtin_server/urls.py | 2 | ||||
| -rw-r--r-- | tests/builtin_server/views.py | 4 |
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 |
