Skip to content

Commit b0c9d55

Browse files
Detect and work with missing --github_actor, --github_branch, and --github_sha (#73)
1 parent 49b189e commit b0c9d55

File tree

3 files changed

+37
-3
lines changed

3 files changed

+37
-3
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[tool.poetry]
22
name = "redisbench-admin"
3-
version = "0.1.60"
3+
version = "0.1.61"
44
description = "Redis benchmark run helper. A wrapper around Redis and Redis Modules benchmark tools ( ftsb_redisearch, memtier_benchmark, redis-benchmark, aibench, etc... )."
55
authors = ["filipecosta90 <[email protected]>"]
66
readme = "README.md"

redisbench_admin/run_remote/args.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,10 +22,10 @@ def create_run_remote_arguments(parser):
2222
default="",
2323
help="specify a test to run. By default will run all of them.",
2424
)
25-
parser.add_argument("--github_actor", type=str, default=None)
25+
parser.add_argument("--github_actor", type=str, default=None, nargs="?", const="")
2626
parser.add_argument("--github_repo", type=str, default=None)
2727
parser.add_argument("--github_org", type=str, default=None)
28-
parser.add_argument("--github_sha", type=str, default=None)
28+
parser.add_argument("--github_sha", type=str, default=None, nargs="?", const="")
2929
parser.add_argument("--github_branch", type=str, default=None, nargs="?", const="")
3030
parser.add_argument("--triggering_env", type=str, default=socket.gethostname())
3131
parser.add_argument("--terraform_bin_path", type=str, default=TERRAFORM_BIN_PATH)

tests/test_args.py

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
import argparse
2+
3+
from redisbench_admin.run_local.args import create_run_local_arguments
4+
from redisbench_admin.run_remote.args import create_run_remote_arguments
5+
6+
7+
def test_create_run_remote_arguments():
8+
parser = argparse.ArgumentParser(
9+
description="test",
10+
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
11+
)
12+
parser = create_run_remote_arguments(parser)
13+
args = parser.parse_args(args=["--github_actor", "--module_path", "mymodule.so"])
14+
assert args.github_actor == ""
15+
args = parser.parse_args(args=["--github_branch", "--module_path", "mymodule.so"])
16+
assert args.github_branch == ""
17+
args = parser.parse_args(args=["--github_sha", "--module_path", "mymodule.so"])
18+
assert args.github_sha == ""
19+
args = parser.parse_args(
20+
args=["--github_actor", "gh.user", "--module_path", "mymodule.so"]
21+
)
22+
assert args.github_actor == "gh.user"
23+
24+
25+
def test_create_run_local_arguments():
26+
parser = argparse.ArgumentParser(
27+
description="test",
28+
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
29+
)
30+
parser = create_run_local_arguments(parser)
31+
args = parser.parse_args(
32+
args=["--module_path", "mymodule.so", "--test", "test1.yml"]
33+
)
34+
assert args.test == "test1.yml"

0 commit comments

Comments
 (0)