Merge pull request confluentinc#109 from t0mk/master

ewencp · web-flow · commit 0af5127f8c54 · 2017-01-23T15:51:10.000-08:00
Fix handling of None messages in AvroConsumer
diff --git a/confluent_kafka/avro/__init__.py b/confluent_kafka/avro/__init__.py
@@ -148,6 +148,8 @@ def poll(self, timeout=None):
         if timeout is None:
             timeout = -1
         message = super(AvroConsumer, self).poll(timeout)
+        if message is None:
+            return None
         if not message.value() and not message.key():
             return message
         if not message.error():
