33、Selenium + Python 实现 UI 自动化测试-正片5

测试完成后，确实生成了报告。最好能自动发送给相关的人员，而不是我到report 目录下找到报告，再发邮件给对方。

一、调整run.py 代码如下：

import unittestimport timefrom first.utils.HTMLTestRunner import HTMLTestRunnerimport smtplibfrom email.mime.text import MIMETextfrom email.mime.multipart import MIMEMultipartif __name__ == '__main__':    testdir = "./cases"    cur_time = time.strftime('%Y%m%d%H%M%S', time.localtime(time.time()))    reportname = "HTMLReport_" + cur_time + ".html"    reportdir = "./report/{}".format(reportname)    discover = unittest.defaultTestLoader.discover(start_dir=testdir,pattern='test*.py')    # runner = unittest.TextTestRunner()    # runner.run(discover)    with open(reportdir,'wb+') as f:        runner = HTMLTestRunner(stream=f,                                title='redmine测试报告名称',                                description='redmine 测试描述信息',                                verbosity=2)        runner.run(discover)    host = 'smtp.163.com'    sender = '[e-mail]' #发送方邮件地址    passwd = 'xxxxx'              #发送方密码    receiver = '[e-mail]'  #接收报告方邮件地址    msg = MIMEMultipart()    msg['from'] = sender    msg['to'] = receiver    msg['subject'] = '主题'    msg.attach(MIMEText('邮件正文'))    att1 = MIMEText(open(reportdir, 'rb').read(), 'base64', 'utf-8')    att1["Content-Type"] = 'application/octet-stream'    # 这里的filename可以任意写，写什么名字，邮件中显示什么名字    att1["Content-Disposition"] = 'attachment; filename="report.html"'    msg.attach(att1)    try:        smtpobj = smtplib.SMTP(host, port=25)        smtpobj.login(sender, passwd)        smtpobj.sendmail(sender, receiver, msg.as_string())        smtpobj.quit()        print('send success')    except:        print('send err')

运行结果：

C:\Python36\python.exe E:/python/test1/first/run.py
ok test_login (test_000_login.TestLogin)
ok test_new_pj (test_001_new_pj.TestNewProject)
ok test_new_bug (test_002_new_bug.TestNewBug)
<_io.TextIOWrapper name='<stderr>' mode='w' encoding='UTF-8'> 
Time Elapsed: 0:01:19.129389
send success


Process finished with exit code 0

邮件收到报告如下：

报告内容如下：

效果还不错，达到了我们的预期。

二、run.py 应该是一个执行用例文件，不应该包含太多不相关的代码，比如上面发送邮件部分代码。我们将发送邮件的功能封装一下，独立出来。

我们在utils下面新进sendmail.py文件：

import smtplibfrom email.mime.text import MIMETextfrom email.mime.multipart import MIMEMultipartfrom first.run import reportdirdef sendmail():    host = 'smtp.163.com'    sender = '[e-mail]' #发送方邮件地址    passwd = 'xxxxxx'              #发送方密码    receiver = '[e-mail]'  #接收报告方邮件地址    msg = MIMEMultipart()    msg['from'] = sender    msg['to'] = receiver    msg['subject'] = '主题'    msg.attach(MIMEText('邮件正文'))    att1 = MIMEText(open(reportdir, 'rb').read(), 'base64', 'utf-8')    att1["Content-Type"] = 'application/octet-stream'    # 这里的filename可以任意写，写什么名字，邮件中显示什么名字    att1["Content-Disposition"] = 'attachment; filename="report.html"'    msg.attach(att1)    try:        smtpobj = smtplib.SMTP(host, port=25)        smtpobj.login(sender, passwd)        smtpobj.sendmail(sender, receiver, msg.as_string())        smtpobj.quit()        print('send success')    except:        print('send err')

然后修改run.py 调用sendmail函数

import unittestimport timefrom first.utils.HTMLTestRunner import HTMLTestRunnerfrom first.utils import sendmailtestdir = "./cases"cur_time = time.strftime('%Y%m%d%H%M%S', time.localtime(time.time()))reportname = "HTMLReport_" + cur_time + ".html"reportdir = "./report/{}".format(reportname)discover = unittest.defaultTestLoader.discover(start_dir=testdir, pattern='test*.py')if __name__ == '__main__':     # runner = unittest.TextTestRunner()    # runner.run(discover)    with open(reportdir,'wb+') as f:        runner = HTMLTestRunner(stream=f,                                title='redmine测试报告名称',                                description='redmine 测试描述信息',                                verbosity=2)        runner.run(discover)    sendmail.sendmail()

好，简单总结一下：

1、引入了自动发送测试报告的功能，每次执行完自动化测试，将报告自动发给相关人；

2、将发送邮件功能，单独写成一个函数，简化run.py