diff options
| -rw-r--r-- | test/test_producer.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/test/test_producer.py b/test/test_producer.py index 263df11..7a2db68 100644 --- a/test/test_producer.py +++ b/test/test_producer.py @@ -6,10 +6,17 @@ 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): + + # LZ4 requires 0.8.2 + if compression == 'lz4' and version() < (0, 8, 2): + 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, |