Python实战之unittest使用详解

一 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 unittest单元框架的视频讲解，请参看：http://i.youku.com/weiworld521第 23 节。

作者：平山_Python
链接：https://www.jianshu.com/p/e0f74e54290d
來源：简书
简书著作权归作者所有，任何形式的转载都请联系作者获得授权并注明出处。