如何将夹具传递给 pytest.mark.parametrize？

Question

我正在尝试将三个不同的装置传递给我的 pytest.mark.parameterize 装饰器，如下所示：

@pytest.mark.parametrize("credentials, return_code", [
    (user1, 200),
    #(user2, 200),
    #(user3, 401)
])
def test_login():
    assert True
# Actual test uses response.return_code == return_code

但是，我收到一条错误消息：

NameError: name 'user1' is not defined

这不正确，我在conftest.py中定义了它：

user1_info = [('user1', 'password')]
@pytest.fixture(params=user1_info)
def user1(request):
    return request.param

---

如果我将测试更改为这样，夹具就可以工作：

def test_login(user1):
    assert True
# Actual test uses response.return_code == return_code

问题是，如果我这样做，我必须编写三个单独的测试，一个用于每个夹具，一个用于每个 status_code 检查。如果我可以使用parameterize，我可以使用一个函数并检查用户/预期的返回码。

我如何利用我的固定装置是pytest.mark.parameterize 装饰器的一部分？

Answer 1

要将夹具作为参数化参数传递，请将它们作为字符串引用并使用关键字参数indirect=['arg_name', ...]。

然后使用参数名称credentials 和return_code 作为测试函数的参数。

@pytest.mark.parametrize(
    "credentials, return_code",
    [
        ("user1", 200),
        ("user2", 200),
        ("user3", 401),
    ],
    # Only the `credentials` parameter is a reference to the
    # fixture. The `return_code` should be passed as is.
    indirect=["credentials"],
)
def test_login(credentials, return_code):
    return True

pytest 的 parametrize 魔术和 indirect 的文档可以在这里找到： https://docs.pytest.org/en/stable/example/parametrize.html#apply-indirect-on-particular-arguments

Answer 2

据我所知，你不能。有一些解决方法：

1：放弃参数化

def test_login():
    for user, return_code in [
      (User("user1", ...), 200),
      (User("user2", ...), 200),
      (User("user3", ...), 401),
    ]:
        # test user
        pass

或

def test_login(user1, user2, user3):
    for user, return_code in [
      (user1, 200),
      (user2, 200),
      (user3, 401),
    ]:
        # test user
        pass

专业版：简单
缺点：如果第一个测试失败，后面的测试就不会运行。错误消息也没有显示哪个测试用例失败了。

2：使用索引或名称进行参数化

@pytest.mark.parametrize('idx', [0,1,2])
def test_login(idx):
    user, return_code = [
      (User("user1", ...), 200),
      (User("user2", ...), 200),
      (User("user3", ...), 401),
    ][idx]
        # test user
        pass

（或使用user1、user2、user3 固定装置的等价物）

专业版：pytest 识别单独的测试
缺点：无用的索引/名称

3：使用间接参数化夹具

@pytest.mark.fixture
def user_and_code(request):
    if request.param == "user1":
        return User("user1", ...), 200
    elif request.param == "user2":
        return User("user2", ...), 200
    elif request.param == "user3":
        return User("user3", ...), 401

@pytest.mark.parametrize('user_and_code', ['user1', 'user2', 'user3'], indirect=True)
def test_login(user_and_code):
    user, return_code = user_and_code

    # test user
    pass

或者

@pytest.mark.fixture
def user_and_code(request, user1, user2, user3):
    return [
        (user1, 200),
        (user2, 200),
        (user3, 401),
    ][request.param]

@pytest.mark.parametrize('user_and_code', [0,1,2], indirect=True)
def test_login(user_and_code):
    user, return_code = user_and_code

    # test user
    pass

Pro：测试代码很干净，fixture 可以在其他测试中重用
缺点：更复杂

根据文档，间接参数化是为了在测试收集期间跳过昂贵的设置，而不是让固定装置可用于参数，正如您所见，这不是很好。

在上述每个选项中，有几个选项可以混合搭配，例如实例化所有用户或使用 if/else 仅实例化您需要的内容，使用 user1 等固定装置或内联创建用户，使用名称或索引等。示例仅显示了一些替代方案。