unittest框架与自动化测试环境的搭建

1、unittest框架简介

首先以selenium IDE录制的一段代码为例（脚本的录制请参考其他文档），说明unittest框架的使用。

图 1

要使用unittest框架，首先需要import unittest。上述代码新建了一个Baidu类，该类继承自unittest.TestCase类。类中，setUp函数完成初始化工作，tearDown函数完成清理工作，这两个函数在以后编写测试用例时，可以为空。类中所有以test开头的函数为测试用例（unittest框架已通过代码约定好），如上图红色矩形框所示。

通过执行unittest.main()，自动完成所有以test开头的测试用例的执行。

2、unittest.TestSuite()容器

多数时候，一项测试可能包含多个测试用例。可以将多个测试用例加入unittest.TestSuite()容器，组成测试用例集，作为测试的入口。下面一个简单的例子说明TestSuite容器的使用：

图 2

上述定义了两个运算，定义了一个MyTest的类对两个运算进行测试。通过TestSuite容器将两个test测试用例联合起来，并通过TextTestRunner执行。执行结果如下：

图 3

以上的只是一个简单的例子。利用 selenium IDE生成的测试脚本同样适用。假如有多个 .py的测试文件，可以建立 test_all.py文件，将每个 .py里的测试类 import进来，利用 TestSuite容器将每个 test类里的测试用例（ test开头的类成员函数）汇总，通过 runner逐个执行。此处，如果某个测试用例测试失败，并不影响其他的测试用例。

3、测试报告的生成（使用HTMLTestRunner.py）

以上介绍了如何执行测试用例及将多个测试用例加入容器一次执行，执行结果打印呈现。接下来讲述如何将测试结果生成漂亮的测试报告。供他人阅读。

HTMLTestRunner.py是Python标准库的unittest模块的一个扩展，它可以生成易于使用的HTML测试报告。下载后放入%path%/Lib下即可使用。

要使用HTMLTestRunner.py，首先要import HTMLTestRunner（如出现import错误，说明HTMLTestRunner.py的路径有问题）。将之前的runner改为HTMLTestRunner的对象，并新建文件指针fp，作为生成的报告文件。代码如下：

图 4

与2中的代码对比可以看出，改动很少，仅图中红色矩形框部分。运行生成的报告在代码的当前目录（通过修改 filename字符串，也可以让报告生成在其他文件夹下，从而使目录结构更合理）。以上代码的运行报告如下图：

图 5

4、测试报告细节

在测试用例中加入注释可以使测试报告更容易阅读。另外可以通过调用time相关模块，使测试报告的文件名自动带有测试时间的标签。

对于3中的代码，可以作如下修改（图中红色矩形框）：

运行后生成的报告名为：

报告内容为：

5、结构改进

一个好的测试结构应该是如下图所示，测试用例、数据、报告、all_case.py分离。test_case下存放具体的测试用例。

对于上述的结构，可以通过for循环将所有测试用例加入TestSuite容器，首先需要将test_case文件夹加入当前的path中，如下：

可以在all_case.py的目录下创建allcase_list.py，作为读取所有case的文件，allcase_list.py内容为：

此时只要在all_case.py里加入如下几行即可以读取所有的测试用例到数组alltestnames中：

通过TestSuite容器汇总所有的测试用例：

接下来只要声明runner并执行即可。

6、自动发邮件功能

将生成的测试报告自动发送到指定邮箱，以邮箱正文的方式显示，方便boss查看报告，而不用逐个打开测试报告文件。python的smtplib模块提供了邮件的功能。以下主要介绍如何将测试报告以HTML形式发送到邮件。

（1）首先导入smtplib模块：import smtplib

再导入email模块，MIMEText和Header主要用来完成邮件内容与邮件标题的定义。

from email.mime.text import MIMEText
from email.header import Header

（2）新建sendEmail.py文件，并在里边定义发送邮件的函数。sendEmail.py的内容如下：

代码包括两个函数，sendEmail(fileName)实现邮件发送，传入的参数是生成的测试报告名。sendReport()首先查找在报告目录下查找最新的测试报告，并调用sendEmail(fileName)将报告发出。

对于第4节中的代码，只要在开头import sendEmail，并在代码最后调用发送邮件的函数即可。如下：

最终的执行打印结果为：

到XX邮箱里查看，可以发现收到一封来自XXX的邮件，内容为：

clipboard17.png

• 以上是一个比较简单的自动化测试环境的搭建，可以以此为基础搭建复杂的、灵活的测试环境。