Python +Selenium 測試框架unittest+郵件+測試報告

集中爲測試用例管理，我們這裏用到了unittest+email和前文提到的修改過後的HTMLTestRunner

集中管理測試用例，我們這裏用到了unittest+email 和前文提到的修改過後的HTMLTestRunner

使用過程中直接：Run 就可以執行指定目錄中的全部測試用例，自動生成測試報告：testOutput\html\gf.html

郵件發送到指定的郵箱中

import HTMLTestRunner, unittest
import os
import smtplib
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from email.utils import formataddr, formatdate
from util_s.config import Config

#配置文件讀取
config = Config("cm")
mail_host = config.get("mail_host") #郵件代理
my_sender = config.get("my_sender")  # 發件人郵箱賬號
my_pass = config.get("my_pass")  # 發件人郵箱密碼（密鑰）
my_user = config.get("my_user") # 收件人郵箱賬號

def allcase():
    case_dir = os.path.join(os.getcwd(), "./test")
    testcase = unittest.TestSuite()
    discover = unittest.defaultTestLoader.discover(case_dir, pattern="case*.py", top_level_dir=None)
    """讀取目錄：/test 下文件名正則爲‘case*.py’ 的全部用例"""
    for test_suite in discover:
        for test_case in test_suite:
            testcase.addTest(test_case)
    return testcase


def mail():
    msg = MIMEMultipart()
    msg['From'] = formataddr(["zl", my_sender])  # 括號裏的對應發件人郵箱暱稱、發件人郵箱賬號
    msg['To'] = formataddr(["zl", my_user])  # 括號裏的對應收件人郵箱暱稱、收件人郵箱賬號
    msg['Subject'] = "自動化測試測試報告"  # 郵件的主題，也可以說是標題
    msg['Date'] = formatdate()

    report_path = os.path.join(os.getcwd(), r"testOutput\html\gf.html")
    part1 = MIMEText(open(report_path, 'rb').read(), 'base64', 'utf-8')
    part1["Content-Type"] = 'application/octet-stream'
    # 這裏的filename可以任意寫，寫什麼名字，郵件中顯示什麼名字
    part1["Content-Disposition"] = 'attachment; filename="result.html"'
    part2 = MIMEText("自動化測試測試報告:附件爲自動化測試報告")

    msg.attach(part1)
    msg.attach(part2)

    server = smtplib.SMTP_SSL(mail_host, 465)  # 發件人郵箱中的SMTP服務器，端口是25
    server.login(my_sender, my_pass)  # 括號中對應的是發件人郵箱賬號、郵箱密碼
    server.sendmail(my_sender, [my_user, ], msg.as_string())  # 括號中對應的是發件人郵箱賬號、收件人郵箱賬號、發送郵件
    server.quit()  # 關閉連接


if __name__ == "__main__":
    report_path = os.path.join(os.getcwd(), r"testOutput\html\gf.html")
    fp = open(report_path, "wb+")
    runner = HTMLTestRunner.HTMLTestReportCN(
        stream=fp,
        title=u'自動化測試測試報告',
        description='詳細測試用例結果',  # 不傳默認爲空
        tester=u"tester"
    )
    runner.run(allcase())
    fp.close()
    mail()