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

1 files changed, 7 insertions, 0 deletions

diff --git a/kafka/conn.py b/kafka/conn.py
index 8f0539b..db56dda 100644
--- a/kafka/conn.py
+++ b/kafka/conn.py
@@ -598,6 +598,13 @@ class BrokerConnection(object):
                 # 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)