mirror of
https://github.com/httprunner/httprunner.py.git
synced 2026-05-08 19:03:11 +08:00
153 lines
4.0 KiB
Python
153 lines
4.0 KiB
Python
import argparse
|
|
import enum
|
|
import os
|
|
import sys
|
|
|
|
import pytest
|
|
from loguru import logger
|
|
|
|
from httprunner import __description__, __version__
|
|
from httprunner.compat import ensure_cli_args
|
|
from httprunner.make import init_make_parser, main_make
|
|
from httprunner.utils import ga4_client, init_logger, init_sentry_sdk
|
|
|
|
init_sentry_sdk()
|
|
|
|
|
|
def init_parser_run(subparsers):
|
|
sub_parser_run = subparsers.add_parser(
|
|
"run", help="Make HttpRunner testcases and run with pytest."
|
|
)
|
|
return sub_parser_run
|
|
|
|
|
|
def main_run(extra_args) -> enum.IntEnum:
|
|
ga4_client.send_event("hrun")
|
|
# keep compatibility with v2
|
|
extra_args = ensure_cli_args(extra_args)
|
|
|
|
tests_path_list = []
|
|
extra_args_new = []
|
|
for item in extra_args:
|
|
if not os.path.exists(item):
|
|
# item is not file/folder path
|
|
extra_args_new.append(item)
|
|
else:
|
|
# item is file/folder path
|
|
tests_path_list.append(item)
|
|
|
|
if len(tests_path_list) == 0:
|
|
# has not specified any testcase path
|
|
logger.error(f"No valid testcase path in cli arguments: {extra_args}")
|
|
sys.exit(1)
|
|
|
|
testcase_path_list = main_make(tests_path_list)
|
|
if not testcase_path_list:
|
|
logger.error("No valid testcases found, exit 1.")
|
|
sys.exit(1)
|
|
|
|
if "--tb=short" not in extra_args_new:
|
|
extra_args_new.append("--tb=short")
|
|
|
|
extra_args_new.extend(testcase_path_list)
|
|
logger.info(f"start to run tests with pytest. HttpRunner version: {__version__}")
|
|
return pytest.main(extra_args_new)
|
|
|
|
|
|
def main():
|
|
"""API test: parse command line options and run commands."""
|
|
parser = argparse.ArgumentParser(description=__description__)
|
|
parser.add_argument(
|
|
"-V", "--version", dest="version", action="store_true", help="show version"
|
|
)
|
|
|
|
subparsers = parser.add_subparsers(help="sub-command help")
|
|
init_parser_run(subparsers)
|
|
sub_parser_make = init_make_parser(subparsers)
|
|
|
|
if len(sys.argv) == 1:
|
|
# httprunner
|
|
parser.print_help()
|
|
sys.exit(0)
|
|
elif len(sys.argv) == 2:
|
|
# print help for sub-commands
|
|
if sys.argv[1] in ["-V", "--version"]:
|
|
# httprunner -V
|
|
print(f"{__version__}")
|
|
elif sys.argv[1] in ["-h", "--help"]:
|
|
# httprunner -h
|
|
parser.print_help()
|
|
elif sys.argv[1] == "run":
|
|
# httprunner run
|
|
pytest.main(["-h"])
|
|
elif sys.argv[1] == "make":
|
|
# httprunner make
|
|
sub_parser_make.print_help()
|
|
sys.exit(0)
|
|
elif (
|
|
len(sys.argv) == 3 and sys.argv[1] == "run" and sys.argv[2] in ["-h", "--help"]
|
|
):
|
|
# httprunner run -h
|
|
pytest.main(["-h"])
|
|
sys.exit(0)
|
|
|
|
extra_args = []
|
|
if len(sys.argv) >= 2 and sys.argv[1] in ["run"]:
|
|
args, extra_args = parser.parse_known_args()
|
|
else:
|
|
args = parser.parse_args()
|
|
|
|
if args.version:
|
|
print(f"{__version__}")
|
|
sys.exit(0)
|
|
|
|
# set log level
|
|
try:
|
|
index = extra_args.index("--log-level")
|
|
if index < len(extra_args) - 1:
|
|
level = extra_args[index + 1]
|
|
else:
|
|
# not specify log level value
|
|
level = "INFO" # default
|
|
except ValueError:
|
|
level = "INFO" # default
|
|
|
|
init_logger(level)
|
|
|
|
if sys.argv[1] == "run":
|
|
sys.exit(main_run(extra_args))
|
|
elif sys.argv[1] == "make":
|
|
main_make(args.testcase_path)
|
|
|
|
|
|
def main_hrun_alias():
|
|
"""command alias
|
|
hrun = httprunner run
|
|
"""
|
|
if len(sys.argv) == 2:
|
|
if sys.argv[1] in ["-V", "--version"]:
|
|
# hrun -V
|
|
sys.argv = ["httprunner", "-V"]
|
|
elif sys.argv[1] in ["-h", "--help"]:
|
|
pytest.main(["-h"])
|
|
sys.exit(0)
|
|
else:
|
|
# hrun /path/to/testcase
|
|
sys.argv.insert(1, "run")
|
|
else:
|
|
sys.argv.insert(1, "run")
|
|
|
|
main()
|
|
|
|
|
|
def main_make_alias():
|
|
"""command alias
|
|
hmake = httprunner make
|
|
"""
|
|
sys.argv.insert(1, "make")
|
|
main()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|