测试小白简单聊聊关于UnitTest（二）

lsekfe

四、TestSuite-测试套件
　　首先新建一个UnitTest类，编写[url=]测试[/url]用例，文件名为unit_for_testA.py，代码如下：

1. # 导包
   
2. import unittest
   
3. 
   
4. class UnitForTestA(unittest.TestCase):
   
5.     # 前置条件
   
6.     def setUp(self) -> None:
   
7.         print('this is AsetUp')
   
8. 
   
9.     # 后置条件
   
10.     def tearDown(self) -> None:
    
11.         print('this is AtearDown')
    
12. 
    
13.     #定义测试用例
    
14.     def test_1(self):
    
15.         print('this is Atest1!!!')
    
16. 
    
17.     def test_2(self):
    
18.         print('this is Aest2!!!')
    
19. 
    
20.     def test_3(self):
    
21.         print('this is Atest3!!!')# 导包
    
22. import unittest
    
23. 
    
24. class UnitForTestA(unittest.TestCase):
    
25.     # 前置条件
    
26.     def setUp(self) -> None:
    
27.         print('this is AsetUp')
    
28. 
    
29.     # 后置条件
    
30.     def tearDown(self) -> None:
    
31.         print('this is AtearDown')
    
32. 
    
33.     # 定义测试用例
    
34.     def test_1(self):
    
35.         print('this is Atest1!!!')
    
36. 
    
37.     def test_2(self):
    
38.         print('this is Aest2!!!')
    
39. 
    
40.     def test_3(self):
    
41.         print('this is Atest3!!!')

复制代码

再新建一个类存放套件，文件名为TestSuite_demo.py，直接在UnitTest类中运行无法生效。
　　下面创建一个测试套件（unittest.TestSuite()），并分别用五种不同的方法给该测试套件添加测试用例。
　　（1）添加测试用例的第一种方法--单个添加（addTest）

1. # 导入unittest的包
   
2. import unittest
   
3. # 导入存放测试用例的类
   
4. from unittest_demo.unit_for_testA import UnitForTestA
   
5. 
   
6. # 创建一个测试套件
   
7. suite = unittest.TestSuite()
   
8. # 添加测试用例的第一种方法
   
9. suite.addTest(UnitForTestA('test_1'))
   
10. suite.addTest(UnitForTestA('test_2'))
    
11. suite.addTest(UnitForTestA('test_3'))
    
12. 
    
13. # 基于Ruunner来运行测试套件
    
14. runner = unittest.TextTestRunner()
    
15. runner.run(suite)

复制代码

（2）添加测试用例的第二种方法--批量添加（addTests）

1. # 导入unittest的包
   
2. import unittest
   
3. # 导入存放测试用例的类
   
4. from unittest_demo.unit_for_testA import UnitForTestA
   
5. 
   
6. # 创建一个测试套件
   
7. suite = unittest.TestSuite()
   
8. 
   
9. # 添加测试用例的第二种方法
   
10. cases = [UnitForTestA('test_1'), UnitForTestA('test_2'), UnitForTestA('test_3')]
    
11. suite.addTests(cases)
    
12. 
    
13. # 基于Ruunner来运行测试套件
    
14. runner = unittest.TextTestRunner()
    
15. runner.run(suite)

复制代码

（3）添加测试用例的第三种方法--根据路径下的文件名匹配（defaultTestLoader.discover）
　　该添加测试用例的方法也为批量执行，代码中指定的目录下，如果文件名称满足筛选条件，则所有满足条件的文件中的所有测试用例都会被执行，为了看清效果，我们再建一个UnitTest类，编写测试用例，文件名为unit_for_testB.py，代码如下：

1. # 导包
   
2. import unittest
   
3. 
   
4. class UnitForTestB(unittest.TestCase):
   
5.     # 前置条件
   
6.     def setUp(self) -> None:
   
7.         print('this is BsetUp')
   
8. 
   
9.     # 后置条件
   
10.     def tearDown(self) -> None:
    
11.         print('this is BtearDown')
    
12. 
    
13.     # 定义测试用例
    
14.     def test_1(self):
    
15.         print('this is Btest1!!!')
    
16. 
    
17.     def test_2(self):
    
18.         print('this is Best2!!!')
    
19. 
    
20.     def test_3(self):
    
21.         print('this is Btest3!!!')

复制代码

目录结构如图：

　　此时在测试套件类（TestSuite_demo.py）中，根据路径下的文件名添加测试用例的代码为：

1. # 导入unittest的包
   
2. import unittest
   
3. 
   
4. # 创建一个测试套件
   
5. suite = unittest.TestSuite()
   
6. 
   
7. # 添加测试用例的第三种方法
   
8. test_dir = './'  # 指定的路径下所有与文件名相匹配的文件下的测试用例（批量执行）
   
9. discover = unittest.defaultTestLoader.discover(start_dir=test_dir, pattern='unit_for_test*.py')
   
10. 
    
11. # 基于Ruunner来运行测试套件
    
12. runner = unittest.TextTestRunner()
    
13. runner.run(discover)

复制代码

　运行结果如下图，由图可见，在当前目录下的两个文件中的用例都执行了。
　　start_dir：需要执行的文件路径；
　　pattern：需要匹配的文件名称，其中unit_for_test*.py匹配所有以unit_for_test开头的python文件。

　　（4）添加测试用例的第四种方法--根据类名去读取（单个类）（TestLoader().loadTestsFromTestCase()）

1. # 导入unittest的包
   
2. import unittest
   
3. 
   
4. # 导入存放测试用例的类
   
5. from unittest_demo.unit_for_testA import UnitForTestA
   
6. 
   
7. # 创建一个测试套件
   
8. suite = unittest.TestSuite()
   
9. 
   
10. # 添加测试用例的第四种方法
    
11. suite.addTests(unittest.TestLoader().loadTestsFromTestCase(UnitForTestA))
    
12. 
    
13. # 基于Ruunner来运行测试套件
    
14. runner = unittest.TextTestRunner()
    
15. runner.run(suite)

复制代码

（5）添加测试用例的第五种方法--根据名称去添加（TestLoader().loadTestsFromName()）

1. # 导入unittest的包
   
2. import unittest
   
3. 
   
4. # 创建一个测试套件
   
5. suite = unittest.TestSuite()
   
6. 
   
7. # 添加测试用例的第五种方法
   
8. suite.addTests(unittest.TestLoader().loadTestsFromName('unit_for_testA.UnitForTestA'))
   
9. 
   
10. # 基于Ruunner来运行测试套件
    
11. runner = unittest.TextTestRunner()
    
12. runner.run(suite)

复制代码

五、跳过用例
　　在执行用例的时候，不是每个用例都必须跑一次，unittest提供了跳过一些跳过用例不执行的方法，在用例前加装饰器，具体有：
　　（1）@unittest.skip：直接跳过本条用例不执行；
　　（2）@unittest.skipUnless(1 > 2, '1>2是假的')：当条件为假时，跳过该用例不执行；
　　（3）@unittest.skipIf(1 < 2, '1<2是真的')：当条件为真时，跳过该用例不执行。