1

我正在尝试围绕 py.test 建立一个很好的基础

我们的一些测试需要某些测试数据才能工作。

今天我们只是指定一个模拟对象作为函数参数并在生成器中进行设置，这显然是不可取的。

以下是它今天的外观示例：

def test_something(self, some_data):
    # some_data is unused in the test

我想做这样的事情：

@uses_some_data
def test_something(self):
    # The data is loaded when the test is run

虽然我还没有弄清楚如何正确地做到这一点。

我不能使用类设置，因为我希望数据在整个会话中保持不变，而不是在每个测试类上设置/拆除。

我的第一个想法是仍然使用 funcargs 但不是让测试有 funcarg 我们让装饰器为函数请求 funcarg ，基本上隐藏了丑陋。

这个问题是我需要一个 py.test 对象来请求一个funcarg。
有什么办法可以得到这样的对象，或者这是错误的方法吗？

如果收集的测试都不需要数据，则不必加载数据将是一个很棒的奖励，这是使用装饰器的缺点，因为无论测试是否运行，它们总是运行。

4

2 回答 2

1

这里有一些可以按原样工作的东西，如果不是，希望能指出你正确的方向。

class TestData(object):
    def __getattr__(self, name):
        if name not in ('data1', 'data2', 'data3'):
            raise AttributeError("TestData has no %s" % name)
        if name == 'data1':
            result = self._generate_data('data1')
            setattr(self.__class__, name, result)
        elif name == 'data2':
            result = self._generate_data('data2')
            setattr(self.__class__, name, result)
        elif name == 'data3':
            result = self._generate_data('data3')
            setattr(self.__class__, name, result)
        return result
    def _generate_data(self, data_name):
        return data_name * int(data_name[-1])

TestData 类使用该__getattr__方法根据需要生成数据，并且通过将生成的日期保存回类（而不是实例！），数据也被保留以供将来使用。

class uses_some_data(object):
    def __init__(self, func):
        self.func = func
    def __call__(self, *args, **kwargs):
        global test_data
        test_data = TestData()
        return self.func(*args, **kwargs)

一个简单的装饰器来设置全局名称绑定test_data。事实上，这个版本的装饰器非常简单，可以很容易地用test_data = TestData().

@uses_some_data
def testing_test():
    print(test_data.data2)

和一个测试功能。

如果您不喜欢全局级别，则test_data可以对装饰器更感兴趣并分配test_data给函数本身：

class uses_some_data(object):
    def __init__(self, func):
        self.func = func
    def __call__(self, *args, **kwargs):
        self.func.test_data = TestData()
        return self.func(*args, **kwargs)

在这种情况下，请确保您的测试函数引用自己

@uses_some_data
def testing_test():
    print(testing_test.test_data.data2)
于 2011-11-07T19:48:07.970 回答
1

在玩了一些之后，我发现这有效：

def pytest_funcarg__some_data(request):
    def create():
        # Load the test data here
        print 'Test data loaded'

    return request.cached_setup(
        setup=create,
        scope='session',
        extrakey='some_data'
    )

def uses_some_data(func):
    # The funcarg is actually requested here
    def wrapper(self, some_data):
        return func
    return wrapper

class TestSomething(object):
    @uses_some_data
    def test_something(self):
        # "Some data" is now available
        pass
于 2011-11-07T22:53:30.077 回答