pytest 夹具的多个副本

Question

假设我有一个像下面这样的简单夹具（使用 pytest-django，但它也适用于 pytest）：

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

当我的测试需要 MyModel 的单个实例时，这非常有用。但是如果我需要两个（或三个或四个）呢？我希望每个实例都是不同的，但要以相同的方式设置。

我可以复制/粘贴代码并重命名fixture函数，但这似乎不太优雅。

同样，我也试过：

@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

我的方法可能是创建一个可以生成您的对象的夹具：

@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）

Answer 2

我很晚才知道这个问题......但是，使用参数化的夹具并简单地返回您想要复制的夹具似乎也可以。

import pytest
import random

@pytest.fixture
def name():
    return random.randint(0, 10)


@pytest.fixture(params=[0, 1])
def parametrized_name(request, name):
    return name


def test_something(parametrized_name):
    print "Testing with name: " + str(parametrized_name)

如果您运行上述测试，您会得到 2 个不同的“名称”固定装置

$ pytest -s blah.py
============================================= test session starts ==============================================
platform linux2 -- Python 2.7.14, pytest-3.3.2, py-1.5.2, pluggy-0.6.0
rootdir: /home/bsquizza/repos/blah/tests, inifile:
collected 2 items                                                                                              

blah.py Testing with name: 7
.Testing with name: 10
.                                                                                               [100%]

=========================================== 2 passed in 0.01 seconds ===========================================