使用fixture返回值作为mark.parametrize()中的值

时间:2016-06-22 15:37:27

标签: python pytest

我的问题是 - 是否可以将fixture中的返回值用作参数化中的值? 问题是 - 我想动态获取可能的值(例如,虚拟服务器上的可用系统)以进行参数化。当其中一个灯具创建虚拟服务器时,我可以访问这些。测试看起来像这样(伪代码):

[conftest.py]

@pytest_fixture(scope='session')
def test_server(request):
    test_server = Server([default_params])
    test_server.add()
    def fin():
        test_server.delete()
    request_addfinalizer(fin)
    return test_server()

[tests.py]

def test_basic_server(test_server):
    systems = test.server.get_available_systems()
    for system in systems:
        test_server.install(system)
        test_server.run_checks()
        test_server.uninstall(system)

def test_other(test_server):
    [other tests]

etc

这样,为每个会话添加一个服务器,然后在其上运行所有测试,并在会话结束后删除服务器。但有没有办法在@ pytest.mark.parametrize中获取可用的系统而不显式列出它们(静态地作为参数化中的列表),使用会话开始时添加的服务器中的方法?这样每个系统都可以在单独的测试中运行。

我尝试在另一个fixture中使用test_server然后返回列表(test_server fixture返回test_server的方式相同,但是我不能将它用作参数化中的值 - 因为在任何测试中调用test_server fixture之前都会评估decorator ,获取列表取决于test_server fixture。

这是理想的:

[tests.py]

@pytest.mark.parametrize('system',[systems_list <- dynamically generated
                                             when the server is created])

def test_basic_server(test_server,system):
    test_server.install(system)
    test_server.run_checks()
    test_server.uninstall(system)

这只是一个非常基本的例子,在我的测试中,我需要根据多个场景和值进行参数化,当我静态执行时,我最终会得到巨型数组。

但原理保持不变 - 基本上:我可以在使用此夹具运行的第一次测试之前调用夹具,或者pytest.mark.parametrize()如何访问夹具值?

2 个答案:

答案 0 :(得分:1)

我认为你可能无法直接达到你想要的效果。因为在收集期间调用@pytest.mark.parametrize,并且在收集完成后将调用灯具。

但我有另一种方法来实现类似的结果,主要是通过扩展pytest插件pytest_generate_tests和使用方法metafunc.parametrizehttps://pytest.org/latest/parametrize.html#basic-pytest-generate-tests-example

这是我的解决方案。 在conftest.py中

class System(object):
    def __init__(self, name):
        self.name = name

    def __repr__(self):
        return "<System '{}'>".format(self.name)

def get_available_systems():
    return [System('A'), System('B'), System('C')]


def pytest_generate_tests(metafunc):
    if 'system' in metafunc.fixturenames:
        available_systems = get_available_systems()

        metafunc.parametrize('system', available_systems)

在测试文件中:

def test_basic_server(system):
    print(system)

这是输出,您将可以访问测试中的每个系统。

collected 3 items

test_01.py::test_basic_server[system0] <System 'A'>
PASSED
test_01.py::test_basic_server[system1] <System 'B'>
PASSED
test_01.py::test_basic_server[system2] <System 'C'>
PASSED

糟糕的是,每次使用fixture get_available_systems时都会调用system,这不是你想要的。但我认为添加一些额外的逻辑以使查询逻辑只执行一次并不困难。

例如:

def pytest_generate_tests(metafunc):
    if 'system' in metafunc.fixturenames:
        if hasattr(metafunc.config, 'available_systems'):
            available_systems = metafunc.config.available_systems
        else:
            available_systems = get_available_systems()
            metafunc.config.available_systems = available_systems
        metafunc.parametrize('system', available_systems)

答案 1 :(得分:0)

我能够处理类似的问题,我必须生成用于动态参数化的测试数据:

class TestFilters(object):
 cls_testdata1 = []

def setup_class(cls):
 r = []
 for i in range(5):
   r.append((x, y, z))
 TestFilters.cls_testdata1 = r

@pytest.mark.parametrize("filter_id", list(range(5)))
def test_func(self, filter_id):
    params = TestFilters.cls_testdata1[filter_id]

...

这将支持动态添加参数,只需要预先确定测试次数。