Question

我正在使用Pytest来测试可执行文件。此.exe文件在启动时读取配置文件。

我已经编写了一个fixture来在每个测试开始时生成这个.exe文件，并在测试结束时将其关闭。但是，我无法弄清楚如何告诉灯具使用哪个配置文件。我希望fixture在生成.exe文件之前将指定的配置文件复制到目录。

    @pytest.fixture
    def session(request):
        copy_config_file(specific_file) # how do I specify the file to use?
        link = spawn_exe()
        def fin():
            close_down_exe()
        return link 

    # needs to use config file foo.xml
    def test_1(session):  
        session.talk_to_exe()

    # needs to use config file bar.xml
    def test_2(session):
        session.talk_to_exe()

如何告诉灯具使用foo.xml功能test_1和bar.xml功能使用test_2？

由于约翰

Answer 1

一种解决方案是使用pytest.mark：

import pytest


@pytest.fixture
def session(request):
    m = request.node.get_closest_marker('session_config')
    if m is None:
        pytest.fail('please use "session_config" marker')
    specific_file = m.args[0]
    copy_config_file(specific_file) 
    link = spawn_exe()
    yield link
    close_down_exe(link)    

@pytest.mark.session_config("foo.xml")
def test_1(session):  
    session.talk_to_exe()

@pytest.mark.session_config("bar.xml")
def test_2(session):
    session.talk_to_exe()

另一种方法是稍微改变你的session夹具，将链接的创建委托给测试函数：

import pytest


@pytest.fixture
def session_factory(request):
    links = []

    def make_link(specific_file):
        copy_config_file(specific_file) 
        link = spawn_exe()
        links.append(link)
        return link 

    yield make_link

    for link in links:
        close_down_exe(link)

def test_1(session_factory):  
    session = session_factory('foo.xml')
    session.talk_to_exe()

def test_2(session):
    session = session_factory('bar.xml')
    session.talk_to_exe()

我更喜欢后者，因为它更容易理解，并允许以后进行更多改进，例如，如果您需要在基于配置值的测试中使用@parametrize。另请注意，后者允许在同一测试中生成多个可执行文件。

如何将值传递给Pytest fixture

1 个答案: