diff --git a/src/steps/common.py b/src/steps/common.py index d7ba494..2af620e 100644 --- a/src/steps/common.py +++ b/src/steps/common.py @@ -118,3 +118,7 @@ class StepsCommon: default_target = [f"http://{self.main_nodes[1 + i % 2].name()}:18080"] proxy_client.run(input_values=default_target) self.client_nodes.append(proxy_client) + + @pytest.fixture(params=["setup_2_node_cluster", "setup_4_node_cluster"]) + def setup_cluster_variant(self, request): + return request.getfixturevalue(request.param) diff --git a/tests/data_confidentiality/test_data_confidentiality.py b/tests/data_confidentiality/test_data_confidentiality.py index fe57153..f3453ca 100644 --- a/tests/data_confidentiality/test_data_confidentiality.py +++ b/tests/data_confidentiality/test_data_confidentiality.py @@ -45,7 +45,7 @@ def modify_key_value(file_path, yaml_key_path): class TestDataConfidentiality(StepsDataAvailability): main_nodes = [] - @pytest.mark.usefixtures("setup_2_node_cluster") + @pytest.mark.usefixtures("setup_cluster_variant") def test_unauthorized_node_cannot_receive_dispersed_data(self): self.disperse_data(DATA_TO_DISPERSE[1], to_app_id(1), to_index(0)) delay(CONSENSUS_SLOT_TIME)