Add end-to-end integration testing for all compression types

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,