Return connection state explicitly after close in connect() (#1778)

Author: dpkp

kafka/conn.py

@@ -354,7 +354,7 @@ def connect(self):
             next_lookup = self._next_afi_sockaddr()
             if not next_lookup:
                 self.close(Errors.KafkaConnectionError('DNS failure'))
-                return
+                return self.state
             else:
                 log.debug('%s: creating new socket', self)
                 self._sock_afi, self._sock_addr = next_lookup
@@ -409,6 +409,7 @@ def connect(self):
                           ' Disconnecting.', self, ret)
                 errstr = errno.errorcode.get(ret, 'UNKNOWN')
                 self.close(Errors.KafkaConnectionError('{} {}'.format(ret, errstr)))
+                return self.state
 
             # Needs retry
             else:
@@ -443,6 +444,7 @@ def connect(self):
             if time.time() > request_timeout + self.last_attempt:
                 log.error('Connection attempt to %s timed out', self)
                 self.close(Errors.KafkaConnectionError('timeout'))
+                return self.state
 
         return self.state