From a5a3f179afeaf3a4eaa80bcb194dddbedfdd788f Mon Sep 17 00:00:00 2001 From: Roman Date: Thu, 16 Jan 2025 19:52:36 +0800 Subject: [PATCH] test: setup main nodes fixture --- src/steps/common.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/src/steps/common.py b/src/steps/common.py index 91fdea5..caf8fa9 100644 --- a/src/steps/common.py +++ b/src/steps/common.py @@ -2,7 +2,7 @@ import inspect import pytest -from src.env_vars import NODE_1, NODE_2 +from src.env_vars import NODE_1, NODE_2, CFGSYNC, NOMOS, NOMOS_EXECUTOR from src.libs.custom_logger import get_custom_logger from src.node.nomos_node import NomosNode @@ -13,8 +13,17 @@ class StepsCommon: @pytest.fixture(scope="function") def setup_main_nodes(self, request): logger.debug(f"Running fixture setup: {inspect.currentframe().f_code.co_name}") - self.node1 = NomosNode(NODE_1, f"node1_{request.cls.test_id}") + self.node1 = NomosNode(CFGSYNC, "cfgsync") + self.node2 = NomosNode(NOMOS, "nomos_node_0") + self.node3 = NomosNode(NOMOS_EXECUTOR, "nomos_node_1") self.node1.start() - self.node2 = NomosNode(NODE_2, f"node2_{request.cls.test_id}") self.node2.start() - self.main_nodes.extend([self.node1, self.node2]) + self.node3.start() + self.main_nodes.extend([self.node1, self.node2, self.node3]) + + try: + self.node2.ensure_ready() + self.node3.ensure_ready() + except Exception as ex: + logger.error(f"REST service did not become ready in time: {ex}") + raise