diff options
author | Dana Powers <dana.powers@gmail.com> | 2016-05-22 17:14:03 -0700 |
---|---|---|
committer | Dana Powers <dana.powers@gmail.com> | 2016-05-22 17:14:03 -0700 |
commit | 1d4251a9efa4c5466ba5095f3ba199bf082a72b5 (patch) | |
tree | 0393a7b7768f449174216fb6b0702a87844a5a81 /test/test_codec.py | |
parent | 96530f6a9c4a31d23b069ba162dba6cf45a5efd0 (diff) | |
download | kafka-python-1d4251a9efa4c5466ba5095f3ba199bf082a72b5.tar.gz |
Use standard LZ4 framing for v1 messages / kafka 0.10 (#695)
* LZ4 framing fixed in 0.10 / message v1 -- retain broken lz4 code for compatibility
* lz4f does not support easy incremental decompression - raise RuntimeError
* Update lz4 codec tests
Diffstat (limited to 'test/test_codec.py')
-rw-r--r-- | test/test_codec.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/test/test_codec.py b/test/test_codec.py index 07a74cd..906b53c 100644 --- a/test/test_codec.py +++ b/test/test_codec.py @@ -8,6 +8,7 @@ from kafka.codec import ( gzip_encode, gzip_decode, snappy_encode, snappy_decode, lz4_encode, lz4_decode, + lz4_encode_old_kafka, lz4_decode_old_kafka, ) from test.testutil import random_string @@ -84,4 +85,26 @@ def test_lz4(): for i in xrange(1000): b1 = random_string(100).encode('utf-8') b2 = lz4_decode(lz4_encode(b1)) + assert len(b1) == len(b2) + assert b1 == b2 + + +@pytest.mark.skipif(not has_lz4(), reason="LZ4 not available") +def test_lz4_old(): + for i in xrange(1000): + b1 = random_string(100).encode('utf-8') + b2 = lz4_decode_old_kafka(lz4_encode_old_kafka(b1)) + assert len(b1) == len(b2) + assert b1 == b2 + + +@pytest.mark.xfail(reason="lz4tools library doesnt support incremental decompression") +@pytest.mark.skipif(not has_lz4(), reason="LZ4 not available") +def test_lz4_incremental(): + for i in xrange(1000): + # lz4 max single block size is 4MB + # make sure we test with multiple-blocks + b1 = random_string(100).encode('utf-8') * 50000 + b2 = lz4_decode(lz4_encode(b1)) + assert len(b1) == len(b2) assert b1 == b2 |