GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-05-13 08:29:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

change: add argument to specify yaml/json testcase file for locusts

Browse Source
This commit is contained in:
debugtalk
2020-04-13 00:42:55 +08:00
parent cea7f33f27
commit fc0a10a193
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
httprunner/ext/locusts/__init__.py
View File

@@ -16,6 +16,8 @@ def init_parser_locusts(subparsers):
sub_parser_locusts.add_argument(
'--locust-help', action='store_true', default=False,
help="Show locust help.")
sub_parser_locusts.add_argument('test_file', nargs='?',
help="Specify YAML/JSON testcase file.")
sub_parser_locusts.add_argument(
"--master", action='store_true', default=False, help="Start locust master.")
sub_parser_locusts.add_argument(
@@ -57,16 +59,14 @@ def main_locusts(args, extra_args):
logger.remove()
logger.add(sys.stdout, level=loglevel)
# convert httprunner yaml/json case to locustfile.py
try:
testcase_index = get_arg_index("-f", "--locustfile")
assert testcase_index and testcase_index < len(sys.argv)
testcase_file_path = sys.argv[testcase_index]
sys.argv[testcase_index] = parse_locustfile(testcase_file_path)
except AssertionError:
print("Testcase file is not specified, exit.")
if not args.test_file:
logger.error("Testcase file is not specified, exit.")
sys.exit(1)
# convert httprunner yaml/json case to locustfile.py
locustfile_path = parse_locustfile(args.test_file)
sys.argv.extend(["-f", locustfile_path])
manager = multiprocessing.Manager()
try:
if args.quickstart: