diff --git a/redis_benchmarks_specification/__runner__/args.py b/redis_benchmarks_specification/__runner__/args.py index 60a0bc77..9d2de4a7 100644 --- a/redis_benchmarks_specification/__runner__/args.py +++ b/redis_benchmarks_specification/__runner__/args.py @@ -190,4 +190,10 @@ def create_client_runner_args(project_name): type=int, help="override memtier number of runs for each benchmark. By default will run once each test", ) + parser.add_argument( + "--cluster-mode", + default=False, + action="store_true", + help="Run client in cluster mode.", + ) return parser diff --git a/redis_benchmarks_specification/__runner__/runner.py b/redis_benchmarks_specification/__runner__/runner.py index 0801fba9..e3fc4b0a 100644 --- a/redis_benchmarks_specification/__runner__/runner.py +++ b/redis_benchmarks_specification/__runner__/runner.py @@ -410,7 +410,7 @@ def process_self_contained_coordinator_stream( port = args.db_server_port host = args.db_server_host password = args.db_server_password - + oss_cluster_api_enabled = args.cluster_mode ssl_cert_reqs = "required" if tls_skip_verify: ssl_cert_reqs = None @@ -553,6 +553,7 @@ def process_self_contained_coordinator_stream( test_tls_cacert, resp_version, password, + oss_cluster_api_enabled, ) execute_init_commands( benchmark_config, r, dbconfig_keyname="dbconfig" @@ -596,6 +597,7 @@ def process_self_contained_coordinator_stream( local_benchmark_output_filename ) ) + if "memtier_benchmark" not in benchmark_tool: # prepare the benchmark command ( @@ -622,7 +624,7 @@ def process_self_contained_coordinator_stream( host, password, local_benchmark_output_filename, - False, + oss_cluster_api_enabled, tls_enabled, tls_skip_verify, test_tls_cert, @@ -970,6 +972,7 @@ def data_prepopulation_step( tls_cacert=None, resp_version=None, password=None, + oss_cluster_api_enabled=False, ): # setup the benchmark ( @@ -999,7 +1002,7 @@ def data_prepopulation_step( host, password, local_benchmark_output_filename, - False, + oss_cluster_api_enabled, tls_enabled, tls_skip_verify, tls_cert,