Python实战之unittest使用详解

来源：平山

链接：http://blog.51cto.com/2681882/2123613

一 unittest是什么？

unittest是python内置的单元测试框架，具备编写用例、组织用例、执行用例、

输出报告等自动化框架的条件。使用unittest前需要了解该框架的五个概念:即test case,test suite,testLoader，test runner,test fixture。

test case ：一个完整的测试单元，执行该测试单元可以完成对某一个问题的

验证，完整体现在：测试前环境准备(setUp)，执行测试代码(run)，

以及测试后环境 还原(tearDown)；test suite ：多个测试用例的集合，测试套件或测试计划；testLoader ：加载TestCase到TestSuite中的，其中loadTestsFrom__()方法用于

寻找TestCase，并创建它们的实例，然后添加到TestSuite中，

返回TestSuite实例；test runner ：执行测试用例，并将测试结果保存到TextTestResult实例中，包括

运行了多少测试用例，成功了多少，失败了多少等信息；test fixture：一个测试用例的初始化准备及环境还原，主要是setUp()和

setDown()方法；

二 unitest的工作原理

通过unittest类调用分析，可将框架的工作流程概况如下：

编写TestCase，由TestLoader加载TestCase到TestSuite，然后由TextTestRunner

来运行TestSuite，最后将运行的结果保存在TextTestResult中。

三 unittest实战举例

了解了unittest框架的五大概念和工作流程后，接下来举例如何使用该框架对模块进行

单元测试。

3.1 待测模块myfunc

myfunc包含三个函数：isprime(),add(),divide(),如下：

3.2 unitest编写测试用例

使用unittest对myfunc进行单元测试，首先需要导入unitest框架和待测模块myfunc，

定义的测试用例方法类，需要继承unittest.TestCase,且测试用例方法是以test开头

作为标识，用例的执行结果以assetxxx断言结果决定，如果断言返回为false，

将抛出assetError异常。

测试用例代码如下：

一些常用的断言方法如下，更多断言方法请查看unittest自带文档：

从上面的设计的测试用例执行结果及对自动化测试的要求，需要考虑以下4个问题，及

给出unitest框架中的解决方法。

3.3 框架如何解决自动化需求的4个问题

3.3.1 问题1： 如何控制用例执行顺序

在unittest中，用例是以test开头的方法定义的，默认执行顺序是根据用例名称

升序进行，如上面的用例，实际执行顺序为：test_add-->test_divide-->test_is_prime,

而不是用例定义的先后顺序。在unittest中解决用例执行顺序的问题是使用TestSuite，代码如下：

3.3.2 问题2：如何让多个用例共用setup、teardown

unittest的setup、teardown会在每个用例执行前后执行一次，如上面测试用例类中

有3个测试用例，那么每个用例执行前会执行setup，执行后会执行teardown，即setup、

teardown总共会调用三次，但考虑实际自动化测试场景，多个用例只需执行一次setup，

全部用例执行完成后，执行一次teardown，针对该种场景，unittest的处理方法是使用

setupclass、teardownclass，注意@classmethod的使用，如下：

3.3.3 问题3：如何跳过用例

在自动化测试中，经常会遇到挑选用例的情况，在unittest中的解决方法是使用skip

装饰器，其中skip装饰器主要有3种：unittest.skip(reason)、unittest.

skipIf(condition,reason)、unittest.skipUnless(condition,reason)，即在满足

condition条件下跳过该用例，reason用于描述跳过的原因，实例代码如下：

3.3.4 问题4：如何生成html格式的测试报告

Unittest中默认生成的报告格式为txt，如果想生成html格式的报告，可以使用

HtmlTestRunner模块，安装后导入该模块，使用HTMLTestRunner代替默认的

TextTestRunner()执行测试用例即可。

实例代码如下：

看完本文有收获？请转发分享给更多人

关注 【小蟒社区】，一起学Python