python之Unittest单元测试框架-低调大师

python之Unittest单元测试框架

2018-04-22 622

前言

unittest是一个python版本的junit，junit是java中的单元测试框架，对java的单元测试，有一句话很贴切：Keep the bar green，相信使用eclipse写过java单元测试的都心领神会。unittest实现了很多junit中的概念，作为标准python中的一个模块，是其它框架和工具的基础，参考资料是它的官方文档：http://docs.python.org/2.7/library/unittest.html和源代码，比如我们非常熟悉的test case, test suite等，总之，原理都是相通的，只是用不同的语言表达出来。

一、unittest工作原理

unittest中最核心的四个概念是：test case, test suite, test runner, test fixture。

工作原理

一个TestCase的实例就是一个测试用例。什么是测试用例呢？就是一个完整的测试流程，包括测试前准备环境的搭建(setUp)，执行测试代码(run)，以及测试后环境的还原(tearDown)。元测试(unit test)的本质也就在这里，一个测试用例是一个完整的测试单元，通过运行这个测试单元，可以对某一个问题进行验证。

而多个测试用例集合在一起，就是TestSuite，而且TestSuite也可以嵌套TestSuite。

TestLoader是用来加载TestCase到TestSuite中的，其中有几个loadTestsFrom__()方法，就是从各个地方寻找TestCase，创建它们的实例，然后add到TestSuite中，再返回一个TestSuite实例。

TextTestRunner是来执行测试用例的，其中的run(test)会执行TestSuite/TestCase中的run(result)方法。测试的结果会保存到TextTestResult实例中，包括运行了多少测试用例，成功了多少，失败了多少等信息。

而对一个测试用例环境的搭建和销毁，是一个fixture。

一个class继承了unittest.TestCase，便是一个测试用例，但如果其中有多个以 test 开头的方法，那么每有一个这样的方法，在load的时候便会生成一个TestCase实例，如：一个class中有四个test_xxx方法，最后在load到suite中时也有四个测试用例。

到这里整个流程就清楚了：

写好TestCase，然后由TestLoader加载TestCase到TestSuite，然后由TextTestRunner来运行TestSuite，运行的结果保存在TextTestResult中，我们通过命令行或者unittest.main()执行时，main会调用TextTestRunner中的run来执行，或者我们可以直接通过TextTestRunner来执行用例。这里加个说明，在Runner执行时，默认将执行结果输出到控制台，我们可以设置其输出到文件，在文件中查看结果（通过HTMLTestRunner将结果输出到HTML中，生成漂亮的报告，它跟TextTestRunner是一样的）。

二、unittest实例-test case

1.准备测试方法

mathfunc.py

#coding:utf-8

import math

def add(a,b):

    return a + b

def minus(a,b):

    return a - b

def multi(a,b):

    return a * b

def divide(a,b):

    return a / b

2.为测试方法写测试用例

run_mathfunc.py

#coding:utf-8

import unittest

from python_ceshikuangjia.mathfuncimport *

class TestMathFunc(unittest.TestCase):

    def test_add(self):

        self.assertEqual(5,add(3.2))

        self.assertNotEqual(3,add(2,2))

    def test_minus(self):

        self.assertEqual(2,minus(4,2))

    def test_multi(self):

        self.assertEqual(6,multi(2,3))

    def test_divide(self):

        self.assertEqual(2,divide(6,3))

        self.assertEqual(2.5,divide(5,2))

if __name__ =='__main__':

    unittest.main()

3.查看运行结果

success

这就是一个简单的测试，有几点需要说明的：

>在第一行给出了每一个用例执行的结果的标识，成功是 .，失败是 F，出错是 E，跳过是 S。从上面也可以看出，测试的执行跟方法的顺序没有关系，test_divide写在了第4个，但是却是第2个执行的。

>每个测试方法均以 test 开头，否则是不被unittest识别的。

>在unittest.main()中加 verbosity 参数可以控制输出的错误报告的详细程度，默认是 1，如果设为 0，则不输出每一用例的执行结果，即没有上面的结果中的第1行；如果设为 2，则输出详细的执行结果，如下：

