python + unittest 做单元测试之学习笔记

单元测试在保证开发效率、可维护性和软件质量等方面有很重要的地位，所谓的单元测试，就是对一个类，一个模块或者一个函数进行正确性检测的一种测试方式。

这里主要是就应用 python + unitest 做单元测试做的一些学习笔记。

Unittest 提供了test cases, test suites, test fixtures，test runner：

• test case ：通过继承TestCase类，我们可以创建一个test，或者一组tests
• test suites ： 测试套件, 使用时要将test case 装载到测试套件里
• test fixtures ： setup + test case + teardown，每个test case执行的时候都会运行setUp->test_case->tearDown ，因此setUp 和 tearDown 可以用来做环境的初始化。例如，写一个有关数据操作的类，setUp放连接数据库，tearDown 放关闭数据库连接。

一、unittest.main() 的使用

所有的test case 要以test开头，例如 def test_pass_case(self)

具体的单元测试例子如下：

import unittest
def calc(x,y):
    return x + y

class TestCalc(unittest.TestCase):
    def test_pass_case(self):
        '''这是通过的测试用例'''
        print('这个是通过用例')
        res = calc(1,2)
        self.assertEqual(3,res)
        self.assertNotEqual(2,3)
    def setUp(self):#每个用例执行前都会执行它
        print('我是setUp')
    def tearDown(self):#每个用例运行之后都会执行它
        print('我是tearDown')

    @classmethod
    def setUpClass(cls):#所有的用例执行之前执行一次，setUp()前
        print('我是setupClass')#所有的用例都运行完之后执行一次，setDown 后
    @classmethod
    def tearDownClass(cls):#所有的用例执行之后执行一次，setDown()后
        print('我是teardownClass')
    def test_a(self):
        '''这是普通的测试用例'''
        print('a')
    def test_fail_case(self):
        '''这是失败的测试用例'''
        print('这个是失败的测试用例')
        res = calc(9,8)
        self.assertEqual(98,res)
    def test_haha(self):
        '''这是哈哈'''
        self.assertEqual(1,2)

if __name__ == '__main__':
    unittest.main()#它会帮你运行当前这个python里面所有的测试用例

由此可以看到，执行顺序是：静态方法 setUpClass()  -> setUp -> test case -> SetDown ->... -> SetUp -> testcase -> testDown -> 静态方法 setDownClass()。

所以可以用setUpClass、tearDownClass做所有用例的前置初始化和结束。

unittest.main()  #它会帮你运行当前这个python里面所有的测试用例

二、 应用 HTMLTestRunner 生成测试报告

if __name__ == '__main__':
    suite = unittest.TestSuite()#定义一个测试套件
    suite.addTest(TestCalc('test_pass_case'))#测试套件中添加测试用例
    suite.addTest(TestCalc('test_a'))
    f = open('report.html','wb')#打开一个保存结果的html文件
    runner = HTMLTestRunner.HTMLTestRunner(stream=f,title='测试报告',description='测试情况')#生成执行用例的对象
    runner.run(suite)#运行

可以将某几个test case加入test suite ，这样生成的测试报告里只有这两个test case 的结果。

具体如下：

可以将类里所有的test case加入test suite，如下

suite.addTests(unittest.makeSuite(TestCalc))#把类里面所有的测试用例加入测试套件

得到的测试报告结果为：

三、针对整个项目的test case，生成测试报告

例如有以下的结构：

test_buy 里关于购买的cases， test_login是关于login 的cases， test_reg是关于注册的cases， run_all_cases执行所有的cases

注意，在test_buy， test_login，test_reg里的case 需要以test 开头，这样才会被执行到，还有这三个文件不一定以test开头，这里只是为了方便才这么写的，但是必须得有相同的规则，比如这里的文件名都以test_ 开头。

run_all_cases.py代码如下所示：

import unittest,HTMLTestRunner
suite = unittest.TestSuite()
all_cases = unittest.defaultTestLoader.discover('.','test_*.py')#第一个参数：从哪个目录，第二个参数：定义规则
# for case in all_cases:
#     suite.addTests(case)
[suite.addTests(case) for case in all_cases]#列表生成式，和上面的注释写法一样
f = open('new_report.html','wb')#打开一个保存结果的html文件
runner = HTMLTestRunner.HTMLTestRunner(stream=f,title='测试报告',description='测试情况')#生成执行用例的对象
runner.run(suite)#运行

四、集成jenkins，生成jenkins能识别的测试报告

这里就要用到  xmlrunner 模块，jenkins能识别xml格式。

runner = xmlrunner.XMLTestRunner('.')#当前目录

 以上就是对Unittest的简单总结，后续会持续更新。