接上篇 python smtplib模块自动收发邮件功能(一),用python smtplib模块实现了发送邮件程序了,那么接下来我们需要现在要解决的问题如何在 test_report\目录下找到最新生成的报告,只有找到了才能把发邮件功能,然后将其集成到我们的自动化测试应用中.
一、获取最新的test_report
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
#coding=utf-8
import smtplib
from email.mime.text import mimetext
from email.header import header
import os,datetime,time
result_dir = 'c:\\python34\\test_report' # test_report的绝对路径
lists = os.listdir(result_dir)
print (lists)
lists.sort(key = lambda fn: os.path.getmtime(result_dir + "\\" + fn)
if not os.path.isdir(result_dir + "\\" + fn) else 0 )
print ( '最新的文件为:' + lists[ - 1 ])
file = os.path.join(result_dir,lists[ - 1 ])
print ( file )
|
f5,运行,得到:
那么 c:\python34\test_report\2016-03-24-16_00_34_result.html是最新的test_report
二、整合自动发送邮件功能
主要实现以下几部分:
1.运行相关的 cases 生成htmltest report。
2.将test report发送到指定邮箱。
直接上脚本:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
import unittest
import htmltestrunner
import os
import time
import datetime
import smtplib
from email.mime.text import mimetext
from email.mime.multipart import mimemultipart
from email.mime.image import mimeimage
from email.header import header
#定义发送邮件
def sentemail(file_new):
#发信邮箱
sender = 'abc@ciexxx.com'
#收信邮箱
receiver = '12345@qq.com'
#定义正文
f = open (file_new, 'rb' )
mail_body = f.read()
f.close()
msg = mimetext(mail_body,_subtype = 'html' ,_charset = 'utf-8' )
#定义标题
msg[ 'subject' ] = u "搜狗搜索测试报告"
msg[ 'date' ] = time.strftime( '%a, %d %b %y %h:%m:%s %z' )
smtp = smtplib.smtp()
#smtpserver='smtp.263xmail.com'
smtp.connect( 'smtp.263xmail.com' )
username = 'abc@ciexxx.com'
password = '123456'
smtp.login(username,password)
smtp.sendmail(sender,receiver,msg.as_string())
smtp.quit()
print ( 'email has been sent out!' )
#查找测试报告,调用收发邮件功能
def sendreport():
result_dir = 'c:\\python34\\test_report'
lists = os.listdir(result_dir)
lists.sort(key = lambda fn: os.path.getmtime(result_dir + "\\" + fn)
if not os.path.isdir(result_dir + "\\" + fn) else 0 )
#print (u'最新测试生成的报告:'+lists[-1])
print (u '最新测试生成的报告:' + lists[ - 2 ])
#找到最新生成的文件
#file_new=os.path.join(result_dir,lists[-1])
file_new = os.path.join(result_dir,lists[ - 2 ])
print (file_new)
sentemail(file_new)
listaa = 'c:\\python34\\test_case'
def createsuitel():
testunit = unittest.testsuite()
'''discover方法定义'''
discover = unittest.defaulttestloader.discover(listaa,
pattern = 'unittesthtml_*.py' ,
top_level_dir = none)
for test_suite in discover:
for test_case in test_suite:
testunit.addtests(test_case)
print (testunit)
return testunit
alltestnames = createsuitel()
now = time.strftime( '%y-%m-%d-%h_%m_%s' ,time.localtime(time.time()))
file_name = 'c:\\python34\\test_report\\'+now+' _result.html'
fp = open (file_name, 'wb' )
runner = htmltestrunner.htmltestrunner(
stream = fp,
title = u '搜狗搜索测试报告' ,
description = u '用例执行情况:' )
if __name__ = = "__main__" :
runner.run(alltestnames)
time.sleep( 2 )
sendreport()
fp.close()
|
f5运行,得到:
查看邮箱,如图所示:
打开邮件内容,如图所示:
ok,就这样实现了实际项目中的自动收发邮件功能。
另,几个知识点:
1. os.listdir()
用于获取目录下的所有文件列表
2. lists.sort()
python 列表有一个内置的列表。sort()方法用于改变列表中元素的位置。
3. key=lambda fn:
key 是带一个参数的函数,用来为每个元素提取比较值.默认为 none, 即直接比较每个元素.
4.os.path.isdir()
isdir()函数判断某一路径是否为目录
5.lists[-1]
-1 表示取文件列表中的最大值,也就是最新被创建的文件.
6.os.path.join()
join()方法用来连接字符串,通过路径与文件名的拼接,我们将得到目录下最新被创建的的文件名的完整路径。
7.sentmail(file_new)
定义一个 sentmail()发邮件函数,接收一个参数 file_new,表示接收最新生成的测试报告文件.
8.open(file_new, ‘rb')
以读写(rb)方式打开最新生成的测试报告文件.
9.sendreport()
定义 sendreport()用于找最新生成的测试报告文件 file_new.
在成功实现这个sample之前,遇到过1个问题:
指定的邮箱可以正常收到邮件,但所得到的邮件内容是空的,这是由于 htmltestrunner 报告文件的机制所引起的。在测试用例运行之前生成报告文件,在整个程序没有彻底运行结束前,程序并没有把运行的结果写入到文件中,所以,在用例运行完成后发邮件,造成邮件内容是空的。
最开始的脚本,其中两行是这样的:
1
2
3
|
print (u '最新测试生成的报告:' + lists[ - 1 ])
#找到最新生成的文件
file_new = os.path.join(result_dir,lists[ - 1 ])
|
于是,运行结束后,出现了问题,指定的邮箱可以正常收到邮件,但所得到的邮件内容是空的。也就是说,脚本运行还没有结束,就已经执行了邮件的自动发送功能。
于是,将上述的两行,改后的脚本:
1
2
3
|
print (u '最新测试生成的报告:' + lists[ - 2 ])
#找到最新生成的文件
file_new = os.path.join(result_dir,lists[ - 2 ])
|
所以,我们不能在整个程序未运行结束时发送当前的测试报告,我们可以选择上一次运行结果的报告进行发送。
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/liujingqiu/article/details/50973848