From b8da199b54b26728f302aaccc5dd0433a8671759 Mon Sep 17 00:00:00 2001
From: Hannu Valtonen <hannu.valtonen@aiven.io>
Date: Tue, 20 Jun 2017 03:09:43 +0300
Subject: producer: Set exit timeout to 0 for atexit handler to match __del__
 (#1126)

Hit a problem with pytest hitting the atexit handler and waiting
for close() timeout forever at teardown.

This commit makes atexit close() equivalent to __del__ behavior,
namely using timeout of 0 for close() completion. If you need a
longer timeout you should be setting it explicitly.
---
 kafka/producer/kafka.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/kafka/producer/kafka.py b/kafka/producer/kafka.py
index 1f78c89..bd891c9 100644
--- a/kafka/producer/kafka.py
+++ b/kafka/producer/kafka.py
@@ -371,7 +371,7 @@ class KafkaProducer(object):
         _self = weakref.proxy(self)
         def wrapper():
             try:
-                _self.close()
+                _self.close(timeout=0)
             except (ReferenceError, AttributeError):
                 pass
         return wrapper
-- 
cgit v1.2.1