整合自动发邮件功能

解决了前面的问题后,现在就可以将自动发邮件功能集成到自动化测试项目中了。下面重新编辑runtest.py文件

#runtest.py
#coding:utf-8 from HTMLTestRunner import HTMLTestRunner from email.mime.text import MIMEText from email.header import Header import smtplib import unittest,time,os #===================定义发送邮件====================== def send_mail(file_new): f=open(file_new,'rb') mail_body=f.read() f.close() msg=MIMEText(mail_body,'html','utf-8') msg['subject']=Header(u"自动化测试报告",'utf-8') msg['from']='xxxxxx@sina.com' msg['to']='xxxxxx@qq.com' smtp=smtplib.SMTP() smtp.connect("smtp.sina.com") smtp.login("xxxxx","xxxxxxx") smtp.sendmail("xxxxxxx@sina.com","xxxxxxxx@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_dir='F:\\study\\web_demo1\\test_case' test_report='F:\\study\\web_demo1\\report' discover=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=u"测试报告",description=u"用例执行情况:") runner.run(discover) fp.close() newreport=new_report(test_report) send_mail(newreport)#发送测试报告

  

执行后,结果如下图所示:

 

分析如下:
1.通过unittest框架的discover()找到匹配测试用例,由HTMLTestRunner的run()方法执行测试用例并生成最新的测试报告。
2.调用new_report()函数找到测试报告目录(report)下最新生成的测试报告,返回测试报告的路径。
3.将得到的最新测试报告的完整路径传给send_mail()函数,实现发邮件功能。