实现给一个路径，去查找test开头的测试用例文件
创建一个计算器的类，方便后面测试用
class Calculator(object): def add(self, x, y): return x + y def sub(self, x, y): return x - y def mul(self, x, y): return x * y def div(self, x, y): return x / y
创建一个unittest的用例集
# coding=utf-8 from unittest import main from unittest import TestCase from src import Calculator class TestCalculator(TestCase): def setUp(self): self.calc = Calculator() print("setUp") def tearDown(self): print("tearDown") def test_add(self): result = self.calc.add(2, 3) self.assertEqual(result, 5, u"测试加法。") print("test_add") def test_sub(self): result = self.calc.sub(2, 3) self.assertEqual(result, -1, u"测试减法。") print("test_sub") def test_mul(self): result = self.calc.mul(2, 3) self.assertEqual(result, 6, u"测试乘法。") print("test_mul") def test_div(self): result = self.calc.div(4, 2) self.assertEqual(result, 2, u"测试除法。") print("test_div") if __name__ == '__main__': main()
查找用例的类
import sys class Discovery(object): def __init__(self): pass def find_case_by_path(self, path): """ 获取用例 """ files = self.get_files(path, include=['.py'], exclude=['.xml', '.json', '.pyc']) cases = self.load_cases(files) return cases def get_files(self, path, include=[], exclude=['.xml', '.iml']): """ path是要列出文件的路径。根据代码规则，黑名单机制优先于白名单机制执行。 """ files = [] # 如果路径不存在，则直接返回空列表。 if not os.path.exists(path): return files # 如果路径是文件，则直接返回文件。 if os.path.isfile(path): # 这里处理exclude规则--黑名单机制。 if exclude: for rule in exclude: if path.find(rule) != -1: return files # 这里处理include规则--白名单机制。 if include: for rule in include: if path.find(rule) != -1: files.append(path) else: files.append(path) return files # 如果路径是目录，则列出目录内容，递归处理。 if os.path.isdir(path): results = os.listdir(path) for result in results: # 将原路径与路径下文件名拼接成新路径。 new_path = "/".join([path, result]) files.extend(self.get_files(new_path, include, exclude)) return files def load_cases(self, files): """ 加载用例 """ cases = [] for file in files: if file.find("test") == -1: # 判断文件名里面没有“test”就查找下一个 continue case = {'file': file} # 如果有就存到case字典里面 directory = os.path.dirname(file) # 获取到文件的目录 sys.path.append(directory) # 用sys把此文件的目录加进去 name = os.path.basename(file)[:-3] # 获取文件名 module = __import__(name) # 反射导入文件 case.setdefault("module", module) # 把文件加入到case for attr in dir(module): if attr.find("TestCase") != -1: # 判断文件里面有没有继承了“TestCase”的类（兼容unittest） continue if attr.find("Test") == -1: # 判断类名里面有没有Test continue object = getattr(module, attr)() # 反射的方式实例化对象 case.setdefault("object", object) # 获取这个对象 case.setdefault("method", []) for attr in dir(object): # 提取测试用例里面的方法 # 这里是断言结果。 if attr.find('result') != -1: case.setdefault("result", getattr(object, attr)) if not callable(getattr(object, attr)): # 从object代表的对象获取attr代表的函数，判断是否可以调用 continue if attr.find("test") != -1: case["method"].append(getattr(object, attr)) if attr.find("setUp") != -1: case.setdefault("setUp", getattr(object, attr)) if attr.find("tearDown") != -1: case.setdefault("tearDown", getattr(object, attr)) cases.append(case) return cases if __name__ == '__main__': discovery = Discovery() import os case_path = os.path.join(os.path.dirname(os.getcwd()), 'case') # case 文件夹 print(case_path) cases = discovery.find_case_by_path(case_path) print(cases)