def test_delete_content_length_if_compression_enabled(): req = make_request('GET', '/') resp = Response(body=b'answer') resp.enable_compression(ContentCoding.gzip) yield from resp.prepare(req) assert resp.content_length is None
async def test_change_content_length_if_compression_enabled() -> None: req = make_request("GET", "/") resp = Response(body=b"answer") resp.enable_compression(ContentCoding.gzip) await resp.prepare(req) assert resp.content_length is not None and resp.content_length != len(b"answer")
async def test_change_content_length_if_compression_enabled(): req = make_request('GET', '/') resp = Response(body=b'answer') resp.enable_compression(ContentCoding.gzip) await resp.prepare(req) assert resp.content_length is not None and \ resp.content_length != len(b'answer')
async def test_change_content_threaded_compression_enabled() -> None: req = make_request("GET", "/") body_thread_size = 1024 body = b"answer" * body_thread_size resp = Response(body=body, zlib_executor_size=body_thread_size) resp.enable_compression(ContentCoding.gzip) await resp.prepare(req) assert gzip.decompress(resp._compressed_body) == body
def test_delete_content_length_if_compression_enabled(self, ResponseImpl): req = self.make_request("GET", "/") resp = Response(body=b"answer") self.assertEqual(6, resp.content_length) resp.enable_compression(ContentCoding.gzip) self.loop.run_until_complete(resp.prepare(req)) self.assertIsNone(resp.content_length)
def test_delete_content_length_if_compression_enabled(): req = make_request('GET', '/') resp = Response(body=b'answer') resp.enable_compression(ContentCoding.gzip) with mock.patch('aiohttp.web_reqrep.PayloadWriter'): yield from resp.prepare(req) assert resp.content_length is None
def test_delete_content_length_if_compression_enabled(): req = make_request('GET', '/') resp = Response(body=b'answer') resp.enable_compression(ContentCoding.gzip) with mock.patch('aiohttp.web_response.PayloadWriter'): yield from resp.prepare(req) assert resp.content_length is None
def test_delete_content_length_if_compression_enabled(self, ResponseImpl): req = self.make_request('GET', '/') resp = Response(body=b'answer') self.assertEqual(6, resp.content_length) resp.enable_compression(ContentCoding.gzip) resp.start(req) self.assertIsNone(resp.content_length)
async def test_change_content_threaded_compression_enabled() -> None: req = make_request('GET', '/') body_thread_size = 1024 body = b'answer' * body_thread_size resp = Response(body=body, zlib_executor_size=body_thread_size) resp.enable_compression(ContentCoding.gzip) await resp.prepare(req) assert gzip.decompress(resp._compressed_body) == body
async def test_change_content_threaded_compression_enabled_explicit() -> None: req = make_request('GET', '/') body_thread_size = 1024 body = b'answer' * body_thread_size with ThreadPoolExecutor(1) as executor: resp = Response(body=body, zlib_executor_size=body_thread_size, zlib_executor=executor) resp.enable_compression(ContentCoding.gzip) await resp.prepare(req) assert gzip.decompress(resp._compressed_body) == body
def test_force_compression_identity_response(): writer = mock.Mock() def write_headers(status_line, headers): assert headers[hdrs.CONTENT_LENGTH] == "6" assert hdrs.TRANSFER_ENCODING not in headers writer.write_headers.side_effect = write_headers req = make_request('GET', '/', payload_writer=writer) resp = Response(body=b'answer') resp.enable_compression(ContentCoding.identity) yield from resp.prepare(req) assert resp.content_length == 6
async def test_rm_content_length_if_compression_http10() -> None: writer = mock.Mock() async def write_headers(status_line, headers): assert hdrs.CONTENT_LENGTH not in headers assert hdrs.TRANSFER_ENCODING not in headers writer.write_headers.side_effect = write_headers req = make_request("GET", "/", version=HttpVersion10, writer=writer) resp = Response(body=BytesPayload(b"answer")) resp.enable_compression(ContentCoding.gzip) await resp.prepare(req) assert resp.content_length is None
async def test_force_compression_identity_response() -> None: writer = mock.Mock() async def write_headers(status_line, headers): assert headers[hdrs.CONTENT_LENGTH] == "6" assert hdrs.TRANSFER_ENCODING not in headers writer.write_headers.side_effect = write_headers req = make_request("GET", "/", writer=writer) resp = Response(body=b"answer") resp.enable_compression(ContentCoding.identity) await resp.prepare(req) assert resp.content_length == 6
async def test_remove_content_length_if_compression_enabled_on_payload_http10(): # noqa writer = mock.Mock() def write_headers(status_line, headers): assert hdrs.CONTENT_LENGTH not in headers assert hdrs.TRANSFER_ENCODING not in headers writer.write_headers.side_effect = write_headers req = make_request('GET', '/', version=HttpVersion10, payload_writer=writer) resp = Response(body=BytesPayload(b'answer')) resp.enable_compression(ContentCoding.gzip) await resp.prepare(req) assert resp.content_length is None
async def test_rm_content_length_if_compression_enabled_on_payload_http10(): writer = mock.Mock() async def write_headers(status_line, headers): assert hdrs.CONTENT_LENGTH not in headers assert hdrs.TRANSFER_ENCODING not in headers writer.write_headers.side_effect = write_headers req = make_request('GET', '/', version=HttpVersion10, writer=writer) resp = Response(body=BytesPayload(b'answer')) resp.enable_compression(ContentCoding.gzip) await resp.prepare(req) assert resp.content_length is None
async def test_force_compression_identity_response(): writer = mock.Mock() async def write_headers(status_line, headers): assert headers[hdrs.CONTENT_LENGTH] == "6" assert hdrs.TRANSFER_ENCODING not in headers writer.write_headers.side_effect = write_headers req = make_request('GET', '/', writer=writer) resp = Response(body=b'answer') resp.enable_compression(ContentCoding.identity) await resp.prepare(req) assert resp.content_length == 6
def test_remove_content_length_if_compression_enabled_on_payload_http11(): writer = mock.Mock() def write_headers(status_line, headers): assert hdrs.CONTENT_LENGTH not in headers assert headers.get(hdrs.TRANSFER_ENCODING, '') == 'chunked' writer.write_headers.side_effect = write_headers req = make_request('GET', '/', payload_writer=writer) payload = BytesPayload(b'answer', headers={"X-Test-Header": "test"}) resp = Response(body=payload) assert resp.content_length == 6 resp.body = payload resp.enable_compression(ContentCoding.gzip) yield from resp.prepare(req) assert resp.content_length is None
async def test_rm_content_length_if_compression_http11() -> None: writer = mock.Mock() async def write_headers(status_line, headers): assert hdrs.CONTENT_LENGTH not in headers assert headers.get(hdrs.TRANSFER_ENCODING, "") == "chunked" writer.write_headers.side_effect = write_headers req = make_request("GET", "/", writer=writer) payload = BytesPayload(b"answer", headers={"X-Test-Header": "test"}) resp = Response(body=payload) assert resp.content_length == 6 resp.body = payload resp.enable_compression(ContentCoding.gzip) await resp.prepare(req) assert resp.content_length is None
async def test_rm_content_length_if_compression_enabled_on_payload_http11(): writer = mock.Mock() async def write_headers(status_line, headers): assert hdrs.CONTENT_LENGTH not in headers assert headers.get(hdrs.TRANSFER_ENCODING, '') == 'chunked' writer.write_headers.side_effect = write_headers req = make_request('GET', '/', writer=writer) payload = BytesPayload(b'answer', headers={"X-Test-Header": "test"}) resp = Response(body=payload) assert resp.content_length == 6 resp.body = payload resp.enable_compression(ContentCoding.gzip) await resp.prepare(req) assert resp.content_length is None
async def test_set_content_length_if_compression_enabled(): writer = mock.Mock() async def write_headers(status_line, headers): assert hdrs.CONTENT_LENGTH in headers assert headers[hdrs.CONTENT_LENGTH] == '26' assert hdrs.TRANSFER_ENCODING not in headers writer.write_headers.side_effect = write_headers req = make_request('GET', '/', writer=writer) resp = Response(body=b'answer') resp.enable_compression(ContentCoding.gzip) await resp.prepare(req) assert resp.content_length == 26 del resp.headers[hdrs.CONTENT_LENGTH] assert resp.content_length == 26
async def test_set_content_length_if_compression_enabled() -> None: writer = mock.Mock() async def write_headers(status_line, headers): assert hdrs.CONTENT_LENGTH in headers assert headers[hdrs.CONTENT_LENGTH] == '26' assert hdrs.TRANSFER_ENCODING not in headers writer.write_headers.side_effect = write_headers req = make_request('GET', '/', writer=writer) resp = Response(body=b'answer') resp.enable_compression(ContentCoding.gzip) await resp.prepare(req) assert resp.content_length == 26 del resp.headers[hdrs.CONTENT_LENGTH] assert resp.content_length == 26