上一篇文章写了给HTMLTestRunner增加饼状图,这让我们的报告看起来更直观一些,但是细节方面还不是太让人满意,比如,我们查看执行通过、失败、错误的用例时,HTMLTestRunner给出了一大段代码,让人看着头大,如果给出截图是否更直观,一目了然的了解用例执行情况呢,答案是肯定的。
给HTMLTestRunner增加截图分两部分,第一部分,需要编写截图功能,第二部分就是修改HTMLTestRunner给HTML样式增加截图路径以及展示样子
----------------------------------------------------------------------------------------------------------------------------------------------------------------------
效果展示:
第一步:增加截图功能
selenium提供了截图函数get_srceenshot_as_file(),我们只要调用就好了,需要我们自己写的是,截图保存位置和截图名称,根据个人喜好来定。
下面代码,以及把截图功能封装在一个函数中,我们只要传入driver、文件名称、路径就好,路径是写死的
# !/usr/bin/env python # -*- coding: UTF-8 –*- __author__ = ‘Mr.Li‘ from selenium import webdriver import os #截图函数 def insert_img(driver,file_name,file_path): base_dir = os.path.dirname(os.path.dirname(os.path.dirname(__file__))) file_path = base_dir + "/YJF-UI_TestCode/report/image/" + file_name driver.get_screenshot_as_file(file_path) print(‘Screenshot_Path:‘,file_path) if __name__ == ‘__main__‘: driver = webdriver.Chrome() driver.get(‘http://www.baidu.com‘) insert_img(driver,‘baidu.png‘) driver.quit()