HTMLTestRunner报告增加截图功能

本文中使用的HTMLTestRunner的模板为：HTMLTestRunnerCN

。

在模板上增加了截图功能，样例如下：



这里我将测试通过的以截图形式展示，测试失败则显示相应的错误异常。

具体相应的代码模块如下：

使用的POM设计模式，封装代码，测试需要截屏时直接调用即可

file_path:根据自己的需求填写

def get_windows_img(self):
self.logger = logging.getLogger(__name__)
file_path = 'D:/python/pom_adv/report/screenshots/'
rq = time.strftime('%Y%m%d%H%M%S', time.localtime(time.time()))
screen_name = file_path + rq + '.png'
print(screen_name)
try:
self.driver.get_screenshot_as_file(screen_name)
self.logger.info("Had take screenshot and save to folder : /screenshots")
except NameError as e:
self.logger.error("Failed to take screenshot! %s" % e)
self.get_windows_img()

修改HTMLTestRunner

def _generate_report_test(self, rows, cid, tid, n, t, o, e):
...
# 插入图片
po = str(uo).find('screenshots')
if (uo and po != -1):
hidde_status = ''
image_url = 'file:///'+uo
else:
hidde_status = '''hidden="hidden"'''
image_url = ''

script = self.REPORT_TEST_OUTPUT_TMPL % dict(
id = tid[2:],
output = saxutils.escape(ue),
hidde=hidde_status,
image=image_url,
)
...

REPORT_TEST_OUTPUT_TMPL = r"""
%(id)s: %(output)s
<img %(hidde)s src="%(image)s" alt="picture_shot" height="200" width="400"></img>
<a  %(hidde)s  href="%(image)s">screen_shot</a>
"""

源码地址：smarthaut