Unittest单元测试框架之unittest构建测试套件

构建测试套件

在实际项目中，随着项目进度的开展，测试类会越来越多，可是直到现在我 们还只会一个一个的单独运行测试类，这在实际项目实践中肯定是不可行的，在 unittest中可以通过测试套件来解决该问题。

测试套件（Test Suite）是由多个测试用例（Test Case）组成的，当然也 可以由多个子测试套件组成

在unittest中，把测试用例加载到测试套件的方式有如下方法：（推荐方式六）

方式一、用unittest.TestSuite()实例化测试套件对象后，内部的addTest()方法对测试类内部的测试案例进行逐一添加。

注：使用addTest()方法，不是以test开头的用例也可以执行

代码示例：

# 1，一条一条添加测试用例
import unittest
 
 
class Test_Case(unittest.TestCase):
    def setUp(self):
        print("我是setup方法：进行测试前的初始化工作")
    def tearDown(self) -> None:  # 表示该方法没有返回值
        print("我是tearDown方法：执行测试后的清除工作")
        print("------------执行结束----------------")
 
    def test_aaa(self):
        print("我是case:test_aaa")
    def test_bbb(self):
        print("我是case:test_bbb")
    def ccc(self):
        print("我是case:ccc")
 
# 测试套件
# 一条一条用例通过addTest方法添加
# 方法一：
# if __name__ == '__main__':
#     suite = unittest.TestSuite()  # 创建测试套件对象
#     suite.addTest(test_case("test_bbb"))  # addTest(类名（“方法名称”）)
#     suite.addTest(test_case("ccc"))  # 不是以test开头的用例也可以执行
#     unittest.main(defaultTest="suite")
 
# 方法二：
def suite():
    suite = unittest.TestSuite()
    suite.addTest(Test_Case("test_bbb"))  # addTest(类名（“方法名称”）)
    suite.addTest(Test_Case("ccc"))
    return suite
if __name__ == '__main__':
    unittest.main(defaultTest="suite")

查看执行结果

方式二、用unittest.TestSuite()实例化测试套件对象后，内部的makeSuite()方法对整个测试类进行添加。

注：使用makeSuite()方法，不是以test开头的方法不会被添加进测试套件中

代码示例：

# 添加整个测试类
import unittest
 
class Test_Case(unittest.TestCase):
    def setUp(self):
        print("我是setup方法：进行测试前的初始化工作")
    def tearDown(self) -> None:  # 表示该方法没有返回值
        print("我的tearDown方法：执行测试后的清除工作")
        print("------------执行结束----------------")
 
    def test_aaa(self):
        print("我的case:test_aaa")
    def test_bbb(self):
        print("我的case:test_bbb")
    def ccc(self):
        print("我的case:ccc")
 
# 能否添加整个测试类？
if __name__ == '__main__':
    # 注意：此时不是以test开头的方法不会被添加进测试套件中
    # 注意：makeSuite 可能需要手动添加
    suite = unittest.TestSuite(unittest.makeSuite(Test_Case))  # 里面是类名
 
    unittest.main(defaultTest="suite")

查看执行结果

方式三、使用TestLoader()方法添加整个测试类

注：unittest.TestLoader().loadTestsFromTestCase() 不是以test开头的方法不会被执行

代码示例：

# 3，添加整个测试类
# unittest中提供一个TestLoader类用于自动创建一个测试类，并且把单个测试自动放入测试集中
# TestLoader 类会自动运行以test开头的测试方法
import unittest
 
class Test_Case1(unittest.TestCase):
    def setUp(self):
        print("我是setup方法：进行测试前的初始化工作")
    def tearDown(self) -> None:  # 表示该方法没有返回值
        print("我是tearDown方法：执行测试后的清除工作")
        print("------------执行结束----------------")
 
    def test_aaa(self):
        print("我是case1:test_aaa")
    def test_bbb(self):
        print("我是case1:test_bbb")
    def ccc(self):
        print("我是case1:ccc")
 
 
class Test_Case2(unittest.TestCase):
    def setUp(self):
        print("我是setup方法：进行测试前的初始化工作")
    def tearDown(self) -> None:  # 表示该方法没有返回值
        print("我是tearDown方法：执行测试后的清除工作")
        print("------------执行结束----------------")
 
    def test_aaa(self):
        print("我是case2:test_aaa")
    def test_bbb(self):
        print("我是case2:test_bbb")
    def test_ccc(self):
        print("我是case2:test_ccc")
 
 
if __name__ == "__main__":
    suite = unittest.TestLoader().loadTestsFromTestCase(Test_Case1) # 填 类名
    unittest.main(defaultTest='suite')

查看执行结果

方式四、添加整个模块下所有的测试类

注：unittest.TestLoader().loadTestsFromName() 不是以test开头的方法不执行

代码示例：

