测开之路五十五：实现类似于unittest查找case

 

实现给一个路径，去查找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)