diff options
Diffstat (limited to 'test/test_producer.py')
-rw-r--r-- | test/test_producer.py | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/test/test_producer.py b/test/test_producer.py index 263df11..36da68d 100644 --- a/test/test_producer.py +++ b/test/test_producer.py @@ -1,3 +1,5 @@ +import sys + import pytest from kafka import KafkaConsumer, KafkaProducer @@ -6,10 +8,21 @@ from test.testutil import random_string @pytest.mark.skipif(not version(), reason="No KAFKA_VERSION set") -def test_end_to_end(kafka_broker): +@pytest.mark.parametrize("compression", [None, 'gzip', 'snappy', 'lz4']) +def test_end_to_end(kafka_broker, compression): + + if compression == 'lz4': + # LZ4 requires 0.8.2 + if version() < (0, 8, 2): + return + # LZ4 python libs dont work on python2.6 + elif sys.version_info < (2, 7): + return + connect_str = 'localhost:' + str(kafka_broker.port) producer = KafkaProducer(bootstrap_servers=connect_str, max_block_ms=10000, + compression_type=compression, value_serializer=str.encode) consumer = KafkaConsumer(bootstrap_servers=connect_str, group_id=None, |