我有一个产生测试用例的基类。类注册测试的具体实现。由于鼻子只支持通过测试函数(不是类)的产量生成测试,我想要一个应用必要的函数包装的类装饰器。
注意:我需要 yield 测试用例的原因是测试类需要做一些不能在 setUp 中运行的初始化。
下面是一个显示预期行为的最小示例,但是,它仅在make_generator
装饰器位于当前模块中时才有效 - 当我将其移动到单独的模块时,不会产生任何测试。谁能告诉我为什么会这样?
def make_generator(cls):
"""returns a function that instantiates ``cls`` and yields tests. """
def generator():
obj = cls()
for t in obj:
yield t
generator.__name__ = 'test_%s' % cls.__name__
return generator
def register(func):
"""Set _check attr of ``func``. """
func._check = True
return func
class BaseMixin(object):
def __init__(self):
"""Register check handlers. """
self.check_handlers = []
attrs = (attr for attr in dir(self) if not attr.startswith('__'))
for attr in attrs:
func = getattr(self, attr)
if getattr(func, '_check', False):
self.check_handlers.append(attr)
def __iter__(self):
"""Generate tests for registered handlers. """
# do stuff
args = []
for check in self.check_handlers:
func = getattr(self, check)
yield func, args
@make_generator
class ConcreteTest(BaseMixin):
@register
def check_foobar(self, args):
assert False