pytest夹具的多个副本

Question

假设我有一个简单的夹具,如下所示(使用pytest-django,但它也适用于pytest):

@pytest.fixture
def my_thing(request, db):
    thing = MyModel.objects.create()
    request.addfinalizer(lambda: thing.delete())
    return thing

当我的测试需要单个MyModel实例时,这很有用.但是,如果我需要两个(或三个或四个)呢？我希望每个实例都是不同的,但要以相同的方式设置.

我可以复制/粘贴代码并重命名夹具功能,但这看起来不够优雅.

同样,我试过:

@pytest.fixture
def my_thing_1(my_thing):
    return my_thing

@pytest.fixture
def my_thing_2(my_thing):
    return my_thing

但是,这些中的每一个似乎都返回MyModel的相同实例.

有没有办法用pytest的内置功能做我想做的事情？或者,我可以将我的装置的设置/拆卸移动到辅助函数中,因此我不会复制太多的代码.

或者我是以错误的方式处理这整件事？

Answer 1

我的方法可能是创建一个可以生成对象的fixture:

@pytest.fixture
def thing(request, db):
    class ThingFactory(object):
        def get(self):
            thing = MyModel.objects.create()
            request.addfinalizer(thing.delete)
            return thing
    return ThingFactory()

def test_thing(thing):
    thing1 = thing.get()
    thing2 = thing.get()

显然你可以做出.get()争论等.

(PS:还要注意在终结器中不需要lambda)