Merge pull request #27 from aiven/keejon/bugfix-check-version

bugfix: catch KeyError in check_version and retry

Author: aiven-anton

kafka/client_async.py

@@ -912,7 +912,13 @@ def check_version(self, node_id=None, timeout=2, strict=False):
                 self._lock.release()
                 raise Errors.NoBrokersAvailable()
             self._maybe_connect(try_node)
-            conn = self._conns[try_node]
+            try:
+                conn = self._conns[try_node]
+            except KeyError:
+                if node_id is not None:
+                    self._lock.release()
+                    raise Errors.NodeNotReadyError()
+                continue
 
             # We will intentionally cause socket failures
             # These should not trigger metadata refresh