diff options
author | Dana Powers <dana.powers@gmail.com> | 2016-01-26 20:34:13 -0800 |
---|---|---|
committer | Dana Powers <dana.powers@gmail.com> | 2016-01-26 20:34:13 -0800 |
commit | 66284e57accec5977d606fc91a0b28177b352eb4 (patch) | |
tree | 762dd84d16b8a10ba563ec5d1465dc4afa838540 /test/test_producer.py | |
parent | 54a735bed082feecd68f15f63453e7f6ca58d547 (diff) | |
download | kafka-python-66284e57accec5977d606fc91a0b28177b352eb4.tar.gz |
Add end-to-end integration testing for all compression types
Diffstat (limited to 'test/test_producer.py')
-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, |