Added hability to select benchmark by glob pattern (#300)

Author: filipecosta90

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "redisbench-admin"
-version = "0.7.1"
+version = "0.7.2"
 description = "Redis benchmark run helper. A wrapper around Redis and Redis Modules benchmark tools ( ftsb_redisearch, memtier_benchmark, redis-benchmark, aibench, etc... )."
 authors = ["filipecosta90 <[email protected]>","Redis Performance Group <[email protected]>"]
 readme = "README.md"

redisbench_admin/run/args.py

@@ -23,6 +23,7 @@
 TRIGGERING_ENV = os.getenv("TRIGGERING_ENV", DEFAULT_TRIGGERING_ENV)
 ENV = os.getenv("ENV", "oss-standalone,oss-cluster")
 SETUP = os.getenv("SETUP", "")
+BENCHMARK_GLOB = os.getenv("BENCHMARK_GLOB", "*.yml")
 S3_BUCKET_NAME = os.getenv("S3_BUCKET_NAME", "ci.benchmarks.redislabs")
 PUSH_S3 = bool(os.getenv("PUSH_S3", False))
 FAIL_FAST = bool(os.getenv("FAIL_FAST", False))
@@ -71,6 +72,13 @@ def common_run_args(parser):
         default="",
         help="specify a test to run. By default will run all of them.",
     )
+    parser.add_argument(
+        "--test-glob",
+        type=str,
+        default=BENCHMARK_GLOB,
+        help="specify a test glob pattern to use on the tests directory. by default uses *.yml. If --test is defined this options has no effect.",
+    )
+
     parser.add_argument(
         "--defaults_filename",
         type=str,

redisbench_admin/utils/benchmark_config.py

@@ -444,7 +444,7 @@ def min_ver_check(
 def get_testfiles_to_process(args):
     defaults_filename = args.defaults_filename
     if args.test == "":
-        files = pathlib.Path().glob("*.yml")
+        files = pathlib.Path().glob(args.test_glob)
         files = [str(x) for x in files]
         if defaults_filename in files:
             files.remove(defaults_filename)