Cluster mode cannot be specified when importing

YaacovHazan · Nov 17, 2023 · 567348c · 567348c
1 parent 55ca438
commit 567348c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/memtier_benchmark.cpp b/memtier_benchmark.cpp
@@ -1484,6 +1484,11 @@ int main(int argc, char *argv[])
         }
         assert(obj_gen != NULL);
     } else {
+        // oss cluster API can't be enabled
+        if (cfg.cluster_mode) {
+            fprintf(stderr, "error: Cluster mode cannot be specified when importing.\n");
+            exit(1);
+        }
         // check paramters
         if (cfg.data_size ||
             cfg.data_size_list.is_defined() ||

diff --git a/tests/tests_oss_simple_flow.py b/tests/tests_oss_simple_flow.py
@@ -420,6 +420,7 @@ def test_default_set_get_rate_limited(env):
             assert_minimum_memtier_outcomes(config, env, memtier_ok, overall_expected_request_count, overall_request_count, request_delta)
 
 def test_data_import(env):
+    env.skipOnCluster()
     benchmark_specs = {"name": env.testName, "args": [f"--data-import={ROOT_FOLDER}/tests/data-import-2-keys.txt",'--ratio=1:1']}
     addTLSArgs(benchmark_specs, env)
     config = get_default_memtier_config()
@@ -451,6 +452,7 @@ def test_data_import(env):
 
 
 def test_data_import_setex(env):
+    env.skipOnCluster()
     benchmark_specs = {"name": env.testName, "args": [f"--data-import={ROOT_FOLDER}/tests/data-import-2-keys-expiration.txt",'--ratio=1:1']}
     addTLSArgs(benchmark_specs, env)
     config = get_default_memtier_config()