Fixup BrokerConnection check_version strict error checking: 0.9 will fail with CorrelationIdError

dpkp · dpkp · commit 254dcacb28f6 · 2016-05-17T07:24:00.000-07:00
diff --git a/kafka/conn.py b/kafka/conn.py
@@ -598,6 +598,13 @@ def connect():
                 # get a basic Request Timeout. This is not ideal, but we'll deal
                 if isinstance(f.exception, Errors.RequestTimedOutError):
                     pass
+
+                # 0.9 brokers do not close the socket on unrecognized api
+                # requests (bug...). In this case we expect to see a correlation
+                # id mismatch
+                elif (isinstance(f.exception, Errors.CorrelationIdError) and
+                      version == '0.10'):
+                    pass
                 elif six.PY2:
                     assert isinstance(f.exception.args[0], socket.error)
                     assert f.exception.args[0].errno in (32, 54, 104)
