fixing _get_leader_for_partition unittests

Author: mrtheb

test/test_unit.py

@@ -417,8 +417,8 @@ def test_load_metadata(self, protocol, conn):
 
         # 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 @@ def test_get_leader_for_partitions_reloads_metadata(self, protocol, conn):
         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 @@ def test_get_leader_for_partitions_reloads_metadata(self, protocol, conn):
         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 @@ def test_get_leader_for_unassigned_partitions(self, protocol, conn):
 
         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 @@ def test_get_leader_returns_none_when_noleader(self, protocol, conn):
         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