（五十三）自动化测试高级应用之自动发邮件功能-整合自动发邮件功能

# !/usr/bin/env python
# -*- coding: UTF-8 –*-
__author__ = ‘Mr.Li‘
from HTMLTestRunner import HTMLTestRunner
from email.mime.text import MIMEText
from email.header import Header
import smtplib
import unittest
import time
import os

# =====================定义发送邮件================
def send_mail(file_new):
    f = open(file_new,‘rb‘)
    mail_bobd = f.read()
    f.close()

    msg = MIMEText(mail_bobd,‘html‘,‘utf-8‘)
    msg[‘Subject‘] = Header(‘自动化测试报告‘,‘utf-8‘)

    smtp = smtplib.SMTP()
    smtp.connect("smtp.qq.com")
    smtp.login(‘xxxx@qq.com‘,‘mbnzfxlnmwbkbcfb‘)
    smtp.sendmail(‘xxxx@qq.com‘,‘xxxx@qq.com‘,msg.as_string())
    smtp.quit()
    print(‘email has send out!‘)

# ===========查找测试报告目录，找到最新生成的测试报告文件============
def new_report(testreport):
    lists = os.listdir(testreport)
    lists.sort(key=lambda fn: os.path.getmtime(testreport + ‘\\‘ + fn))
    file_new = os.path.join(testreport,lists[-1])
    print(file_new)
    return file_new

#执行测试类的测试方法
if __name__ == ‘__main__‘:
    test_report = ‘C:\\Users\\86136\\PycharmProjects\\spider\\test\\test_project\\report\\‘
    test_dir = ‘C:\\Users\\86136\\PycharmProjects\\spider\\test\\test_project\\test_case‘

    #把测试用例加载到一个套件
    dicscover = unittest.defaultTestLoader.discover(test_dir, pattern=‘test*.py‘)

    # 按照一定格式获取当前时间
    now = time.strftime("%Y-%m-%d %H_%M_%S")

    # 定义报告的存放路径
    filename = test_report + ‘\\‘ + now + ‘result.html‘
    fp = open(filename, ‘wb‘)
    # 定义测试报告
    runner = HTMLTestRunner(stream=fp,
                            title=‘测试报告‘,
                            description=‘用例执行情况：‘)
    runner.run(dicscover)  # 运行测试用例
    fp.close()  # 关闭报告文件

    new_report = new_report(test_report)#找到最新测试报告
    send_mail(new_report) #发送测试报告