python 装饰器(实现dataprovider)

时间:2022-02-11 22:27:09

接触装饰器,是因为在使用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/