python 装饰器(实现dataprovider)

接触装饰器,是因为在使用unittest测试时,想要看起来更简洁的传递数据,看了下Java的TestNG有DataProvider这个东东,python里第三方库也有实现的。但还是满足不了自己的要求,基础的东西还是学一下更好。

我的设想:
测试用例的数据都放在一个test_cases_data的dict中(test_case_name为key),通过dataprovider根据test_case_name获取其对应的测试数据,并传递给test method,如下:

@data_provider(['login_success'])
    def test_login_success(self, user, password, verifycode):
    ......

献出我的data_provider文件:

import data.config


def data_provider(name_list):
    """第一个自己的decorator"""
    test_cases_data = get_data_by_name(name_list)

    def case_decorator(func):
        def repl(self, **kwargs):
            for test_case_data in test_cases_data:
                    func(self, **test_case_data)
        return repl
    return case_decorator


def get_data_by_name(name_list):
    test_cases_data = data.config.get_test_cases_data()
    case_data = []
    for name in name_list:
        case_data.append(test_cases_data[name])
    return case_data

强烈推荐一篇很好的文章&参考:
http://python.jobbole.com/85056/
https://pypi.python.org/pypi/unittest-data-provider/

猜你喜欢

转载自blog.csdn.net/u010895119/article/details/79231950