如何动态选择 pytest 夹具？

Question

import pytest


@pytest.fixture()
def fixture1(fixture_a):
    print('In fixture 1')
    <do>
    step a
    step b
    <end>
    return <some object1> 



@pytest.fixture()
def fixture2(fixture_b):
    print('In fixture 2')
    <do>
    step x
    step y
    <end>
    return <some object2> 


def decide():
    a = 1
    if a == 1:
        return fixture1: Object1
    else:
        return fixture2: Object2


def test_me():
    res = decide()
    assert res == Object

我有两个夹具 arg1 和 arg2，现在我想将其中一个夹具返回给测试，但这必须是基于条件的动态选择。如何做到这一点？

更新：fixture arg1 和 arg2 有一个依赖链，它们被用于不同的测试。

另外，决定函数需要在多个测试中使用。

Answer 1

我想这会有所帮助 - 让夹具构建器在某些情况下运行函数（将夹具 A、B 转换为函数）：

def arg1():
    print('In arg 1')
    return 1

def arg2():
    print('In arg 2')
    return 2

@pytest.fixture
def env_builder():
    if condition:
        return arg1()
    else
        return arg2()

def test_smth(env_builder):
   assert 2 == env_builder

你的条件取决于你，os.getenv('environment')，或者可能在运行测试时定位 cli 参数

Answer 2

您可以将arg1 和arg2 传递给test 并测试条件本身。

def decide():
    a = 1
    if a == 1:
        return 1
    else:
        return 2

def test_me(fixture1, fixture2):
    arg = decide()
    if arg == 1:
        assert fixture1 == 1
    else:
        assert fixture2 == 2