mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-11 18:11:21 +08:00
84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
import argparse
|
|
import logging
|
|
import os
|
|
from collections import OrderedDict
|
|
|
|
import PyUnitReport
|
|
from ate import __version__
|
|
from ate.task import create_task
|
|
|
|
|
|
def main():
|
|
""" 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.")
|
|
|
|
try:
|
|
from jenkins_mail_py import MailgunHelper
|
|
mailer = MailgunHelper(parser)
|
|
except ImportError:
|
|
mailer = None
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.version:
|
|
print("ApiTestEngine version: {}".format(__version__))
|
|
exit(0)
|
|
|
|
log_level = getattr(logging, args.log_level.upper())
|
|
logging.basicConfig(level=log_level)
|
|
|
|
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 = {}
|
|
subject = "SUCCESS"
|
|
|
|
for testset_path in set(args.testset_paths):
|
|
|
|
testset_path = testset_path.strip('/')
|
|
task_suite = create_task(testset_path)
|
|
|
|
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:
|
|
subject = "FAILED"
|
|
|
|
flag_code = 0 if subject == "SUCCESS" else 1
|
|
if mailer and mailer.config_ready:
|
|
mailer.send_mail(subject, results, flag_code)
|
|
|
|
return flag_code
|