diff options
| author | mrtheb <mrlabbe@gmail.com> | 2014-02-15 22:05:11 -0500 |
|---|---|---|
| committer | mrtheb <mrlabbe@gmail.com> | 2014-02-15 22:05:11 -0500 |
| commit | 828133cff064f4f8fba753183ac21619355ac005 (patch) | |
| tree | e4dec0a369cab032bc6b0b58e1b32e19dd484ea4 /test | |
| parent | 5e5d7097a70e79d90d573aef6a0adba40ada5f03 (diff) | |
| download | kafka-python-828133cff064f4f8fba753183ac21619355ac005.tar.gz | |
fixing _get_leader_for_partition unittests
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_unit.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/test/test_unit.py b/test/test_unit.py index 1439d8b..0b2b339 100644 --- a/test/test_unit.py +++ b/test/test_unit.py @@ -417,8 +417,8 @@ class TestKafkaClient(unittest.TestCase): # client loads metadata at init client = KafkaClient(host='broker_1', port=4567) - self.assertItemsEqual({ - TopicAndPartition('topic_1', 0): brokers[0], + self.assertDictEqual({ + TopicAndPartition('topic_1', 0): brokers[1], TopicAndPartition('topic_noleader', 0): None, TopicAndPartition('topic_noleader', 1): None, TopicAndPartition('topic_3', 0): brokers[0], @@ -443,7 +443,7 @@ class TestKafkaClient(unittest.TestCase): client = KafkaClient(host='broker_1', port=4567) # topic metadata is loaded but empty - self.assertItemsEqual({}, client.topics_to_brokers) + self.assertDictEqual({}, client.topics_to_brokers) topics['topic_no_partitions'] = { 0: PartitionMetadata('topic_no_partitions', 0, 0, [0, 1], [0, 1]) @@ -455,7 +455,7 @@ class TestKafkaClient(unittest.TestCase): leader = client._get_leader_for_partition('topic_no_partitions', 0) self.assertEqual(brokers[0], leader) - self.assertItemsEqual({ + self.assertDictEqual({ TopicAndPartition('topic_no_partitions', 0): brokers[0]}, client.topics_to_brokers) @@ -475,7 +475,7 @@ class TestKafkaClient(unittest.TestCase): client = KafkaClient(host='broker_1', port=4567) - self.assertItemsEqual({}, client.topics_to_brokers) + self.assertDictEqual({}, client.topics_to_brokers) self.assertRaises( PartitionUnavailableError, client._get_leader_for_partition, @@ -500,7 +500,7 @@ class TestKafkaClient(unittest.TestCase): protocol.decode_metadata_response.return_value = (brokers, topics) client = KafkaClient(host='broker_1', port=4567) - self.assertItemsEqual( + self.assertDictEqual( { TopicAndPartition('topic_noleader', 0): None, TopicAndPartition('topic_noleader', 1): None |