您的位置:首页 > 编程语言 > Python开发

python 装饰器(实现dataprovider)

2018-02-01 18:27 183 查看
接触装饰器,是因为在使用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/
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: