def test_zip_bomb(self): """Verify zip_decompress always returns small chunks.""" original = '\x00' * 100000 bomb = ''.join(isolateserver.zip_compress(original)) decompressed = [] chunk_size = 1000 for chunk in isolateserver.zip_decompress([bomb], chunk_size): self.assertLessEqual(len(chunk), chunk_size) decompressed.append(chunk) self.assertEqual(original, ''.join(decompressed))
def test_compress_and_decompress(self): """Test data === decompress(compress(data)).""" original = [str(x) for x in xrange(0, 1000)] processed = isolateserver.zip_decompress( isolateserver.zip_compress(original)) self.assertEqual(''.join(original), ''.join(processed))