详细结果输出

三、组织TestSuite

上面的代码示例了如何编写一个简单的测试，但有两个问题，我们怎么控制用例执行的顺序呢？（这里的示例中的几个测试方法并没有一定关系，但之后你写的用例可能会有先后关系，需要先执行方法A，再执行方法B），我们就要用到TestSuite了。我们添加到TestSuite中的case是会按照添加的顺序执行的。

问题二是我们现在只有一个测试文件，我们直接执行该文件即可，但如果有多个测试文件，怎么进行组织，总不能一个个文件执行吧，答案也在TestSuite中。

请看run_suite.py

#coding:utf-8

import unittest

from python_ceshikuangjia.run_mathfuncimport TestMathFunc

if __name__ =='__main__':

    suite = unittest.TestSuite()

    tests = [TestMathFunc("test_add"),TestMathFunc("test_minus"),TestMathFunc("test_divide")]

    suite.addTests(tests)

    runner = unittest.TextTestRunner(verbosity=2)

    runner.run(suite)

运行结果：

运行结果

可以看到，执行情况跟我们预料的一样：执行了三个case，并且顺序是按照我们添加进suite的顺序执行的。那么，如何将结果输出到文件呢，请看下面操作方法，修改run_suite.py代码，如下：

#coding:utf-8

import unittest

from python_ceshikuangjia.run_mathfuncimport TestMathFunc

if __name__ =='__main__':

    suite = unittest.TestSuite()

    tests = [TestMathFunc("test_add"),TestMathFunc("test_minus"),TestMathFunc("test_divide")]

    suite.addTests(tests)

    with open('D:/work/python_ceshikuangjia/run_suite_log.txt','a')as f:

        runner = unittest.TextTestRunner(stream=f,verbosity=2)

        runner.run(suite)

运行结果

四、test fixture之setUp() tearDown()

1.假如我的测试需要在每次执行之前准备环境，或者在每次执行完之后需要进行一些清理怎么办？比如执行前需要连接数据库，执行完成之后需要还原数据、断开连接。总不能每个测试方法中都添加准备环境、清理环境的代码吧，这时，就轮到test fixture之setUp() tearDown()大展身手的时候了，请看如下代码：在run_mathfunc.py下的class TestMathFunc类中添加如下代码

class TestMathFunc(unittest.TestCase):

    def setUp(self):

        print("do something before test.Prepare environment.")

    def tearDown(self):

        print("do something after test.Clean up.")

setUp() 和 tearDown() 两个方法（其实是重写了TestCase的这两个方法），这两个方法在每个测试方法执行前以及执行后执行一次，setUp用来为测试准备环境，tearDown用来清理环境，已备之后的测试。

运行结果如下：

运行结果

可以看到setUp和tearDown在每次执行case前后都执行了一次。

2.如果想要在所有case执行之前准备一次环境，并在所有case执行结束之后再清理环境，我们可以用 setUpClass() 与 tearDownClass():请看如下代码：在run_mathfunc.py下的class TestMathFunc类中添加如下代码

class TestMathFunc(unittest.TestCase):

    @classmethod

def setUpClass(cls):

        print("This setUpClass() method only called once.")

    @classmethod

def tearDownClass(cls):

        print("This tearDownClass() method only called once too.")

运行结果：

运行结果

可以看到setUpClass以及tearDownClass均只执行了一次。

3.运行测试用例时不想全部运行，或者说想跳过某一个用例，那么这时skip装饰器就起作用了。

skip装饰器一共有三个 unittest.skip(reason)、unittest.skipIf(condition, reason)、unittest.skipUnless(condition, reason)，skip无条件跳过，skipIf当condition为True时跳过，skipUnless当condition为False时跳过。以下分两种情况进行解析。

>skip装饰器

class TestMathFunc(unittest.TestCase):

    @classmethod

def setUpClass(cls):

        print("This setUpClass() method only called once.")

    @classmethod