# 4,添加整个模块下所有的测试类
import unittest
 
class Test_Case1(unittest.TestCase):
    def setUp(self):
        print("我是setup方法：进行测试前的初始化工作")
    def tearDown(self) -> None:  # 表示该方法没有返回值
        print("我的tearDown方法：执行测试后的清除工作")
        print("------------执行结束----------------")
 
    def test_aaa(self):
        print("我是case1:test_aaa")
    def test_bbb(self):
        print("我是case1:test_bbb")
    def ccc(self):
        print("我是case1:test_ccc")
 
class Test_Case2(unittest.TestCase):
    def setUp(self):
        print("我是setup方法：进行测试前的初始化工作")
    def tearDown(self) -> None:  # 表示该方法没有返回值
        print("我的tearDown方法：执行测试后的清除工作")
        print("------------执行结束----------------")
 
    def test_aaa(self):
        print("我是case2:test_aaa")
    def test_bbb(self):
        print("我是case2:test_bbb")
    def test_ccc(self):
        print("我是case2:test_ccc")
 
 
if __name__ == "__main__":
    # 通过TestLoader中字符串的方法加载用例；不以test开头的方法不执行
    # 1，增加一个测试用例  模块名称.类名.方法名称
    # suite = unittest.TestLoader().loadTestsFromName("demo11_测试套件4.test_case1.test_aaa")
    # 2,增加整个测试类
    # suite = unittest.TestLoader().loadTestsFromName("demo11_测试套件4.test_case1")
    # 3，增加整个.py模块下所有的测试类
    suite = unittest.TestLoader().loadTestsFromName("demo11_测试套件4")  # 填 py文件名
 
    unittest.main(defaultTest='suite')

查看执行结果

方式五、将多个子测试集进行整合到一个大的测试套件中

代码示例：

# 将多个子测试集进行整合到一个大的测试套件中
import unittest
from unittest框架 import  demo11_测试套件4  # 导入其他模块
from unittest框架 import  demo08_测试套件1  # 导入其他模块
 
 
# 依次创建多个套件
# 1，单个用例套件
suite01 = unittest.TestLoader().loadTestsFromName('demo08_测试套件1.Test_Case.test_aaa')
# 2,整个类
suite02 = unittest.TestLoader().loadTestsFromName("demo08_测试套件1.Test_Case")
# 3,整个.py模块
suite03 = unittest.TestLoader().loadTestsFromName("demo11_测试套件4")
 
main_suite = unittest.TestSuite()  # 主套件
main_suite.addTest(suite01)  # 添加子套件到主套件中
main_suite.addTest(suite02)
main_suite.addTest(suite03)
 
unittest.main(defaultTest="main_suite")

查看执行结果

方式六、使用discover()来实现添加执行整个目录下所有的测试用例  （=====推荐=====）

当测试用例存放在多个不同子目录下，我们用之前的把用例加载到测试集合中的方 式还是不太方便，需要不断的去导入和添加测试用例模块，此时可以通过discover()方法 来实现

方法如下：
discover(start_dir,pattern =’test *.py‘, top_level_dir = None)

start_dir：要测试的模块名或测试用例目录

pattern=”test*.py“：表示用例文件名的匹配原则，例子中匹配文件名为以”test”开头的”.py“文件，星号”*”表示任意多个字符；

top_level_dir=None：测试模块的顶层目录，如果没有顶层目录，默认为None。

该方法通过从指定的开始目录递归到子目录中查找所有测试模块，并返回包含它们的TestSuite对象，只要与模式匹配测试文件和可导入的模块名称才会被加载。

如果一个测试文件的名称符合pattern，会自动查找该文件中派生自TestCase 的类包含的 test 开头的方法作为测试方法

代码示例：

# 使用discover()来实现添加执行整个目录下所有的测试用例
import os
import unittest
import HTMLTestRunner
import time
 
# 获取当前路径
curren_path = os.path.dirname(__file__)
# 获取测试用例目录的路径
case_path = os.path.join(curren_path,"all_case")
 
# 从当前路径回到上一层路径
# case_path = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
 
# 匹配测试用例路径下的所有的测试方法
discover = unittest.defaultTestLoader.discover(start_dir=case_path,  # 用例路径
                                               pattern="*_case*.py",
                                               top_level_dir=None)   # 文件类型
# 创建主套件
main_suite = unittest.TestSuite()
# 把测试用例路径添加到主套件中
main_suite.addTest(discover)
# 执行主套件里的测试用例
unittest.main(defaultTest="main_suite")

查看执行结果

最后感谢每一个认真阅读我文章的人，礼尚往来总是要有的，虽然不是什么很值钱的东西，如果你用得到的话可以直接拿走：

这些资料，对于【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴上万个测试工程师们走过最艰难的路程，希望也能帮助到你！ 

代码007普通

打赏 收藏 海报 链接