diff --git a/src/steps/common.py b/src/steps/common.py index 78a905b..fd59c10 100644 --- a/src/steps/common.py +++ b/src/steps/common.py @@ -1,14 +1,24 @@ import inspect +import os +import shutil import pytest -from src.env_vars import NODE_1, NODE_2, CFGSYNC, NOMOS, NOMOS_EXECUTOR +from src.env_vars import CFGSYNC, NOMOS, NOMOS_EXECUTOR from src.libs.custom_logger import get_custom_logger from src.node.nomos_node import NomosNode logger = get_custom_logger(__name__) +def prepare_cluster_config(node_count): + cwd = os.getcwd() + config_dir = "cluster_config" + src = f"{cwd}/{config_dir}/cfgsync-{node_count}node.yaml" + dst = f"{cwd}/{config_dir}/cfgsync.yaml" + shutil.copyfile(src, dst) + + class StepsCommon: @pytest.fixture(scope="function", autouse=True) def cluster_setup(self): @@ -18,6 +28,7 @@ class StepsCommon: @pytest.fixture(scope="function") def setup_2_node_cluster(self, request): logger.debug(f"Running fixture setup: {inspect.currentframe().f_code.co_name}") + prepare_cluster_config(2) self.node1 = NomosNode(CFGSYNC, "cfgsync") self.node2 = NomosNode(NOMOS, "nomos_node_0") self.node3 = NomosNode(NOMOS_EXECUTOR, "nomos_node_1") @@ -36,6 +47,7 @@ class StepsCommon: @pytest.fixture(scope="function") def setup_5_node_cluster(self, request): logger.debug(f"Running fixture setup: {inspect.currentframe().f_code.co_name}") + prepare_cluster_config(5) self.node1 = NomosNode(CFGSYNC, "cfgsync") self.node2 = NomosNode(NOMOS, "nomos_node_0") self.node3 = NomosNode(NOMOS, "nomos_node_1") diff --git a/src/steps/da.py b/src/steps/da.py index 9b2e91e..bbf1379 100644 --- a/src/steps/da.py +++ b/src/steps/da.py @@ -1,5 +1,3 @@ -import json - import allure from src.env_vars import NOMOS_EXECUTOR