fix: copy appropriate config for desired cluster

This commit is contained in:
Roman 2025-01-22 15:19:18 +08:00
parent 82abae7a2d
commit 1b0731f2ad
No known key found for this signature in database
GPG Key ID: B8FE070B54E11B75
2 changed files with 13 additions and 3 deletions

View File

@ -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")

View File

@ -1,5 +1,3 @@
import json
import allure
from src.env_vars import NOMOS_EXECUTOR