第 7 章 unittest 扩展

在第 6 章中，我们介绍了 unittest 的主要功能，但是如果只用它来写 Web 自动化测试，则仍稍显不足。例如，它不能生成 HTML 格式的报告、它不能提供参数化功能等。不过，我们可以借助第三方扩展来弥补这些不足。

7.1 HTML 测试报告

HTMLTestRunner 是 unittest 的一个扩展，它可以生成易于使用的 HTML 测试报告。HTMLTestRunner 是在 BSD 许可证下发布的。
下载地址：http://tungwaiyip.info/software/HTMLTestRunner.html

因为该扩展不支持 Python 3，所以做了一些修改，使它可以在 Python 3 下运行。
GitHub 地址：https://github.com/defnngj/HTMLTestRunner

7.1.2 生成 HTML 测试报告

如果想用 HTMLTestRunner 生成测试报告，那么请查看本书 6.1.4 节 run_tests.py 文件的实现。测试用例的执行是通过 TextTestRunner 类提供的 run()方法完成的。这里需要把 HTMLTestRunner.py 文件中的 HTMLTestRunner 类替换 TextTestRunner 类。
打开 HTMLTestRunner.py 文件，在第 877 行（如果代码更新，则行号会发生变化）可以找到 HTMLTestRunner 类。

使用Sublime打开.py文件，查找关键字“HTMLTestRunner”

这段代码是 HTMLTestRunner 类的部分实现，主要看__init__()初始化方法的参数。

● stream：指定生成 HTML 测试报告的文件，必填。
● verbosity：指定日志的级别，默认为 1。如果想得到更详细的日志，则可以将参数修改为 2。
● title：指定测试用例的标题，默认为 None。
● description：指定测试用例的描述，默认为 None。

在 HTMLTestRunner 类中，同样由 run()方法来运行测试套件中的测试用例。修改 run_tests.py 文件如下。

import unittest
from HTMLTestRunner import HTMLTestRunner


    # 定义测试用例的目录为当前目录
    test_dir = './test_case'
    suit = unittest.defaultTestLoader.discover(test_dir, pattern='test*.py')

if __name__ == '__main__':


    # 取当前日期时间
    now_time = time.strftime("%Y-%m-%d %H_%M_%S")
    html_report = './test_report/' + now_time + 'result.html'
    fp = open(html_report, 'wb')
    # 调用HTMLTestRunner，运行测试用例
    runner = HTMLTestRunner(stream=fp,
                            title="百度搜索测试报告",
                            description="运行环境：Windows 10, Chrome浏览器"
                            )
    runner.run(suit)
    fp.close()

View Code

首先，使用 open()方法打开 result.html 文件，用于写入测试结果。如果没有 result.html 文件，则会自动创建该文件，并将该文件对象传给 HTMLTestRunner 类的初始化参数 stream。然后，调用 HTMLTestRunner 类中的 run()方法来运行测试套件。最后，关闭 result.html 文件。

打开/test_report/result.html 文件，将会得到一张 HTML 格式的报告。HTMLTestRunner 测试报告如图。