diff --git a/tests/basic/basic.py b/tests/basic/basic.py index f4d7f0793efab6938b621306fe99cedef3cfd2f0..4bae2468792a2435eb5e79fef1ef77338a967f86 100644 --- a/tests/basic/basic.py +++ b/tests/basic/basic.py @@ -9,7 +9,7 @@ from ...utils.kvs_api import DEFAULT_TIMEOUT def basic_kv_1(conductor: ClusterConductor, dir, log: Logger): with KVSTestFixture(conductor, dir, log, node_count=2) as fx: c = KVSMultiClient(fx.clients, "client", log) - view = {'shard1': [conductor.get_nodes()[0].get_view()], 'shard2': [conductor.get_nodes()[1].get_view()]} + view = {'shard1': [conductor.get_node(0).get_view()], 'shard2': [conductor.get_node(1).get_view()]} fx.broadcast_view(view) r = c.put(0, "x", "1") diff --git a/utils/containers.py b/utils/containers.py index abba8fd73451e72fd5a4b958817a7be52b4998ea..3e2e3620bbbc1875a34382eba4b1fcc4505c425f 100644 --- a/utils/containers.py +++ b/utils/containers.py @@ -468,7 +468,9 @@ class ClusterConductor: def get_nodes(self): return self.nodes - + + def get_node(self, index): + return self.nodes[index] def get_partition_view(self, partition_id: str): net_name = f"kvs_{self.group_id}_net_{partition_id}"