rename package name from ate to httprunner

httprunner/cli.py

@@ -0,0 +1,127 @@
+import argparse
+import logging
+import os
+import sys
+from collections import OrderedDict
+
+import PyUnitReport
+from PyUnitReport import __version__ as pyu_version
+from httprunner import __version__ as ate_version
+from httprunner import exception
+from httprunner.task import TaskSuite
+from httprunner.utils import create_scaffold
+
+
+def main_ate():
+    """ API test: parse command line options and run commands.
+    """
+    parser = argparse.ArgumentParser(
+        description='Api Test Engine.')
+    parser.add_argument(
+        '-V', '--version', dest='version', action='store_true',
+        help="show version")
+    parser.add_argument(
+        'testset_paths', nargs='*',
+        help="testset file path")
+    parser.add_argument(
+        '--log-level', default='INFO',
+        help="Specify logging level, default is INFO.")
+    parser.add_argument(
+        '--report-name',
+        help="Specify report name, default is generated time.")
+    parser.add_argument(
+        '--failfast', action='store_true', default=False,
+        help="Stop the test run on the first error or failure.")
+    parser.add_argument(
+        '--startproject',
+        help="Specify new project name.")
+
+    args = parser.parse_args()
+
+    if args.version:
+        print("HttpRunner version: {}".format(ate_version))
+        print("PyUnitReport version: {}".format(pyu_version))
+        exit(0)
+
+    log_level = getattr(logging, args.log_level.upper())
+    logging.basicConfig(level=log_level)
+
+    project_name = args.startproject
+    if project_name:
+        project_path = os.path.join(os.getcwd(), project_name)
+        create_scaffold(project_path)
+        exit(0)
+
+    report_name = args.report_name
+    if report_name and len(args.testset_paths) > 1:
+        report_name = None
+        logging.warning("More than one testset paths specified, \
+                        report name is ignored, use generated time instead.")
+
+    results = {}
+    success = True
+
+    for testset_path in set(args.testset_paths):
+
+        testset_path = testset_path.rstrip('/')
+
+        try:
+            task_suite = TaskSuite(testset_path)
+        except exception.TestcaseNotFound:
+            success = False
+            continue
+
+        output_folder_name = os.path.basename(os.path.splitext(testset_path)[0])
+        kwargs = {
+            "output": output_folder_name,
+            "report_name": report_name,
+            "failfast": args.failfast
+        }
+        result = PyUnitReport.HTMLTestRunner(**kwargs).run(task_suite)
+        results[testset_path] = OrderedDict({
+            "total": result.testsRun,
+            "successes": len(result.successes),
+            "failures": len(result.failures),
+            "errors": len(result.errors),
+            "skipped": len(result.skipped)
+        })
+
+        if len(result.successes) != result.testsRun:
+            success = False
+
+        for task in task_suite.tasks:
+            task.print_output()
+
+    return 0 if success is True else 1
+
+def main_locust():
+    """ Performance test with locust: parse command line options and run commands.
+    """
+    try:
+        from httprunner import locusts
+    except ImportError:
+        print("Locust is not installed, exit.")
+        exit(1)
+
+    sys.argv[0] = 'locust'
+    if len(sys.argv) == 1:
+        sys.argv.extend(["-h"])
+
+    if sys.argv[1] in ["-h", "--help", "-V", "--version"]:
+        locusts.main()
+        sys.exit(0)
+
+    try:
+        testcase_index = sys.argv.index('-f') + 1
+        assert testcase_index < len(sys.argv)
+    except (ValueError, AssertionError):
+        print("Testcase file is not specified, exit.")
+        sys.exit(1)
+
+    testcase_file_path = sys.argv[testcase_index]
+    sys.argv[testcase_index] = locusts.parse_locustfile(testcase_file_path)
+
+    if "--full-speed" in sys.argv:
+        locusts.run_locusts_at_full_speed(sys.argv)
+    else:
+        locusts.main()