From db49c8fbdd657b652d1c3715044e6b10af6ec77b Mon Sep 17 00:00:00 2001
From: Dana Powers <dana.powers@gmail.com>
Date: Sat, 17 Sep 2016 18:43:38 -0700
Subject: Fixup iterators

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

(limited to 'kafka/consumer/fetcher.py')

diff --git a/kafka/consumer/fetcher.py b/kafka/consumer/fetcher.py
index 62190fb..15fa1c9 100644
--- a/kafka/consumer/fetcher.py
+++ b/kafka/consumer/fetcher.py
@@ -448,10 +448,10 @@ class Fetcher(six.Iterator):
         return self
 
     def __next__(self):
-        ret = self.fetched_records(max_records=1)
+        ret, _ = self.fetched_records(max_records=1)
         if not ret:
             raise StopIteration
-        assert len(ret.keys()) == 1
+        assert len(ret) == 1
         (messages,) = ret.values()
         assert len(messages) == 1
         return messages[0]
-- 
cgit v1.2.1