我可以向pytest fixtures传递参数吗？

Question

我所有测试的基准是，总有一辆出租车上至少有一名乘客。我可以通过一些基本的固定装置轻松实现此设置：

from blah import Passenger, Taxi

@pytest.fixture
def passenger():
    return Passenger()

@pytest.fixture
def taxi(passenger):
    return Taxi(rear_seat=passenger)

测试基线很简单：

def test_taxi_contains_passenger(taxi)
    assert taxi.has_passenger()

当我开始需要更复杂的测试设置时，我的问题就出现了。在某些情况下，我需要出租车有多个乘客，在某些情况下，我需要定义乘客属性。例如：

def test_three_passengers_in_taxi(taxi)
    assert taxi.has_passengers(3)
    assert taxi.front_passenger_is_not_a_child()

我可以通过为特定的测试设置特定的设备来解决这个问题。对于上述测试，我将创建以下fixture：

@pytest.fixture
def three_passenger_test_setup(taxi)
    taxi.add_front_seat_passenger(Passenger(child=False))
    taxi.add_rear_seat_passenger(Passenger())
    return taxi

我可以把上面的fixture放进我的测试用例中，一切都很好，但是如果我沿着这条路走下去，我可能会在每次测试中都有fixture，感觉应该有一种更有效的方法来完成这项工作。

是否有方法将参数传递给fixture，以便这些参数可以用于创建fixture返回的对象？我应该参数化测试函数吗？固定装置？或者我是在浪费时间，每次测试都要固定下来吗？

Answer 1

那个fixture只是一个Python装饰器。

@decorator
def function(args):
  ...

喜欢

def function(args):
  ...
function = decorator(function)

因此，您可以编写自己的decorator，将要修饰的函数包装成所需的和函数fixture：

def myFixture(parameter):
  def wrapper(function):
    def wrapped(*args, **kwargs):
      return function(parameter, *args, **kwargs)
    return wrapped
  return pytest.fixture(wrapper)

@myFixture('foo')
def function(parameter, ...):
  ...

它的作用类似于fixture，但将一个值（'foo'）作为parameter传递给function。