1
0
mirror of https://github.com/webrecorder/pywb.git synced 2025-03-24 06:59:52 +01:00

tests: update test to support uncompressed followed after compressed block

This commit is contained in:
Ilya Kreymer 2016-12-08 14:20:46 -08:00
parent bc219acb33
commit 4f9b963e13

View File

@ -141,20 +141,20 @@ def test_brotli():
x.read() == b'The quick brown fox jumps over the lazy dog' * 4096 x.read() == b'The quick brown fox jumps over the lazy dog' * 4096
# Compression
# Errors def test_compress_mix():
# error: compressed member, followed by not compressed -- now allowed!
def test_err_compress_mix():
# error: compressed member, followed by not compressed -- considered invalid
x = DecompressingBufferedReader(BytesIO(compress('ABC') + b'123'), decomp_type = 'gzip') x = DecompressingBufferedReader(BytesIO(compress('ABC') + b'123'), decomp_type = 'gzip')
b = x.read() b = x.read()
assert b == b'ABC' assert b == b'ABC'
x.read_next_member() x.read_next_member()
assert x.read() == b'' assert x.read() == b'123'
#with pytest.raises(zlib.error): #with pytest.raises(zlib.error):
# x.read() # x.read()
#error: Error -3 while decompressing: incorrect header check #error: Error -3 while decompressing: incorrect header check
# Errors
def test_err_chunk_cut_off(): def test_err_chunk_cut_off():
# Chunked data cut off with exceptions # Chunked data cut off with exceptions
c = ChunkedDataReader(BytesIO(b"4\r\n1234\r\n4\r\n12"), raise_exceptions=True) c = ChunkedDataReader(BytesIO(b"4\r\n1234\r\n4\r\n12"), raise_exceptions=True)