HTMLTestRunner
生成测试报告时,报告的标题在运行前就已经写死在代码了,假如我现在需要在执行完毕后,根据执行结果,把执行的状态写在标题里面,类似的效果如图:
如果有一条执行错误的,就在后面加ERROR
,如果有一条执行失败的,就在后面加FAIL
,如果都通过,就在后面加PASS
要达到这个效果,我们就要研究,HTMLTestRunner
里面是怎么生成报告的标题的
在HTMLTestRunner
类的run()
方法里面,有一句self.generateReport(test, result)
def run(self, test):
"Run the given test case or test suite."
result = _TestResult(self.verbosity)
test(result)
self.stopTime = datetime.datetime.now()
self.generateReport(test, result)
print('\nTime Elapsed: %s' % (self.stopTime-self.startTime), file=sys.stderr)
return result
很显然,generateReport
是生成测试报告的方法,继续研究这个方法
def generateReport(self, test, result):
report_attrs = self.getReportAttributes(result)
generator = 'HTMLTestRunner %s' % __version__
stylesheet = self._generate_stylesheet()
heading = self._generate_heading(report_attrs)
report = self._generate_report(result)
ending = self._generate_ending()
output = self.HTML_TMPL % dict(
title = saxutils.escape(self.title),
generator = generator,
stylesheet = stylesheet,
heading = heading,
report = report,
ending = ending,
)
self.stream.write(output.encode('utf8'))
我们分别研究这里面的几个self
的方法,发现self._generate_heading
很可疑
def _generate_heading(self, report_attrs):
a_lines = []
for name, value in report_attrs:
line = self.HEADING_ATTRIBUTE_TMPL % dict(
name = saxutils.escape(name),
value = saxutils.escape(value),
)
a_lines.append(line)
heading = self.HEADING_TMPL % dict(
title = saxutils.escape(self.title),
parameters = ''.join(a_lines),
description = saxutils.escape(self.description),
)
return heading
看源码,这确实是处理报告头部信息的方法,包括标题,描述,测试结果等等,那么就可以改造了。现在我们需要根据测试结果,来重新定义标题,而该方法中没有测试结果相关的内容,所以我们把测试结果当成参数传进来,而在这里面,测试结果都是result
,所以这里可以改造为:
def _generate_heading(self, report_attrs, result):
a_lines = []
for name, value in report_attrs:
line = self.HEADING_ATTRIBUTE_TMPL % dict(
name=saxutils.escape(name),
value=saxutils.escape(value),
)
a_lines.append(line)
# 在标题中显示当前测试任务的结果
if result.error_count > 0:
self.title = self.title + ' --[ERROR]'
elif result.failure_count > 0:
self.title = self.title + ' --[FAIL]'
else:
self.title = self.title + ' --[PASS]'
heading = self.HEADING_TMPL % dict(
title=saxutils.escape(self.title),
parameters=''.join(a_lines),
description=saxutils.escape(self.description)
)
return heading
那么相应的generateReport
方法中
heading = self._generate_heading(report_attrs)
需要修改为
heading = self._generate_heading(report_attrs, result)
再次执行测试任务,就可以根据测试任务的结果显示标题了