diff options
| author | Dana Powers <dana.powers@rd.io> | 2015-06-10 09:38:24 -0700 |
|---|---|---|
| committer | Dana Powers <dana.powers@rd.io> | 2015-06-10 10:15:45 -0700 |
| commit | e60de943175c45d64602c0388b502b9763b34f27 (patch) | |
| tree | debf0237b8e909d0cd06f3d2aa7190c9a5c8f401 | |
| parent | 043e9bb04f1149ab8f9f8942ee591227f659128d (diff) | |
| download | kafka-python-e60de943175c45d64602c0388b502b9763b34f27.tar.gz | |
Split consumer test class between Simple and MultiProcess
| -rw-r--r-- | test/test_consumer.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/test/test_consumer.py b/test/test_consumer.py index 08fd620..bbced02 100644 --- a/test/test_consumer.py +++ b/test/test_consumer.py @@ -25,10 +25,11 @@ class TestMultiProcessConsumer(unittest.TestCase): client = MagicMock() partitions = (0,) with patch.object(MultiProcessConsumer, 'fetch_last_known_offsets') as fetch_last_known_offsets: - consumer = MultiProcessConsumer(client, 'testing-group', 'testing-topic', partitions=partitions) + MultiProcessConsumer(client, 'testing-group', 'testing-topic', partitions=partitions) self.assertEqual(fetch_last_known_offsets.call_args[0], (partitions,) ) self.assertEqual(client.get_partition_ids_for_topic.call_count, 0) # pylint: disable=no-member +class TestSimpleConsumer(unittest.TestCase): def test_simple_consumer_failed_payloads(self): client = MagicMock() consumer = SimpleConsumer(client, group=None, |