Use version-indexed lists for request/response protocol structsprotocol_versions

1 files changed, 10 insertions, 10 deletions

diff --git a/test/test_client.py b/test/test_client.py
index 42d7dbd..38235fd 100644
--- a/test/test_client.py
+++ b/test/test_client.py
@@ -137,7 +137,7 @@ class TestSimpleClient(unittest.TestCase):
                 (NO_ERROR, 2, 0, [0, 1], [0, 1])
             ])
         ]
-        protocol.decode_metadata_response.return_value = MetadataResponse(brokers, topics)
+        protocol.decode_metadata_response.return_value = MetadataResponse[0](brokers, topics)
 
         # client loads metadata at init
         client = SimpleClient(hosts=['broker_1:4567'])
@@ -179,7 +179,7 @@ class TestSimpleClient(unittest.TestCase):
                 (NO_LEADER, 1, -1, [], []),
             ]),
         ]
-        protocol.decode_metadata_response.return_value = MetadataResponse(brokers, topics)
+        protocol.decode_metadata_response.return_value = MetadataResponse[0](brokers, topics)
 
         client = SimpleClient(hosts=['broker_1:4567'])
 
@@ -209,7 +209,7 @@ class TestSimpleClient(unittest.TestCase):
                 (NO_LEADER, 1, -1, [], []),
             ]),
         ]
-        decode_metadata_response.return_value = MetadataResponse(brokers, topics)
+        decode_metadata_response.return_value = MetadataResponse[0](brokers, topics)
 
         client = SimpleClient(hosts=['broker_1:4567'])
 
@@ -237,7 +237,7 @@ class TestSimpleClient(unittest.TestCase):
         topics = [
             (NO_LEADER, 'topic_no_partitions', [])
         ]
-        protocol.decode_metadata_response.return_value = MetadataResponse(brokers, topics)
+        protocol.decode_metadata_response.return_value = MetadataResponse[0](brokers, topics)
 
         client = SimpleClient(hosts=['broker_1:4567'])
 
@@ -249,7 +249,7 @@ class TestSimpleClient(unittest.TestCase):
                 (NO_ERROR, 0, 0, [0, 1], [0, 1])
             ])
         ]
-        protocol.decode_metadata_response.return_value = MetadataResponse(brokers, topics)
+        protocol.decode_metadata_response.return_value = MetadataResponse[0](brokers, topics)
 
         # calling _get_leader_for_partition (from any broker aware request)
         # will try loading metadata again for the same topic
@@ -275,7 +275,7 @@ class TestSimpleClient(unittest.TestCase):
             (NO_LEADER, 'topic_no_partitions', []),
             (UNKNOWN_TOPIC_OR_PARTITION, 'topic_unknown', []),
         ]
-        protocol.decode_metadata_response.return_value = MetadataResponse(brokers, topics)
+        protocol.decode_metadata_response.return_value = MetadataResponse[0](brokers, topics)
 
         client = SimpleClient(hosts=['broker_1:4567'])
 
@@ -304,7 +304,7 @@ class TestSimpleClient(unittest.TestCase):
                 (NO_LEADER, 1, -1, [], []),
             ]),
         ]
-        protocol.decode_metadata_response.return_value = MetadataResponse(brokers, topics)
+        protocol.decode_metadata_response.return_value = MetadataResponse[0](brokers, topics)
 
         client = SimpleClient(hosts=['broker_1:4567'])
         self.assertDictEqual(
@@ -330,7 +330,7 @@ class TestSimpleClient(unittest.TestCase):
                 (NO_ERROR, 1, 1, [1, 0], [1, 0])
             ]),
         ]
-        protocol.decode_metadata_response.return_value = MetadataResponse(brokers, topics)
+        protocol.decode_metadata_response.return_value = MetadataResponse[0](brokers, topics)
         self.assertEqual(brokers[0], client._get_leader_for_partition('topic_noleader', 0))
         self.assertEqual(brokers[1], client._get_leader_for_partition('topic_noleader', 1))
 
@@ -350,7 +350,7 @@ class TestSimpleClient(unittest.TestCase):
                 (NO_LEADER, 1, -1, [], []),
             ]),
         ]
-        protocol.decode_metadata_response.return_value = MetadataResponse(brokers, topics)
+        protocol.decode_metadata_response.return_value = MetadataResponse[0](brokers, topics)
 
         client = SimpleClient(hosts=['broker_1:4567'])
 
@@ -375,7 +375,7 @@ class TestSimpleClient(unittest.TestCase):
         topics = [
             (UNKNOWN_TOPIC_OR_PARTITION, 'topic_doesnt_exist', []),
         ]
-        protocol.decode_metadata_response.return_value = MetadataResponse(brokers, topics)
+        protocol.decode_metadata_response.return_value = MetadataResponse[0](brokers, topics)
 
         client = SimpleClient(hosts=['broker_1:4567'])