时间戳

首页 > 解决方案 > 如何使用 pytest 选项作为夹具而不重复自己？

python - 如何使用 pytest 选项作为夹具而不重复自己？

问题描述

我有一个测试套件，其中conftest.py定义了一些选项和一些固定装置来检索它们：

def pytest_addoption(parser):
  parser.addoption("--ip", action="store")
  parser.addoption("--port", action="store")

@pytest.fixture
def ip(request):
  return request.config.getoption("ip")

@pytest.fixture
def port(request):
  return request.config.getoption("ip")

（我输入了一个复制粘贴错误来说明问题）

我的测试可以非常雄辩地表达他们需要的选项：

def test_can_ping(ip):
  ...

def test_can_net_cat(ip, port):
  ...

但 ...

我试图避免在这里重复自己：我必须在三个地方指定配置参数的名称才能使其工作。

如果我有这样的东西，我本可以避免复制粘贴错误：

# does not exist:
@pytest.option_fixture
def ip(request, parser):
  return request.config.getoption(this_function_name)

或这个

def pytest_addoption(parser):
  # does not exist: an as_fixture parameter
  parser.addoption("--ip", action="store", as_fixture=True)
  parser.addoption("--port", action="store", as_fixture=True)

有没有办法告诉 pytest 添加一个选项和一个相应的夹具来实现 DRY/SPOT 代码？

标签： pythonpytestdryfixtures

解决方案

与其更改pytest装饰器，不如创建您自己的装饰器：

parse_options = []

@addOption(parse_options)
@pytest
def ip(...): ...

装饰器不必修改传入的函数。因此，在这种情况下，查看方法对象，使用f.__name__获取名称并在列表parse_options中为其添加一个条目。

下一步是修改pytest_addoption以迭代列表并创建选项。在函数执行的时候，装饰器应该已经完成​​了他们的工作。

推荐阅读