#!/usr/bin/env python # -*- coding=utf-8 -*- __CreateAt__ = '2020/4/19-17:34' import shutil from airtest.cli.runner import AirtestCase, run_script from argparse import * from air_case.report import report import jinja2 import io from util.send_email import SendEmail from util.compress_file import copy_and_zip from util.android_util import attached_devices from util.common import * PATH = lambda p: os.path.abspath( os.path.join(os.path.dirname(__file__), p) ) def run_case(data): devices = attached_devices() if not devices: print("无可用设备") return test = CustomAirtestCase(data["root_path"]) device = ["Android://127.0.0.1:5037/%s?touch_method=ADBTOUCH" % data["dev"]] # "android:///%s" % data["dev"] test.run_air(device, data) def run(root_path, test_case, device, log_date): script = os.path.join(root_path, test_case["module"], test_case["case"]) log = os.path.join(root_path, 'log', test_case["module"], log_date, test_case["case"].replace('.air', '')) if os.path.isdir(log): shutil.rmtree(log) else: os.makedirs(log) print(str(log) + 'is created') output_file = os.path.join(log, 'log.html') args = Namespace(device=device, log=log, compress=None, recording=None, script=script, no_image=None) try: run_script(args, AirtestCase) is_success = True except: is_success = False return {"is_success": is_success, "output_file": output_file, "script": script, "log": log} class CustomAirtestCase(AirtestCase): @classmethod def setUpClass(cls): super(CustomAirtestCase, cls).setUpClass() def __init__(self, root_dir): self.fail_data = [] self.results = {"dev": "", "modules": [], "total_time": "", "data": []} self.log_list = [] super().__init__() def setUp(self): super(CustomAirtestCase, self).setUp() def tearDown(self): print("custom tearDown") super(CustomAirtestCase, self).setUp() def run_air(self, device, data): root_log = os.path.join(data["root_path"], "log") # remove_log表示如果传值就会删除整个log文件夹，删除后无法查看历史报告 if os.path.isdir(root_log): if data.get("remove_log"): shutil.rmtree(root_log, ignore_errors=True) print("删除log文件夹") else: os.makedirs(root_log) get_data_list = get_test_case(data) if not get_data_list: print("无可用用例") return # 整个用例开始执行时间 start_time = datetime.now().strftime("%H:%M:%S") modules = [] # 日志按照日期格式生成 log_date = datetime.now().strftime("%Y-%m-%d-%H-%M-%S") for j in get_data_list: # 日志按照日期格式生成 # log_date = datetime.now().strftime("%Y-%m-%d-%H-%M-%S") # 用例开始执行日期 st_date = datetime.now().strftime("%Y-%m-%d %H:%M:%S") # 用例开始执行时间 s_time = datetime.now().strftime("%H:%M:%S") # 循环运行用例 get_run = run(data["root_path"], j, device, log_date) # 用例结束执行日期 end_date = datetime.now().strftime("%Y-%m-%d %H:%M:%S") # 用例结束执行时间 e_time = datetime.now().strftime("%H:%M:%S") # 用例耗时时间 sum_time = get_case_total_time(s_time, e_time) # 生成测试用例的详情报告 rpt = report.LogToHtml(get_run["script"], get_run["log"], "../../../../report") # rpt = report.LogToHtml(get_run["script"], get_run["log"]) rpt.report("log_template.html", output_file=get_run["output_file"]) # 记录测试结果 result = {"name": j["case"].replace(".air", ""), "result": get_run["is_success"], "start_date": st_date, "end_date": end_date, "sum_time": sum_time, "module": j["module"], "log_date": log_date} modules.append(j["module"]) self.results["data"].append(result) self.log_list.append(get_run["log"]) # 记录失败用例 if not get_run["is_success"]: self.fail_data.append({"module": j["module"], "case": j["case"]}) # 整个用例结束执行时间 end_time = datetime.now().strftime("%H:%M:%S") # 以小时，分钟，秒钟的方式记录所有用例耗时时间 total_time = get_case_total_time(start_time, end_time) self.results["total_time"] = total_time # 记录测试模块 self.results["modules"] = get_test_modules(modules) # 记录设备名字 self.results["phone"] = data["phone"] # 打印失败用例 if self.fail_data: print("存在失败用例") print(self.fail_data) # 生成测试报告 env = jinja2.Environment( loader=jinja2.FileSystemLoader(data["root_path"]), extensions=(), autoescape=True ) print("用例结果为：%s" % self.results) t = os.path.join(data["root_path"], "summary_template.html") if not os.path.exists(t): shutil.copy(PATH("util/summary_template.html"), data["root_path"]) template = env.get_template("summary_template.html", data["root_path"]) html = template.render({"results": self.results}) output_file = os.path.join(data["root_path"], "summary_%s.html" % datetime.now().strftime("%Y-%m-%d-%H-%M-%S")) with io.open(output_file, 'w', encoding="utf-8") as f: f.write(html) # 按日期生成测试报告,方便对比历史报告，但是程序入口字段需要设置为"remove_log": False print("测试报告为：%s" % output_file) # 固定输出给CI output_file = os.path.join(data["root_path"], "summary.html") with io.open(output_file, 'w', encoding="utf-8") as f: f.write(html) # 当发送邮件参数为真，就对文件进行压缩并发送测试报告到指定邮箱 if data.get("send_email"): # 压缩测试报告 report_file = os.path.join(data["root_path"], "report") case_log = os.path.join(data["root_path"], "log") case_html = output_file zip_list = [report_file, case_log, case_html] zip_path = copy_and_zip(zip_list, "report") # 发送测试报告邮件 to_addr = ["284772894@qq.com"] SendEmail.send(zip_path, to_addr) if __name__ == '__main__': """ python E:/project/trade-auto/runner1.py """ root_path = PATH("air_case") # test_plan=1 表示调试用例需要配合test_module使用，0表示全部用例,比如:"test_plan": 0, "test_module": ["我的"] # remove_log为真 表示是否运行用例之前，删除log文件夹，删除后会影响查看历史报告 # send_email为真，表示发送邮件，否则就不发送 # dev参数使用adb devices 获取到设备udid，填进去，若想多台并行，这里的dev必须填正确，不能是同一台手机 # data = {"root_path": root_path, "test_plan": 0, "test_module": [], "dev": "TPG5T18130013404", "phone": "Nova2s"} data = {"root_path": root_path, "remove_log": True, "send_email": True, "test_plan": 0, "test_module": [], "dev": "TPG5T18130013404", "phone": "Nova2s"} run_case(data)