Add topology filter to runners.py

Author: fcostaoliveira

redis_benchmarks_specification/__runner__/args.py

@@ -208,4 +208,9 @@ def create_client_runner_args(project_name):
         action="store_true",
         help="Run client in cluster mode.",
     )
+    parser.add_argument(
+        "--topology",
+        default="",
+        help="Filter tests to run only with the specified topology (e.g. oss-standalone)",
+    )
     return parser

redis_benchmarks_specification/__self_contained_coordinator__/args.py

@@ -165,4 +165,10 @@ def create_self_contained_coordinator_args(project_name):
         default=100000,
         help="Run a subset of the tests based uppon a preset priority. By default runs all tests.",
     )
+    parser.add_argument(
+        "--topology",
+        type=str,
+        default="",
+        help="Filter tests to run only with the specified topology (e.g. oss-standalone)",
+    )
     return parser

redis_benchmarks_specification/__self_contained_coordinator__/runners.py

@@ -116,6 +116,7 @@ def process_self_contained_coordinator_stream(
     docker_air_gap=False,
     verbose=False,
     run_tests_with_dataset=False,
+    args=None,
 ):
     stream_id = "n/a"
     overall_result = False
@@ -197,6 +198,17 @@ def process_self_contained_coordinator_stream(
                                 )
                             )
                     for topology_spec_name in benchmark_config["redis-topologies"]:
+                        # Filter by topology if specified
+                        if (
+                            args.topology
+                            and args.topology != ""
+                            and topology_spec_name != args.topology
+                        ):
+                            logging.info(
+                                f"Skipping topology {topology_spec_name} as it doesn't match the requested topology {args.topology}"
+                            )
+                            continue
+
                         test_result = False
                         try:
                             current_cpu_pos = cpuset_start_pos

redis_benchmarks_specification/__self_contained_coordinator__/self_contained_coordinator.py

@@ -367,6 +367,13 @@ def self_contained_coordinator_blocking_read(
     if len(newTestInfo[0]) < 2 or len(newTestInfo[0][1]) < 1:
         stream_id = ">"
     else:
+        # Create args object with topology parameter
+        class Args:
+            def __init__(self):
+                self.topology = ""
+
+        args = Args()
+
         (
             stream_id,
             overall_result,
@@ -398,6 +405,7 @@ def self_contained_coordinator_blocking_read(
             default_metrics_str,
             docker_keep_env,
             restore_build_artifacts_default,
+            args,
         )
         num_process_streams = num_process_streams + 1
         num_process_test_suites = num_process_test_suites + total_test_suite_runs