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

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

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

import unittest
import time
from first.utils.HTMLTestRunner import HTMLTestRunner
import smtplib
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart

if __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 = 'xxx@163.com' #发送方邮件地址
passwd = 'xxxxx'              #发送方密码
receiver = 'yyy@qq.com'  #接收报告方邮件地址

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 smtplib
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from first.run import reportdir

def sendmail():
host = 'smtp.163.com'
sender = 'xxx@163.com' #发送方邮件地址
passwd = 'xxxxxx'              #发送方密码
receiver = 'yyy@qq.com'  #接收报告方邮件地址

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 unittest
import time
from first.utils.HTMLTestRunner import HTMLTestRunner
from first.utils import sendmail

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')

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