diff options
| author | Dana Powers <dana.powers@rd.io> | 2015-12-10 09:57:39 -0800 |
|---|---|---|
| committer | Dana Powers <dana.powers@rd.io> | 2015-12-10 09:57:39 -0800 |
| commit | c3d2fda3c368771cb93a09bb2f1edaa7a3cf9c2b (patch) | |
| tree | eaaa8184a491341a4087bf663c4415cf981347b8 | |
| parent | eea162eb0366ec15782568ae29e482814b06cc0e (diff) | |
| download | kafka-python-c3d2fda3c368771cb93a09bb2f1edaa7a3cf9c2b.tar.gz | |
Fix test_consumer generator/list mock, reenable tests
| -rw-r--r-- | test/test_consumer.py | 7 |
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)) |