粗暴解决 HTMLTestRunner 加入截图展示功能

lsekfe

1. 最近使用appium+python来做自动化，测试报告用的网上共享的HTMLTestRunner模块，但是里面貌似没有展示截图的功能，于是自己动手稍微修改了一下！
   
2. 先上成果图，点击图片名字就能打开图片
   
3. 
   
4. 
   
5. 总体思路为：在HTMLTestRunner的report模板中加入一列picture，然后将用来展示图片的html打印到每个case的日志中，再从日志中将这段文字截取出来放入report中（主要是为了简单粗暴的将每个case与相关截图正确对应起来~想不到其他办法啦！欢迎高手们给出更加简单快捷的办法~）
   
6. 以下为我的操作方法，有点长~
   
7. 一、在HTMLTestRunner.py里的原有的html表格中插入了一列picture（在<td>View</td>下面加入一行<td>Picture</td>，在<td> </td>下加入一行<td><a href="" target="_blank"></a></td>即可），用来展示脚本中的截图超链接
   
8. 
   
9. <tr id='header_row'>
   
10.     <td>Test Group/Test case</td>
    
11.     <td>Count</td>
    
12.     <td>Pass</td>
    
13.     <td>Fail</td>
    
14.     <td>Error</td>
    
15.     <td>View</td>
    
16.     <td>Picture</td>
    
17. </tr>
    
18. %(test_list)s
    
19. <tr id='total_row'>
    
20.     <td>Total</td>
    
21.     <td>%(count)s</td>
    
22.     <td>%(Pass)s</td>
    
23.     <td>%(fail)s</td>
    
24.     <td>%(error)s</td>
    
25.     <td> </td>
    
26.     <td><a href="" target="_blank"></a></td>
    
27. </tr>
    
28. </table>
    
29. 二、还是在HTMLTestRunner.py里插入一行%(html)s来放置对应脚本的截图
    
30. 
    
