python - Python。Pytest 夹具碰撞

Question

我正在尝试pytest yield-fixture在一项测试中多次使用默认范围。

@pytest.fixture
def create_temp_file():
    nonlocal_maker_data = {'path': None}  # nonlocal for python2

    def maker(path, data):
        nonlocal_maker_data['path'] = path
        with open(path, 'wb') as out:
            out.write(data)
    try:
        yield maker
    finally:
        path = nonlocal_maker_data['path']
        if os.path.exists(path):
            os.remove(path)


def test_two_call_of_fixture(create_temp_file):
    create_temp_file('temp_file_1', data='data1')
    create_temp_file('temp_file_2', data='data2')

    with open('temp_file_1') as f:
        assert f.read() == 'data1'

    with open('temp_file_2') as f:
        assert f.read() == 'data2'

    assert False
    # temp_file_2 removed
    # temp_file_1 doesn't removed

我有碰撞。第一个夹具没有清理 - temp_file_1 没有删除，而第二个文件删除得很好。是否可以多次正确使用夹具？

PS：我知道tmpdir- 标准 pytest 夹具。这只是一个例子。

Answer 1

在这个例子中，夹具产生了一个功能。我认为直接的方法是累积传递的参数：

@pytest.fixture
def create_temp_file():
    nonlocal_maker_data = set()  # nonlocal for python2

    def maker(path, data):
        with open(path, 'wb') as out:
            out.write(data)
        nonlocal_maker_data.add(path)
    try:
        yield maker
    finally:
        for path in nonlocal_maker_data:
            if os.path.exists(path):
                os.remove(path)