Python3学习笔记26-unittest模块

unittest单元测试框架，主要由四部分组成：测试固件、测试用例、测试套件、测试执行器

测试固件(test fixture)

测试固件有两部分，执行测试前的准备部分setUp()，测试执行完后的清扫部分tearDown()

测试用例(test case)

测试用例中，包含测试固件，具体的测试代码的函数。测试固件可以不写，但是至少要有一个以test开头的函数。

unittest会自动识别test开头的函数是测试代码，一定要用小写的test开头！，下面看一个实例

在PyCharm中新建一个项目，在下面建一个包叫做lesson_unittest，然后在这个包下面新建一个.py文件叫做lesson_test_fixture1，然后写下面的代码

import unittest
from io import StringIO
class lesson(unittest.TestCase):

    def setUp(self):
        self.f = StringIO()
        print("创建StringIO")

    def tearDown(self):
        s = self.f.getvalue()
        print("读取StringIO：",s)

    def test_case1(self):
        self.f.write("test case 1")
        print("写入test case 1")

    def test_case2(self):
        self.f.write("test case 2")
        print("写入test case 2")

if __name__ =='__main__':
    unittest.main()

所有方法中一定要带self这个变量

unittest.main()在PyCharm可以不要，如果要在cmd中运行，一定要加上这行

setUp()执行成功之后，不管test中有没有异常发生，tearDown()都会执行

可以看出上面这个实例在执行每个test之前和之后都会去调用setUp()和tearDown()这个方法

很多时候我们只想执行一次测试固件：比如setUp()中放着连接数据库操作，多个test case放着不同数据库操作，然后tearDown()断开数据连接

使用setUpClass()和tearDownClass()可以的得到这样的效果，必须被装饰成classmethod

使用装饰器@unittest.skip可以跳过某个测试用例不执行

在刚才lesson_unittest包下，再新建一个.py文件叫做lesson_test_fixture2，然后写下面的代码

import unittest
from io import StringIO
class lesson(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.f = StringIO()
        print("创建StringIO")

    @classmethod
    def tearDownClass(cls):
        s = cls.f.getvalue()
        print("读取StringIO：",s)

    def test_case1(self):
        self.f.write("test case 1")
        print("写入test case 1")

    def test_case2(self):
        self.f.write("test case 2")
        print("写入test case 2")

if __name__ =='__main__':
    unittest.main()

测试执行器(test runner)

test runner用来执行加载的测试用例，可以是单个测试用例，也可以是测试套件

一种就是上方实例中的直接使用unittest.main()去执行，会搜索所有以test开头的测试用例，按照ASCII的顺序执行多个用例

另一种就是使用TextTestRunner()方法初始化一个测试执行器，然后使用run()去执行测试套件

runner = unittest.TextTestRunner()

runner.run(suite)

这里suite是指测试套件的实例

测试套件(test suite)

很多测试用例的集合就是测试套件，通过测试套件来管理多个测试用例，会根据用例的加载顺序执行用例

使用TestSuite()初始化一个Suite实例

使用addTest()方法加载测试方法到测试套件中，addTest()如果要在传入别的包下的测试方法：  包名.类名(测试方法函数)

suite = unittest.TestSuite()
suite.addTest(lesson('test_case1'))#lesson是类名，test_case1是测试函数
suite.addTest(lesson('test_case2'))#如果是lesson是其他包下的，就是包名.lesson（‘test_case2’）
runner = unittest.TextTestRunner()
runner.run(suite)

将文章第一个示例中最后一行unittest.main()换成上方代码执行

使用makeSuite()可以把一个类下面所有测试方法都加载到测试套件中

目前项目路径应该是这样的，最上面是项目，然后包，最后是.py文件。一定要有__init__.py文件，不然引用不到

在项目下新建一个.py文件，名字随便不要带test，或lesson就行。在项目下新建，不要在包里，不要在包里，不要在包里

import unittest
from lesson_unittest.lesson_test_fixture1 import lesson
suite = unittest.TestSuite(unittest.makeSuite(lesson))
if __name__=='__main__':
    runner = unittest.TextTestRunner()
    runner.run(suite)

可以看到把lesson两个测试用例都执行了，但是使用makeSuite()还是有些繁琐。要把写的所有的类都加到测试套件中

最常用的是使用discover()方法可以把指定路径下所有测试方法都加载到测试套件中

使用刚才实验makeSuite()方法的那个.py文件，清空代码输入以下代码

import unittest
test_dir = 'D:/PycharmProjects/PythonLesson/lesson_unittest'
discover = unittest.defaultTestLoader.discover(test_dir, pattern='lesson*.py')
runner = unittest.TextTestRunner()
runner.run(discover)

test_dir是指定的路径，pattern是指定的文件名。pattern参数是定义文件名匹配规则的，可以随便修改

比如这里lesson*.py是指文件名以lesson开头的文件

也可以写成*test*文件带有test的文件

如果是*lesson.py，那就是指文件名以lesson.py结尾的文件

*起到跟通配符差不多的作用