aritest发送测试报告到邮件

#!/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)