x/opencafe
Code Issues Proposed changes

Merge "Update datagen decorator to support lists"

Browse Source
This commit is contained in:
Jenkins 2014-05-02 16:26:33 +00:00 committed by Gerrit Code Review
commit 0d3d333cee
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
cafe/drivers/unittest/decorators.py
View File

@ -47,12 +47,16 @@ def tags(*tags, **attrs):
return decorator
def data_driven_test(dataset_source=None):
def data_driven_test(*dataset_sources, **kwargs):
"""Used to define the data source for a data driven test in a
DataDrivenFixture decorated Unittest TestCase class"""
def decorator(func):
setattr(func, DATA_DRIVEN_TEST_ATTR, dataset_source)
# dataset_source checked for backward compatibility
combined_lists = kwargs.get("dataset_source") or []
for list_ in dataset_sources:
combined_lists += list_
setattr(func, DATA_DRIVEN_TEST_ATTR, combined_lists)
return func
return decorator