summaryrefslogtreecommitdiff
path: root/test/test_producer.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_producer.py')
-rw-r--r--test/test_producer.py15
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,