Validate max_records in KafkaConsumer.pollmax_records_int

1 files changed, 2 insertions, 0 deletions

diff --git a/kafka/consumer/group.py b/kafka/consumer/group.py
index debbd2d..110df55 100644
--- a/kafka/consumer/group.py
+++ b/kafka/consumer/group.py
@@ -595,6 +595,8 @@ class KafkaConsumer(six.Iterator):
         assert timeout_ms >= 0, 'Timeout must not be negative'
         if max_records is None:
             max_records = self.config['max_poll_records']
+        assert isinstance(max_records, int), 'max_records must be an integer'
+        assert max_records > 0, 'max_records must be positive'
 
         # Poll for new data until the timeout expires
         start = time.time()