Python+Unittest - discover()构建不同目录下的用例

一、执行指定路径的所有用例

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

举例：如目录层级如下图所示，如何使用最简洁的语法执行目录的所有用例

 语法：unittest.defaultTestLoader.discover(start_dir=start_path,pattern='test_*.py',top_level_dir=start_path)

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

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

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

用例模块代码写法如下，不同模块复制粘贴修改类名即可：

import unittest

class test_one_demo01(unittest.TestCase):
    def test_add(self):
        print('exec test_one_demo01_test_add')
    def test_sub(self):
        print('exec test_one_demo01_test_sub')


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

举例①：执行test_one目录下的所有用例时，目录路径写法

start_path = os.path.join(current_path,'test_one')

举例②：执行test_two目录下所有用例时，目录路径写法

start_path = os.path.join(current_path,'test_one/test_two')

举例③：执行test_03目录下的所有用例，目录路径写法

start_path = os.path.join(current_path,'test_one/test_two/test_03')

test_all源代码如下

import os.path
import unittest

current_path = os.path.dirname(__file__)#获取当前目录路径
start_path = os.path.join(current_path,'test_one')#连接路径语法

print(start_path)
#执行以下路径的所有测试用例
discover = unittest.defaultTestLoader.discover(start_dir=start_path,pattern='test_*.py',top_level_dir=start_path)

main_suite = unittest.TestSuite()
main_suite.addTest(discover)
unittest.main(defaultTest='main_suite')

二、执行指定类的指定用例

1、方法一：方法名不以test开头或直接注释用例

2、方法二：使用unittest.TestLoader()方法

import unittest
from unittest import loader


class test_one_demo01(unittest.TestCase):
    def test_add(self):
        print('exec test_one_demo01_test_add')
    def test_sub(self):
        print('exec test_one_demo01_test_sub')

#仅执行指定用例

def load_tests(loader=None,tests=None,pattrn=None):
    if not loader:
        loader = unittest.TestLoader()
    suite_list =  [loader.loadTestsFromName('test_one_01.test_one_demo01.test_sub')]
    return unittest.TestSuite(suite_list)

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