将参数传递给 Pytest Fixture，参数将在 if 语句中使用

Question

我正在尝试创建一个 Pytest 夹具并在夹具中包含“user_type”参数。然后在 if 语句中评估传递的参数。

conftest.py

import pytest

# Some custom modules...

@pytest.fixture
def generate_random_user(_db, user_type):
    # Do stuff
    if user_type == "no-subscription":
        # Do stuffs

        yield _db

        _db.drop_all()

    elif user_type == "with-subscription":
        pass

test.py

@pytest.mark.usefixtures("live_server", "client")
def test_checkout_new_customer(selenium, generate_random_user("no-subscription")):
    pass

Answer 1

我通过使用参数化和间接=True 解决了这个问题

conftest.py

@pytest.fixture
def generate_random_user(_db, request):

    user_type = request.param

    if user_type == "no-subscription":
        # Create User

        yield _db

        _db.drop_all()

    elif user_type == "with-subscription":
        pass

test.py

@pytest.mark.usefixtures("live_server", "client", "firefox_headless", "_db")
@pytest.mark.parametrize("generate_random_user", ["no-subscription"], indirect=True)
def test_checkout_new_customer(selenium, generate_random_user):
    # Check if user has been created successfully
    user5 = models.User.query.get(5)
    print(user5)