def tearDownClass(cls):

        print("This tearDownClass() method only called once too.")

@unittest.skip(u"我不想运行此用例！！.")

    def test_add(self):

        self.assertEqual(5,add(3,2))

        self.assertNotEqual(3,add(2,2))

运行结果：

运行结果

>TestCase.skipTest()方法

class TestMathFunc(unittest.TestCase):

    @classmethod

def setUpClass(cls):

        print("This setUpClass() method only called once.")

    @classmethod

def tearDownClass(cls):

        print("This tearDownClass() method only called once too.")

# @unittest.skip(u"我不想运行此用例！！.")

def test_add(self):

        self.assertEqual(5,add(3,2))

        self.assertNotEqual(3,add(2,2))

    def test_minus(self):

        self.skipTest(u"我不想运行此用例！！")

        self.assertEqual(2,minus(4,2))

运行结果：

运行结果

通过以上两种不同方式，可以看到总的test数量还是3个，但add()和minus()方法都被skip了。

五、用HTMLTestRunner输出HTML报告

HTMLTestRunner是一个第三方的unittest HTML报告库，首先我们下载HTMLTestRunner.py，并放到当前目录下，或者你的’python’安装目录下，就可以导入运行了。

下载地址：HTMLTestRunner模板（下载的模板只支持python2.x，要想在python3.x中使用可以看下这个：HTMLTestRunner修改成Python3版本）

修改我们的 run_suite.py：

#coding:utf-8

import unittest

import HTMLTestRunner

from python_ceshikuangjia.run_mathfuncimport TestMathFunc

if __name__ =='__main__':

    suite = unittest.TestSuite()

    tests = [TestMathFunc("test_add"),TestMathFunc("test_minus"),TestMathFunc("test_divide"),TestMathFunc('test_multi')]

    suite.addTests(tests)

# with open('D:/work/python_ceshikuangjia/run_suite_log.txt','a') as f:

# runner = unittest.TextTestRunner(stream=f,verbosity=2)

# runner.run(suite)

#输出HTML格式报告

with open('D:/work/python_ceshikuangjia/HTMLReport.html','wb')as f:

        runner = HTMLTestRunner.HTMLTestRunner(stream=f,

title=u'软件测试报告 Test Report',

description=u'用例执行情况',

verbosity =2)

        runner.run(suite)

运行结果：

运行结果1

运行结果2

这下漂亮的HTML报告也有了。其实你能发现，HTMLTestRunner的执行方法跟TextTestRunner很相似，你可以跟我上面的示例对比一下，就是把类图中的runner换成了HTMLTestRunner，并将TestResult用HTML的形式展现出来，如果你研究够深，可以写自己的runner，生成更复杂更漂亮的报告。

单元测试小结：

1.unittest是Python自带的单元测试框架，我们可以用其来作为我们自动化测试框架的用例组织执行框架。

2.unittest的流程：写好TestCase，然后由TestLoader加载TestCase到TestSuite，然后由TextTestRunner来运行TestSuite，运行的结果保存在TextTestResult中，我们通过命令行或者unittest.main()执行时，main会调用TextTestRunner中的run来执行，或者我们可以直接通过TextTestRunner来执行用例。

3.项目命名不可用小写‘test’开头（大写无影响），否则会出错，一个class继承unittest.TestCase即是一个TestCase，其中以 test 开头的方法在load时被加载为一个真正的TestCase。

4.verbosity参数可以控制执行结果的输出，0 是简单报告、1 是一般报告、2 是详细报告。

5.可以用 setUp()、tearDown()、setUpClass()以及 tearDownClass()可以在用例执行前布置环境，以及在用例执行后清理环境

6.我们可以通过skip，skipIf，skipUnless装饰器跳过某个case，或者用TestCase.skipTest方法。

7.参数中加stream，可以将报告输出到文件：可以用TextTestRunner输出txt报告，以及可以用HTMLTestRunner输出html报告，或者自己研究生成更复杂更漂亮的报告。

微信关注我们

