Unittest中的TestSuite和TestRunner

1、Unittest中主要使用TestSuite添加测试用例到测试套件中，再使用Testrunner运行测试用例，并生成测试报告。TestSuite可以指定运行的测试用例，要比main函数运行测试用例要灵活一些。

2、TestSuite和TestRunner用法

unittest_testsuite_testrunner_demo.py
# 导包
import unittest

# 创建测试用例的类
class TestSuiteDemo1(unittest.TestCase):
    def test01(self):
        print("测试用例：大田03")

    def test02(self):
        print("测试用例：大田04")

class TestSuiteDemo2(unittest.TestCase):
    def test01(self):
        print("测试用例：大田01")

    def test02(self):
        print("测试用例：大田02")


if __name__ == '__main__':

    #运行指定测试用例方法：

     # 实例化测试套件
     suite = unittest.TestSuite()

     # 添加测试用例到测试套件当中
     suite.addTest(TestSuiteDemo1('test01'))

     # 实例化test_runner
     runner = unittest.TextTestRunner()

     # 使用runner运行测试套件
     runner.run(suite)

运行结果：可以看到仅运行了自己想运行的测试用例

 3、TestSuite添加测试用例的三种方法

1）suite.addTest：添加单个测试用例

可以看上面的例子即可

2）suite.addTests：添加多个测试用例

# 添加多个测试用例
if __name__ == '__main__':

    #运行指定测试用例方法：

     # 实例化测试套件
     suite = unittest.TestSuite()

     # 添加测试用例到测试套件当中：添加单个测试用例方法
     # suite.addTest(TestSuiteDemo1('test01'))

     #添加多个测试用例
     suite.addTests([TestSuiteDemo2('test01'),TestSuiteDemo2('test02')])

     # 实例化test_runner
     runner = unittest.TextTestRunner()

     # 使用runner运行测试套件
     runner.run(suite)

运行结果：

 

3）TestLoader

第一种：TestLoader().discover(("./", "*.py")

根据执行的路径来寻找指定的py文件中的测试用例，加载到测试套件当中

第二种：TestLoader().loadTestsFromTestCase(测试用例类名)

根据测试用例的类名来加载测试用例

unittest_testloader_demo.py

import unittest
def testXX(aa):
  print("编外的测试用例")

class TestSuiteDemo1(unittest.TestCase):
 def test01(self):
   print("执行测试用例aaaa")

 def test02(self):
   print("执行测试用例bbbb")

class TestSuiteDemo2(unittest.TestCase):
   def test01(self):
     print("执行测试用例xxxx")
   def test02(self):
     print("执行测试用例yyyy")

class TestSuiteDemo3(unittest.TestCase):
   def test01(self):
     print("执行接口测试")
   def test02(self):
     print("执行单元测试")
   def test03(self):
     print("执行安全测试")


if __name__ == '__main__':
  t1 = unittest.TestLoader()
  #方法1
  suite = t1.discover("./", "*.py") # 使用discover来寻找测试用例
  #方法2
  # suite = t1.loadTestsFromTestCase(TestSuiteDemo3)
  runner = unittest.runner.TextTestRunner() # 实例化runner
  runner.run(suite) # 使用runner执行测试套件