From 66284e57accec5977d606fc91a0b28177b352eb4 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Tue, 26 Jan 2016 20:34:13 -0800 Subject: Add end-to-end integration testing for all compression types --- test/test_producer.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) 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, -- cgit v1.2.1