openstack/deb-python-kafka
Code Issues Proposed changes

Update lz4 codec tests

Browse Source
This commit is contained in:
Dana Powers
2016-05-22 13:31:23 -07:00
parent 4cbc1e3374
commit 355859ba16
1 changed files with 23 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
test/test_codec.py
View File

@@ -8,6 +8,7 @@ from kafka.codec import (
gzip_encode, gzip_decode, gzip_encode, gzip_decode,
snappy_encode, snappy_decode, snappy_encode, snappy_decode,
lz4_encode, lz4_decode, lz4_encode, lz4_decode,
lz4_encode_old_kafka, lz4_decode_old_kafka,
) )
from test.testutil import random_string from test.testutil import random_string
@@ -84,4 +85,26 @@ def test_lz4():
for i in xrange(1000): for i in xrange(1000):
b1 = random_string(100).encode('utf-8') b1 = random_string(100).encode('utf-8')
b2 = lz4_decode(lz4_encode(b1)) 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 assert b1 == b2