From 6ff8510d616b63e3d2bd7b00801522f99bc730df Mon Sep 17 00:00:00 2001 From: Roman Date: Wed, 5 Mar 2025 04:03:27 +0000 Subject: [PATCH] test: client node run --- src/cli/nomos_cli.py | 16 +++++++++------- src/steps/common.py | 4 ++-- tests/dos_robustness/test_high_load_dos.py | 2 ++ 3 files changed, 13 insertions(+), 9 deletions(-) diff --git a/src/cli/nomos_cli.py b/src/cli/nomos_cli.py index 75f4793..36286b0 100644 --- a/src/cli/nomos_cli.py +++ b/src/cli/nomos_cli.py @@ -44,12 +44,15 @@ class NomosCli: self._port_map = {} - cmd = [NOMOS_CLI, self._command] - for flag in nomos_cli[self._command]["flags"]: - for f, indexes in flag.items(): - cmd.append(f) - for j in indexes: - cmd.append(input_values[j]) + if self._command == "client_node": + cmd = ["tail", "-f", "/dev/null"] + else: + cmd = [NOMOS_CLI, self._command] + for flag in nomos_cli[self._command]["flags"]: + for f, indexes in flag.items(): + cmd.append(f) + for j in indexes: + cmd.append(input_values[j]) logger.debug(f"NomosCli command to run {cmd}") @@ -72,7 +75,6 @@ class NomosCli: decode_only = kwargs.get("decode_only", False) return self.reconstruct(decode_only=decode_only) case "client_node": - delay(3600) return None case _: return None diff --git a/src/steps/common.py b/src/steps/common.py index 8842848..2ca8f56 100644 --- a/src/steps/common.py +++ b/src/steps/common.py @@ -44,7 +44,7 @@ class StepsCommon: def cluster_setup(self): logger.debug(f"Running fixture setup: {inspect.currentframe().f_code.co_name}") self.main_nodes = [] - self.cli_nodes = [] + self.client_nodes = [] @pytest.fixture(scope="function") def setup_2_node_cluster(self, request): @@ -101,6 +101,6 @@ class StepsCommon: for i in range(num_clients): cli_node = NomosCli(command="client_node") - self.cli_nodes.append(cli_node) + self.client_nodes.append(cli_node) delay(1) diff --git a/tests/dos_robustness/test_high_load_dos.py b/tests/dos_robustness/test_high_load_dos.py index 225bb8b..ab5f258 100644 --- a/tests/dos_robustness/test_high_load_dos.py +++ b/tests/dos_robustness/test_high_load_dos.py @@ -121,6 +121,8 @@ class TestHighLoadDos(StepsDataAvailability): successful_downloads = 0 unsuccessful_downloads = 0 + self.client_nodes[0].run() + while True: if time.time() - start_time > timeout: break