粗暴解决 HTMLTestRunner 加入截图展示功能 - 51Testing软件测试论坛

最近使用appium+python来做自动化，测试报告用的网上共享的HTMLTestRunner模块，但是里面貌似没有展示截图的功能，于是自己动手稍微修改了一下！
先上成果图，点击图片名字就能打开图片
总体思路为：在HTMLTestRunner的report模板中加入一列picture，然后将用来展示图片的html打印到每个case的日志中，再从日志中将这段文字截取出来放入report中（主要是为了简单粗暴的将每个case与相关截图正确对应起来~想不到其他办法啦！欢迎高手们给出更加简单快捷的办法~）
以下为我的操作方法，有点长~
一、在HTMLTestRunner.py里的原有的html表格中插入了一列picture（在<td>View</td>下面加入一行<td>Picture</td>，在<td> </td>下加入一行<td><a href="" target="_blank"></a></td>即可），用来展示脚本中的截图超链接
<tr id='header_row'>
<td>Test Group/Test case</td>
<td>Count</td>
<td>Pass</td>
<td>Fail</td>
<td>Error</td>
<td>View</td>
<td>Picture</td>
</tr>
%(test_list)s
<tr id='total_row'>
<td>Total</td>
<td>%(count)s</td>
<td>%(Pass)s</td>
<td>%(fail)s</td>
<td>%(error)s</td>
<td> </td>
<td><a href="" target="_blank"></a></td>
</tr>
</table>
二、还是在HTMLTestRunner.py里插入一行%(html)s来放置对应脚本的截图
REPORT_TEST_WITH_OUTPUT_TMPL = r"""
<tr id='%(tid)s' class='%(Class)s'>
<td class='%(style)s'><div class='testcase'>%(desc)s</div></td>
<td colspan='5' align='center'>
<a class="popup_link" onfocus='this.blur();' href="javascript:showTestDetail('div_%(tid)s')" >
%(status)s</a>
<div id='div_%(tid)s' class="popup_window">
<div style='text-align: right; color:red;cursor:pointer'>
<a onfocus='this.blur();' onclick="document.getElementById('div_%(tid)s').style.display = 'none' " >
[x]</a>
</div>
<pre>
%(script)s
</pre>
</div>
</td>
%(html)s
</tr>
三、依然在HTMLTestRunner.py中，_generate_report_test()函数中，插入html的内容，因为本身初学者，对unittest和htmltestrunner都不怎么了解，观察htmltestrunner的报告发现它能展示脚本执行过程中打印的log文本，所以我强行把自己组装的用于展示图片html代码打印到执行日志中，然后通过关键字截取出来，放置在script后面的位置，也就是刚才新建的picture列（要是觉得最后report中打印的log中加入了这段html影响美观，可以在下面代码中script里删掉~我懒得写了~）
script = self.REPORT_TEST_OUTPUT_TMPL % dict(
id = tid,
output = saxutils.escape(uo+ue),
)
s = uo+ue
html = s[s.find('htmlbegin')+9:s.find('htmlend')]
row = tmpl % dict(
tid = tid,
Class = (n == 0 and 'hiddenRow' or 'none'),
style = n == 2 and 'errorCase' or (n == 1 and 'failCase' or 'none'),
desc = desc,
script = script,
html = html,
status = self.STATUS[n],
)
四、在自己封装的手机截屏方法中加入分别将截图名称和截图存放路径插入对应的数组中（也可以使用二维数组），以下分别为打印日志，截屏，组装html的方法，在组装完成的html代码开头和结尾分别叫上htmlbegin和htmlend作为之后截取使用的关键字
# _*_ coding:utf-8 _*_
__author__ = 'gjj14453'
import logging
import os
import common.myDate as myDate
PATH = lambda p: os.path.abspath(
os.path.join(os.path.dirname(__file__), p))
pics = []
picpath = []
##################################
# 日志
# 0: debug
# 1：info
# 2：warning
# -1：error
###################################
def mylogger(msg, flag=1):
logging.basicConfig(level=logging.INFO,
format='%(asctime)s %(filename)s %(levelname)s %(message)s',
datefmt='%a, %d %b %Y %H:%M:%S',
filename='mylog.log',
filemode='w')
console = logging.StreamHandler()
console.setLevel(logging.DEBUG)
formatter = logging.Formatter('%(name)-12s: %(levelname)-8s %(message)s')
console.setFormatter(formatter)
logging.getLogger('').addHandler(console)
if flag == 0:
logging.debug(msg)
elif flag == 1:
logging.info(msg)
elif flag == 2:
logging.warning(msg)
screenshot()
elif flag == -1:
logging.error(msg)
screenshot()
logging.getLogger('').removeHandler(console)
# 截屏
def screenshot():
dirname = PATH('D:\\MyConfiguration\\gjj14453\\PycharmProjects\\untitled\\list_out' + "/screenshot")
os.popen("adb wait-for-device")
os.popen("adb shell screencap -p /data/local/tmp/tmp.png")
if not os.path.isdir(dirname):
os.makedirs(dirname)
pic = myDate.timestampname() + ".png"
path = dirname + "/" + pic
os.popen("adb pull /data/local/tmp/tmp.png " + PATH(path))
os.popen("adb shell rm /data/local/tmp/tmp.png")
mylogger("截图为：" + pic)
pics.append(pic)
picpath.append(path)
def creathtml(path, pic):
html = ''
if range(len(path)) > 0:
for i in range(len(path)):
if i == 0:
html = '<a href=' + path[i] + ' target="_blank">' + pic[i] + '</a>'
else:
html = html + '<br /><a href=' + path[i] + ' target="_blank">' + pic[i] + '</a>'
else:
html = ''
htmls = 'htmlbegin<td>' + html +'</td>htmlend'
return htmls
五、之后在case中加入打印html到日志中的语句就OK啦
class Test(unittest.TestCase):
def setUp(self):
log.picpath = []
log.pics = []
pass
def tearDown(self):
pass
def test(self):
log.mylogger('aaa', -1)
log.mylogger('bbb', -1)
log.mylogger(log.creathtml(log.picpath, log.pics))
最后，还是希望有高手能给我一个更加简单的插入对应截图的办法~本来想把截图存放的数组传到脚本执行的result中再一一提取，试了几次毫无成果~

复制代码