• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• httprunner-master
  • .github
    • ISSUE_TEMPLATE
      • feature_request_zh.md
      • feature_request.md
      • bug_report.md
      • bug_report_zh.md
    • workflows
      • unittest.yml
      • integration_test.yml
  • poetry.lock
  • examples
    • postman_echo
      • cookie_manipulation
        • hardcode.yml
        • set_delete_cookies.yml
      • debugtalk.py
      • request_methods
        • demo_testsuite_yml
          • request_with_testcase_reference_test.py
          • request_with_functions_test.py
          • __init__.py
        • request_with_functions.yml
        • hardcode.yml
        • request_with_variables_test.py
        • hardcode_test.py
        • validate_with_functions.yml
        • request_with_parameters.yml
        • validate_with_variables_test.py
        • request_with_testcase_reference_test.py
        • validate_with_variables.yml
        • request_with_variables.yml
        • request_with_functions_test.py
        • __init__.py
        • validate_with_functions_test.py
        • account.csv
        • request_with_testcase_reference.yml
        • conftest.py
        • demo_testsuite.yml
      • __init__.py
    • __init__.py
    • httpbin
      • basic_test.py
      • upload.yml
      • test.env
      • debugtalk.py
      • validate_test.py
      • load_image_test.py
      • load_image.yml
      • basic.yml
      • upload_test.py
      • hooks_test.py
      • __init__.py
      • user_agent.csv
      • account.csv
      • hooks.yml
      • validate.yml
  • mkdocs.yml
  • LICENSE
  • httprunner
    • builtin
      • functions.py
      • __init__.py
      • comparators.py
    • exceptions.py
    • cli.py
    • client.py
    • testcase.py
    • models.py
    • response.py
    • make.py
    • __main__.py
    • ext
      • locust
        • locustfile.py
        • __init__.py
      • uploader
        • __init__.py
      • __init__.py
      • har2case
        • __init__.py
        • utils.py
        • core.py
    • __init__.py
    • utils.py
    • parser.py
    • loader.py
    • runner.py
    • app
      • __init__.py
      • main.py
      • routers
        • deps.py
        • debugtalk.py
        • __init__.py
        • debug.py
    • compat.py
    • scaffold.py
  • README.md
  • tests
    • cli_test.py
    • parser_test.py
    • loader_test.py
    • make_test.py
    • utils_test.py
    • runner_test.py
    • data
      • a-b.c
        • 2 3.yml
        • 1.yml
        • 中文case.yml
      • debugtalk.py
      • .csv
    • ext
      • __init__.py
      • har2case
        • har_utils_test.py
        • data
          • demo-quickstart.har
          • demo.har
        • core_test.py
        • __init__.py
    • __init__.py
    • compat_test.py
    • app
      • debug_test.py
      • __init__.py
    • scaffold_test.py
    • response_test.py
  • pyproject.toml
  • .gitignore
  • docs
    • quickstart.md
    • images
    • user
      • scaffold.md
      • run_testcase.md
      • gen_tests.md
      • run_loadtest.md
      • write_testcase.md
      • testing_report.md
    • CNAME
    • index.md
    • sponsors.md
    • CHANGELOG.md
    • installation.md
    • assets
      • sentry-logo-black.svg

import argparse
import enum
import os
import sys

import pytest
from loguru import logger
from sentry_sdk import capture_message

from httprunner import __description__, __version__
from httprunner.compat import ensure_cli_args
from httprunner.ext.har2case import init_har2case_parser, main_har2case
from httprunner.make import init_make_parser, main_make
from httprunner.scaffold import init_parser_scaffold, main_scaffold
from httprunner.utils import 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:
    capture_message("start to run")
    # 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")
    sub_parser_run = init_parser_run(subparsers)
    sub_parser_scaffold = init_parser_scaffold(subparsers)
    sub_parser_har2case = init_har2case_parser(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] == "startproject":
            # httprunner startproject
            sub_parser_scaffold.print_help()
        elif sys.argv[1] == "har2case":
            # httprunner har2case
            sub_parser_har2case.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", "locusts"]:
        args, extra_args = parser.parse_known_args()
    else:
        args = parser.parse_args()

    if args.version:
        print(f"{__version__}")
        sys.exit(0)

    if sys.argv[1] == "run":
        sys.exit(main_run(extra_args))
    elif sys.argv[1] == "startproject":
        main_scaffold(args)
    elif sys.argv[1] == "har2case":
        main_har2case(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()


def main_har2case_alias():
    """ command alias
        har2case = httprunner har2case
    """
    sys.argv.insert(1, "har2case")
    main()


if __name__ == "__main__":
    main()