如何通过命令行将多个列表参数传递给多个Pytest fixtures

Question

我想从命令行将以下列表传递到我的pytest代码中（类似于Python中的argparser），有人可以帮我如何做到这一点吗？ pytest -vs test_sample.py --html=results.html --A_list=['A1', 'A2'] --B_list=['B1'] --C_list=['C1', 'C2']

这是我的pytest代码。

import pytest

@pytest.fixture(scope="session", params=A_list) # here A_list is a list and it has to be passed from command line
def get_A_list(request):
    print("inside fixture get_A_list - element in A_list: ".format(request.param))
    yield request.param


@pytest.fixture(scope="module", params=B_list) # here B_list is a list and it has to be passed from command line
def get_B_list(request, get_A_list):
    print("inside fixture get_B_list - element in A_list: {}, element in B_list: {}: ".format(get_A_list, request.param))
    yield request.param


@pytest.mark.parametrize("c", C_list) # here C_list is a list and it has to be passed from command line
def test_functionality(get_A_list, get_B_list, c):
    print("inside function test_functionality - element in A_list: {}, element in B_list: {}, element in C_list: {}: ".format(get_A_list, get_B_list, c))

test_functionality 是我的测试用例函数。测试用例的数量应该基于命令行传递的列表的长度。

它必须按照下面的顺序执行上述列表中的测试用例。

test_functionality[A1-B1-C1]
test_functionality[A1-B1-C2]
test_functionality[A2-B1-C1]
test_functionality[A2-B1-C2]

我需要在fixture get_A_list中使用A_list元素执行某些操作...同样，我需要在fixture get_B_list中使用A_list和B_list元素执行某些操作......这样，我有4个fixtures，最后我需要在test_functionality中执行测试用例，它将使用A_list、B_list、C_list等元素。

最终输出应该像下面这样...

    inside fixture get_A_list - element in A_list: A1
    inside fixture get_B_list - element in A_list: A1, element in B_list: B1
    inside function test_functionality - element in A_list: A1, element in B_list: B1, element in C_list: C1
    inside function test_functionality - element in A_list: A1, element in B_list: B1, element in C_list: C2
    inside function test_functionality - element in A_list: A1, element in B_list: B1, element in C_list: C3
    inside fixture get_A_list - element in A_list: A2
    inside fixture get_B_list - element in A_list: A2, element in B_list: B1
    inside function test_functionality - element in A_list: A2, element in B_list: B1, element in C_list: C1
    inside function test_functionality - element in A_list: A2, element in B_list: B1, element in C_list: C2
    inside function test_functionality - element in A_list: A2, element in B_list: B1, element in C_list: C3

- Srikanth K

Answer 1

这里的问题是params参数在加载时被读取，而命令行参数的信息尚不可用，因此您不能像示例中那样简单地设置它们。

最好的方法可能是重新构造代码，使其不再需要。如果您能够做到这一点，就不需要额外的fixtures了。

无论如何，您首先必须注册您的命令行参数：

conftest.py

def pytest_addoption(parser):
    parser.addoption("--A_list", action="store")
    parser.addoption("--B_list", action="store")
    parser.addoption("--C_list", action="store")

如果您可以重构代码使fixtures不依赖于参数，则可以基于传递的命令行参数对测试进行参数化。只要将参数作为逗号分隔列表传递，您就可以执行以下操作：

test.py

import pytest

@pytest.hookimpl
def pytest_generate_tests(metafunc):
    # collect fixture names with values
    option_values = {
        "a": metafunc.config.option.A_list,
        "b": metafunc.config.option.B_list,
        "c": metafunc.config.option.C_list
    }

    # add parametrization for each fixture name
    for name, value in option_values.items():
        if name in metafunc.fixturenames and value is not None:
            metafunc.parametrize(name, value.split(","))

def test_functionality1(a, b, c):
    print(a, b, c)

这将按照所需顺序创建参数化测试。

