python学习笔记(十四): unittest

Python中有一个自带的单元测试框架是unittest模块，用它来作单元测试，它里面封装好了一些校验返回的结果方法和一些用例执行前的初始化操做。

在说unittest以前，先说几个概念：

TestCase 也就是测试用例

TestSuite 多个测试用例集合在一块儿，就是TestSuite

TestLoader是用来加载TestCase到TestSuite中的

TestRunner是来执行测试用例的,测试的结果会保存到TestResult实例中，包括运行了多少测试用例，成功了多少，失败了多少等信息

下面写一个简单的单元测试用例

        import unittest
        class MyTest(unittest.TestCase):#继承unittest.TestCase
            def tearDown(self):
                #每一个测试用例执行以后作操做
                print('111')
            def setUp(self):
                #每一个测试用例执行以前作操做
                print(22222)
            def test_run(self):
                self.assertEqual(1,1)
                #测试用例
        
        
        if __name__ == '__main__':
            unittest.main()#运行全部的测试用例

下面是一些经常使用的断言，也就是校验结果

        assertEqual(a, b)     a == b      
        assertNotEqual(a, b)     a != b      
        assertTrue(x)     bool(x) is True      
        assertFalse(x)     bool(x) is False      
        assertIsNone(x)     x is None     
        assertIsNotNone(x)     x is not None   
        assertIn(a, b)     a in b    
        assertNotIn(a, b)     a not in b

那如何生成一个测试报告呢，须要加入另一个模块了，HTMLTestRunner，这个模块须要本身安装，使用执行测试用例就会生成一个html的测试报告，里面会有每一个测试用例的执行结果，代码以下：

        import HTMLTestRunner        
        import unittest
        class MyTest(unittest.TestCase):#继承unittest.TestCase
            def tearDown(self):
                #每一个测试用例执行以后作操做
                print('111')
            def setUp(self):
                #每一个测试用例执行以前作操做
                print(22222)
            def test_run(self):
                # self.assertEqual(1,1)
                self.assertIs(1,1)
                #测试用例
            def test_run2(self):
                # self.assertEqual(1,1)
                self.assertIs(1,1)
                #测试用例
            def test_run3(self):
                # self.assertEqual(1,1)
                self.assertIs(1,1)
                #测试用例
            def test_run1(self):
                # self.assertEqual(1,1)
                self.assertIs(1,1)
                #测试用例
        if __name__ == '__main__':
            test_suite = unittest.TestSuite()#建立一个测试集合
            test_suite.addTest(MyTest('test_run1'))#测试套件中添加测试用例
            #test_suite.addTest(unittest.makeSuite(MyTest))#使用makeSuite方法添加全部的测试方法
            fp = open('res.html','wb')#打开一个保存结果的html文件
            runner = HTMLTestRunner.HTMLTestRunner(stream=fp,title='api测试报告',description='测试状况')
            #生成执行用例的对象
            runner.run(test_suite)
            #执行测试套件

若是咱们有不少个模块，每一个模块下面都写了不少python文件，每一个python文件里面都有测试用例，那怎么把这个目录下的用例都执行了呢，就要先找到这个目录下的全部python文件，而后找到里面的测试用例，逐个执行，代码以下：

       import unittest,HTMLTestRunner
        suite = unittest.TestSuite()#建立测试套件
        all_cases = unittest.defaultTestLoader.discover('.','test_*.py')
        #找到某个目录下全部的以test开头的Python文件里面的测试用例
        for case in all_cases:
            suite.addTests(case)#把全部的测试用例添加进来
        fp = open('res.html','wb')
        runner = HTMLTestRunner.HTMLTestRunner(stream=fp,title='all_tests',description='全部测试状况')
        runner.run(suite)
        #运行测试

咱们在后续进行持续集成的时候，要让代码自动运行，就会用到Jenkins了，可是上面产生的测试报告都是html格式的，Jenkins不认识，就在Jenkins里面显示不出来。那我们就要产生一些Jenkins认识的测试报告，Jenkins认识xml格式的报告，那我们就产生xml格式的呗，就须要用一个新的模块，xmlrunner，安装直接 pip install xmlrunner便可，代码以下：

import unittest
import xmlrunner
#导入这个模块
class My(unittest.TestCase):
 
    def test1(self,a,b,c):
        self.assertEqual(a+b,c)
 
if __name__=='__main__':
    test_suite = unittest.TestSuite()
    test_suite.addTest(unittest.makeSuite(My))
    runner = xmlrunner.XMLTestRunner(output='report')#指定报告放的目录
    runner.run(test_suite)