def test_compress_raises_a_non_zlib_error___error_is_raised_without_converting(self):
class FakeCompressor(object):
def compress(self, *args, **kwargs):
raise ValueError()
with patch('zlib.compressobj', Mock(return_value=FakeCompressor())), self.assertRaises(ValueError):
compress_data(b'data')
def test_compress_raises_a_zlib_error___error_is_converted_to_oasis_error(self):
class FakeCompressor(object):
def compress(self, *args, **kwargs):
raise zlib.error()
with patch('zlib.compressobj', Mock(return_value=FakeCompressor())), self.assertRaises(OasisException):
compress_data(b'data')
def test_data_is_larger_than_the_chunk_size___result_is_the_concatenated_compressed_version_of_the_chunks(self, front, overflow):
compressor = zlib.compressobj()
expected = compressor.compress(front) + compressor.compress(overflow) + compressor.flush()
result = compress_data(front + overflow)
self.assertEqual(expected, result)
def test_data_is_equal_to_the_the_chunk_size___result_is_the_compressed_version_of_the_full_data(self, data):
compressor = zlib.compressobj()
expected = compressor.compress(data) + compressor.flush()
result = compress_data(data)
self.assertEqual(expected, result)
