Fix test_consumer generator/list mock, reenable tests

1 files changed, 2 insertions, 5 deletions

diff --git a/test/test_consumer.py b/test/test_consumer.py
index 0cab116..31b7e72 100644
--- a/test/test_consumer.py
+++ b/test/test_consumer.py
@@ -30,7 +30,6 @@ class TestMultiProcessConsumer(unittest.TestCase):
         self.assertEqual(client.get_partition_ids_for_topic.call_count, 0) # pylint: disable=no-member
 
 class TestSimpleConsumer(unittest.TestCase):
-    @unittest.skip
     def test_simple_consumer_failed_payloads(self):
         client = MagicMock()
         consumer = SimpleConsumer(client, group=None,
@@ -45,7 +44,6 @@ class TestSimpleConsumer(unittest.TestCase):
         # This should not raise an exception
         consumer.get_messages(5)
 
-    @unittest.skip
     def test_simple_consumer_leader_change(self):
         client = MagicMock()
         consumer = SimpleConsumer(client, group=None,
@@ -66,7 +64,6 @@ class TestSimpleConsumer(unittest.TestCase):
         self.assertGreaterEqual(client.reset_topic_metadata.call_count, 1)
         self.assertGreaterEqual(client.load_metadata_for_topics.call_count, 1)
 
-    @unittest.skip
     def test_simple_consumer_unknown_topic_partition(self):
         client = MagicMock()
         consumer = SimpleConsumer(client, group=None,
@@ -129,10 +126,10 @@ class TestSimpleConsumer(unittest.TestCase):
         def fail_requests(payloads, **kwargs):
             responses = [
                 FetchResponsePayload(payloads[0].topic, payloads[0].partition, 0, 0,
-                              (OffsetAndMessage(
+                              [OffsetAndMessage(
                                   payloads[0].offset + i,
                                   "msg %d" % (payloads[0].offset + i))
-                               for i in range(10))),
+                               for i in range(10)]),
             ]
             for failure in payloads[1:]:
                 responses.append(error_factory(failure))