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)
再次执行测试任务，就可以根据测试任务的结果显示标题了
