From c6b54036af4aebaca545898fe0d8100a15f93ee8 Mon Sep 17 00:00:00 2001
From: Dana Powers <dana.powers@gmail.com>
Date: Mon, 5 Feb 2018 14:02:24 -0800
Subject: Fix for PartitionRecords has no attribute message_idx; add test

---
 kafka/consumer/fetcher.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'kafka')

diff --git a/kafka/consumer/fetcher.py b/kafka/consumer/fetcher.py
index 6597448..c9bbb97 100644
--- a/kafka/consumer/fetcher.py
+++ b/kafka/consumer/fetcher.py
@@ -846,6 +846,10 @@ class Fetcher(six.Iterator):
             for i, msg in enumerate(messages):
                 if msg.offset == fetch_offset:
                     self.message_idx = i
+                    break
+            else:
+                self.message_idx = 0
+                self.messages = None
 
         # For truthiness evaluation we need to define __len__ or __nonzero__
         def __len__(self):
-- 
cgit v1.2.1