From bbab7aadb62517464420abfd1eed771af68fd7be Mon Sep 17 00:00:00 2001
From: Dana Powers <dana.powers@gmail.com>
Date: Sat, 14 Oct 2017 17:21:36 -0700
Subject: _sasl_auth_future can be reset if recv() causes disconnect

---
 kafka/conn.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/kafka/conn.py b/kafka/conn.py
index dc15a5e..2f560d0 100644
--- a/kafka/conn.py
+++ b/kafka/conn.py
@@ -444,7 +444,10 @@ class BrokerConnection(object):
             sasl_response.add_errback(lambda f, e: f.failure(e), future)
             self._sasl_auth_future = future
         self.recv()
-        if self._sasl_auth_future.failed():
+        # A connection error could trigger close() which will reset the future
+        if self._sasl_auth_future is None:
+            return False
+        elif self._sasl_auth_future.failed():
             ex = self._sasl_auth_future.exception
             if not isinstance(ex, Errors.ConnectionError):
                 raise ex  # pylint: disable-msg=raising-bad-type
-- 
cgit v1.2.1