31.     REPORT_TEST_WITH_OUTPUT_TMPL = r"""
    
32. <tr id='%(tid)s' class='%(Class)s'>
    
33.     <td class='%(style)s'><div class='testcase'>%(desc)s</div></td>
    
34.     <td colspan='5' align='center'>
    
35. 
    
36.     <!--css div popup start-->
    
37.     <a class="popup_link" onfocus='this.blur();' href="javascript:showTestDetail('div_%(tid)s')" >
    
38.         %(status)s</a>
    
39. 
    
40.     <div id='div_%(tid)s' class="popup_window">
    
41.         <div style='text-align: right; color:red;cursor:pointer'>
    
42.         <a onfocus='this.blur();' onclick="document.getElementById('div_%(tid)s').style.display = 'none' " >
    
43.            [x]</a>
    
44.         </div>
    
45.         <pre>
    
46.         %(script)s
    
47.         </pre>
    
48.     </div>
    
49.     <!--css div popup end-->
    
50. 
    
51.     </td>
    
52.     %(html)s
    
53. </tr>
    
54. 
    
55. 三、依然在HTMLTestRunner.py中，_generate_report_test()函数中，插入html的内容，因为本身初学者，对unittest和htmltestrunner都不怎么了解，观察htmltestrunner的报告发现它能展示脚本执行过程中打印的log文本，所以我强行把自己组装的用于展示图片html代码打印到执行日志中，然后通过关键字截取出来，放置在script后面的位置，也就是刚才新建的picture列（要是觉得最后report中打印的log中加入了这段html影响美观，可以在下面代码中script里删掉~我懒得写了~）
    
56. 
    
57.         script = self.REPORT_TEST_OUTPUT_TMPL % dict(
    
58.             id = tid,
    
59.             output = saxutils.escape(uo+ue),
    
60.         )
    
61. 
    
62.         s = uo+ue
    
63.         html = s[s.find('htmlbegin')+9:s.find('htmlend')]
    
64. 
    
65.         row = tmpl % dict(
    
66.             tid = tid,
    
67.             Class = (n == 0 and 'hiddenRow' or 'none'),
    
68.             style = n == 2 and 'errorCase' or (n == 1 and 'failCase' or 'none'),
    
69.             desc = desc,
    
70.             script = script,
    
71.             html = html,
    
72.             status = self.STATUS[n],
    
73.         )
    
74. 四、在自己封装的手机截屏方法中加入分别将截图名称和截图存放路径插入对应的数组中（也可以使用二维数组），以下分别为打印日志，截屏，组装html的方法，在组装完成的html代码开头和结尾分别叫上htmlbegin和htmlend作为之后截取使用的关键字
    
75. 
    
76. # _*_ coding:utf-8 _*_
    
77. __author__ = 'gjj14453'
    
78. import logging
    
79. import os
    
80. import common.myDate as myDate
    
81. 
    
82. 
    
83. PATH = lambda p: os.path.abspath(
    
84.     os.path.join(os.path.dirname(__file__), p))
    
85. 
    
86. 
    
87. pics = []
    
88. picpath = []
    
89. 
    
90. 
    
91. ##################################
    
92. #  日志
    
93. # 0: debug
    
94. # 1：info
    
95. # 2：warning
    
96. # -1：error
    
97. ###################################
    
98. def mylogger(msg, flag=1):
    
99.     logging.basicConfig(level=logging.INFO,
    
100.                     format='%(asctime)s %(filename)s %(levelname)s %(message)s',
     
101.                     datefmt='%a, %d %b %Y %H:%M:%S',
     
102.                     filename='mylog.log',
     
103.                     filemode='w')
     
104.     console = logging.StreamHandler()
     
105.     console.setLevel(logging.DEBUG)
     
106.     formatter = logging.Formatter('%(name)-12s: %(levelname)-8s %(message)s')
     
107.     console.setFormatter(formatter)
     
108.     logging.getLogger('').addHandler(console)
     
109.     if flag == 0:
     
110.         logging.debug(msg)
     
111.     elif flag == 1:
     
112.         logging.info(msg)
     
113.     elif flag == 2:
     
114.         logging.warning(msg)
     
115.         screenshot()
     
116.     elif flag == -1:
     
117.         logging.error(msg)
     
118.         screenshot()
     
119.     logging.getLogger('').removeHandler(console)
     
120. 
     
121. 
     
122. #  截屏
     
123. def screenshot():
     
124.     dirname = PATH('D:\\MyConfiguration\\gjj14453\\PycharmProjects\\untitled\\list_out' + "/screenshot")
     
125.     os.popen("adb wait-for-device")
     
126.     os.popen("adb shell screencap -p /data/local/tmp/tmp.png")
     
127.     if not os.path.isdir(dirname):
     
128.         os.makedirs(dirname)
     
129.     pic = myDate.timestampname() + ".png"
     
130.     path = dirname + "/" + pic
     
131.     os.popen("adb pull /data/local/tmp/tmp.png " + PATH(path))
     
132.     os.popen("adb shell rm /data/local/tmp/tmp.png")
     
133.     mylogger("截图为：" + pic)
     
134.     pics.append(pic)
     
135.     picpath.append(path)
     
136. 
     
137. 
     
138. def creathtml(path, pic):
     
139.     html = ''
     
140.     if range(len(path)) > 0:
     
141.         for i in range(len(path)):
     
142.             if i == 0:
     
143.                 html = '<a href=' + path[i] + ' target="_blank">' + pic[i] + '</a>'
     
144.             else:
     
145.                 html = html + '<br /><a href=' + path[i] + ' target="_blank">' + pic[i] + '</a>'
     
146.     else:
     
147.         html = ''
     
148.     htmls = 'htmlbegin<td>' + html +'</td>htmlend'
     
149.     return htmls
     
150. 五、之后在case中加入打印html到日志中的语句就OK啦
     
151. 
     
152. class Test(unittest.TestCase):
     
153. 
     
154.     def setUp(self):
     
155.         log.picpath = []
     
156.         log.pics = []
     
157.         pass
     
158. 
     
159.     def tearDown(self):
     
160.         pass
     
161. 
     
162.     def test(self):
     
163.         log.mylogger('aaa', -1)
     
164.         log.mylogger('bbb', -1)
     
165.         log.mylogger(log.creathtml(log.picpath, log.pics))
     
166. 最后，还是希望有高手能给我一个更加简单的插入对应截图的办法~本来想把截图存放的数组传到脚本执行的result中再一一提取，试了几次毫无成果~

复制代码

文章出自：https://testerhome.com/topics/4620

若尘_51

学习一下， 膜拜大侠大神。 顶~   

254761258

最近正好在这里遇到问题了，
想问下楼主
def test(self):
        log.mylogger('aaa', -1)
        log.mylogger('bbb', -1)
        log.mylogger(log.creathtml(log.picpath, log.pics))
这里的‘aaa’，‘bbb’是什么参数，是一个固定的字符串？还是什么？；test方法中的执行用例写哪里呢？