原文链接：https://yq.aliyun.com/articles/664269

转载内容版权归作者及来源网站所有！

低调大师中文资讯倾力打造互联网数据资讯、行业资源、电子商务、移动互联网、网络营销平台。持续更新报道IT业界、互联网、市场资讯、驱动更新,是最及时权威的产业资讯及硬件资讯报道平台。

Android基础混淆

#---------------------------------实体类--------------------------------- -keep class com.yuyigufen.hymanager.beans.**{*;} # 自定义数据模型的bean目录 #---------------------------------基本指令区---------------------------------- -optimizationpasses 5 -dontusemixedcaseclassnames -dontskipnonpubliclibraryclasses -dontskipnonpubliclibraryclassmembers -dontpreverify -verbose -printmapping proguardMapping.txt -optimizations !code/simplification/cast,!field/*,!class/merging/* -keepattributes *Annotation*,InnerClasses -...

2018-04-23

571

前言不小心就鸽了几天没有更新了，这个星期回家咯。在学校的日子要努力一点才行! 只有光头才能变强回顾前面：多线程三分钟就可以入个门了！ Thread源码剖析本文章的知识主要参考《Java并发编程实战》这本书的前4章，这本书的前4章都是讲解并发的基础的。要是能好好理解这些基础，那么我们往后的学习就会事半功倍。当然了，《Java并发编程实战》可以说是非常经典的一本书。我是未能完全理解的，在这也仅仅是抛砖引玉。想要更加全面地理解我下面所说的知识点，可以去阅读一下这本书，总的来说还是不错的。首先来预览一下《Java并发编程实战》前4章的目录究竟在讲什么吧：第1章　简介 1.1　并发简史 1.2　线程的优势 1.2.1　发挥多处理器的强大能力 1.2.2　建模的简单性 1.2.3　异步事件的简化处理 1.2.4　响应更灵敏的用户界面 1.3　线程带来的风险 1.3.1　安全性问题 1.3.2　活跃性问题 1.3.3　性能问题 1.4　线程无处不在 ps：这一部分我就不讲了，主要是引出我们接下来的知识点，有兴趣的同学可翻看原书~ 第2章　线程安全性 2.1　什么是线程安全性 2.2　原...

2018-04-23

521

资源下载

更多资源

优质分享App

近一个月的开发和优化，本站点的第一个app全新上线。该app采用极致压缩，本体才4.36MB。系统里面做了大量数据访问、缓存优化。方便用户在手机上查看文章。后续会推出HarmonyOS的适配版本。

Nacos

Nacos /nɑ:kəʊs/ 是 Dynamic Naming and Configuration Service 的首字母简称，一个易于构建 AI Agent 应用的动态服务发现、配置管理和AI智能体管理平台。Nacos 致力于帮助您发现、配置和管理微服务及AI智能体应用。Nacos 提供了一组简单易用的特性集，帮助您快速实现动态服务发现、服务配置、服务元数据、流量管理。Nacos 帮助您更敏捷和容易地构建、交付和管理微服务平台。

Spring

Spring框架（Spring Framework）是由Rod Johnson于2002年提出的开源Java企业级应用框架，旨在通过使用JavaBean替代传统EJB实现方式降低企业级编程开发的复杂性。该框架基于简单性、可测试性和松耦合性设计理念，提供核心容器、应用上下文、数据访问集成等模块，支持整合Hibernate、Struts等第三方框架，其适用范围不仅限于服务器端开发，绝大多数Java应用均可从中受益。

Rocky Linux

Rocky Linux（中文名：洛基）是由Gregory Kurtzer于2020年12月发起的企业级Linux发行版，作为CentOS稳定版停止维护后与RHEL（Red Hat Enterprise Linux）完全兼容的开源替代方案，由社区拥有并管理，支持x86_64、aarch64等架构。其通过重新编译RHEL源代码提供长期稳定性，采用模块化包装和SELinux安全架构，默认包含GNOME桌面环境及XFS文件系统，支持十年生命周期更新。