如果不可能，您可以尝试在运行时创建您的fixtures（有关更多信息，请参见此文章）：

conftest.py

# define the functions used as fixtures
def a_list_fixture(request):
    print(f"get_A_list: {request.param}")
    yield request.param

def b_list_fixture(request, get_A_list):
    print(f"get_B_list: {get_A_list},  {request.param}")
    yield request.param

# at session start, create the fixtures with the correct params
def pytest_sessionstart(session):
    a_list = session.config.option.A_list
    if a_list:
        a_list = a_list.split(",")
        name = "get_A_list"
        fn = pytest.fixture(scope='session', name=name,
                            params=a_list)(a_list_fixture)
        setattr(sys.modules[__name__], "{}_func".format(name), fn)
    b_list = session.config.option.B_list
    if b_list:
        b_list = b_list.split(",")
        name = "get_B_list"
        fn = pytest.fixture(scope='session', name=name,
                            params=b_list)(b_list_fixture)
        setattr(sys.modules[__name__], "{}_func".format(name), fn)

请注意，您仍然需要像上面那样对函数的 c 进行参数化：

test.py

import pytest

@pytest.hookimpl
def pytest_generate_tests(metafunc):
    c_value = metafunc.config.option.C_list
    if "c" in metafunc.fixturenames and value is not None:
            metafunc.parametrize("c", value.split(","))

def test_functionality(get_A_list, get_B_list, c):
    print(f"test_functionality {get_A_list}, {get_B_list}, {c}")

这将创建所需的参数化测试： pytest -vv --A_list="A1,A2" --B_list="B1" --C_list="C1,C2"：

================================================= test session starts =================================================
...
collected 4 items

so/cmd_line_args/test.py::test_functionality[C1-A1-B1] PASSED                                                    [ 25%]
so/cmd_line_args/test.py::test_functionality[C2-A1-B1] PASSED                                                    [ 50%]
so/cmd_line_args/test.py::test_functionality[C1-A2-B1] PASSED                                                    [ 75%]
so/cmd_line_args/test.py::test_functionality[C2-A2-B1] PASSED                                                    [100%]

================================================== 4 passed in 0.28s ==================================================

请注意，参数顺序不是我们想要的，但这只是一个外观问题。这里是print函数的输出结果： pytest -s --A_list="A1,A2" --B_list="B1" --C_list="C1,C2"

================================================= test session starts =================================================
...
collected 4 items

test.py get_A_list: A1
get_B_list: A1,  B1
test_functionality A1, B1, C1
.test_functionality A1, B1, C2
.get_A_list: A2
get_B_list: A2,  B1
test_functionality A2, B1, C1
.test_functionality A2, B1, C2
.

================================================== 4 passed in 0.27s ==================================================

- MrBean Bremen

---

感谢@MrBean Bremen......这并没有解决我的问题....我需要在fixture get_A_list中使用A_list中的元素执行一些操作.........同样，我需要在fixture get_B_list中使用A_list和B_list中的元素执行一些操作.......像这样，我有4个fixtures，最后我需要执行测试用例在test_functionality中使用A_list，B_list，C_list等元素........我已经更新了我的问题，包括最终输出以及执行顺序。 - Srikanth K

我添加了一个更复杂的选项来实现这个目标（尽管我保留了第一个答案，因为它对于大多数情况已经足够了）。 - MrBean Bremen

谢谢MrBean，我会采用你的第二个答案并查看它是否解决了我的问题。 我将检查是否需要进行任何修改以实现我的输出及顺序...这只是一个小例子，但实际上我需要在test_functionality中执行大约60个测试用例，并且在此之前我需要进行一些操作（计划在多个固定装置中完成这些操作）。 - Srikanth K

谢谢MrBean。除了参数顺序之外，它运行得很好，但这不是一个功能性问题。正如您所提到的，这只是一个外观问题。我根据您的解决方案添加了更多的参数以从命令行传递。再次感谢您的解决方案。 - Srikanth K