def test_gzip_encoding(self):
original_data = 'foo bar ponies, wooo gzip'
if PY3:
from io import BytesIO
string_io = BytesIO()
else:
string_io = StringIO()
stream = gzip.GzipFile(fileobj=string_io, mode='w')
stream.write(b(original_data))
stream.close()
compressed_data = string_io.getvalue()
self._mock_response.read.return_value = compressed_data
self._mock_response.getheaders.return_value = \
{'Content-Encoding': 'gzip'}
response = Response(response=self._mock_response,
connection=self._mock_connection)
body = response.parse_body()
self.assertEqual(body, original_data)
self._mock_response.getheaders.return_value = \
{'Content-Encoding': 'x-gzip'}
response = Response(response=self._mock_response,
connection=self._mock_connection)
body = response.parse_body()
self.assertEqual(body, original_data)
def test_deflate_encoding(self):
original_data = 'foo bar ponies, wooo zlib'
compressed_data = zlib.compress(b(original_data))
self._mock_response.read.return_value = compressed_data
self._mock_response.getheaders.return_value = \
{'Content-Encoding': 'deflate'}
response = Response(response=self._mock_response,
connection=self._mock_connection)
body = response.parse_body()
self.assertEqual(body, original_data)
self._mock_response.getheaders.return_value = \
{'Content-Encoding': 'zlib'}
response = Response(response=self._mock_response,
connection=self._mock_connection)
body = response.parse_body()
self.assertEqual(body, original_data)
