3

我正在使用 Python3.8 和 pytest。如何根据某些命令行规范为我的会话激活夹具?现在我有一个 tests/conftest.py 文件,其中包括

@pytest.fixture(name="fixture1", scope="session")
def fixture1():
    ...

@pytest.fixture(name="fixture2", scope="session")
def fixture2():
    ...

但我只希望会话中包含某些固定装置,并且我希望能够在 pytest.ini 文件中或通过传递给 pytest 的命令行参数指定它。我怎么做?

编辑:这是我的用例...

如果启用了命令行选项,我想依赖一个夹具

@pytest.fixture(name="az_sql_db_inst")
def az_db_connection_fixture(az_sql_creds, wait_for_sql_server)

但是如果没有启用命令行,我想依赖另一个夹具......

@pytest.fixture(name="az_sql_db_inst")
def az_db_connection_fixture(az_sql_creds, wait_for_docker_sql_server)
4

1 回答 1

4

您可以添加可用于启用固定装置的命令行标志:

conftest.py

def pytest_addoption(parser):
    parser.addoption('--option2', action='store_const', const=True)
    parser.addoption('--option2', action='store_const', const=True)

比你必须在你的夹具中检查这些参数:

@pytest.fixture(name="fixture1", scope="session", autouse=True)
def fixture1(request):
   # only use with option1 command line argument
   if request.config.getoption("--option1"):
       ...
    
@pytest.fixture(name="fixture2", scope="session", autouse=True)
def fixture2(request):
   # only use with option2 command line argument
   if request.config.getoption("--option2"):
       ...

@pytest.fixture(name="fixture3", scope="session", autouse=True)
def fixture3(request):
   # only use if option1 command line argument is not provided
   if not request.config.getoption("--option1"):
       ...

autouse=True在这里使用,因为我希望夹具执行不同的设置代码,当然您的使用可能会有所不同。

您现在可以调用:

  • pytest-> 不会应用任何夹具
  • pytest --option1-> 将应用fixture1
  • pytest --option1 --option2-> 两个夹具都将被应用

您还可以将这些参数添加到您的pytest.ini

[pytest]
# always apply fixture2
addopts = --option2

编辑:
关于继承夹具的后续问题,您可以执行以下操作:

@pytest.fixture
def wait_for_sql_server(request):
   if request.config.getoption("--my_option"):
       ...

@pytest.fixture
def wait_for_docker(request):
   if not request.config.getoption("--my_option"):
       ...

@pytest.fixture(name="az_sql_db_inst")
def az_db_connection_fixture(
    az_sql_creds, wait_for_sql_server, wait_for_docker):
    ...

EDIT2:
如果您不能自己编写或调整基本装置(wait_for_在问题的编辑部分),您可以采取另一种方式。

您可以在单独的插件中编写基本夹具的单独实现,并根据配置加载所需的插件:

plugin_docker.py

@pytest.fixture
def wait_for_service(wait_for_docker):
    yield

插件服务器.py

@pytest.fixture
def wait_for_service(wait_for_sql_server):
    yield

conftest.py

def pytest_addoption(parser):
    parser.addoption('--docker', action='store_const', const=True)

def pytest_configure(config):    
    use_docker = config.getoption("--docker")
    plugin_name = 'plugin_docker' if use_docker else 'plugin_server'
    if not config.pluginmanager.has_plugin(plugin_name):
        config.pluginmanager.import_plugin(plugin_name)

@pytest.fixture(name="az_sql_db_inst")
def az_db_connection_fixture(az_sql_creds, wait_for_service):
    ...    

固定装置只是实际固定装置的wait_for_service包装,但这样您可以在两种情况下从同一个固定装置派生。

于 2020-08-05T19:08:13.810 回答