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
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_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)
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
