diff options
| author | Dana Powers <dana.powers@gmail.com> | 2016-03-12 23:51:34 -0800 |
|---|---|---|
| committer | Dana Powers <dana.powers@gmail.com> | 2016-03-13 09:35:49 -0700 |
| commit | 5d28581b40c654d80b282e489c7149e72cf1b954 (patch) | |
| tree | b7e01b65f532c51c03108733c7816d5ab91734f8 /test/test_producer.py | |
| parent | f6af0fac6b116a21ad3514991cade86f0e6c8086 (diff) | |
| download | kafka-python-5d28581b40c654d80b282e489c7149e72cf1b954.tar.gz | |
Use producer retries and flush timeout in producer end-to-end test
Diffstat (limited to 'test/test_producer.py')
| -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]) |