diff options
| -rw-r--r-- | test/test_producer.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/test/test_producer.py b/test/test_producer.py index 36da68d..829c6f2 100644 --- a/test/test_producer.py +++ b/test/test_producer.py @@ -21,6 +21,7 @@ def test_end_to_end(kafka_broker, compression): connect_str = 'localhost:' + str(kafka_broker.port) producer = KafkaProducer(bootstrap_servers=connect_str, + retries=5, max_block_ms=10000, compression_type=compression, value_serializer=str.encode) @@ -34,7 +35,7 @@ def test_end_to_end(kafka_broker, compression): for i in range(1000): producer.send(topic, 'msg %d' % i) - producer.flush() + producer.flush(timeout=30) producer.close() consumer.subscribe([topic]) |