我如何使用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):
  ...

但是…

我试图避免在这里重复自己:我必须在三个位置指定config参数的名称,以使其起作用.

如果我有类似以下内容,则可以避免复制粘贴错误:

# 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代码?

解决方法:

经过一些测试,我发现有些工作.这可能不是最好的方法,但我认为这很令人满意.
除了这两个测试,下面的所有代码都已添加到conftest.py模块中.

首先定义一个包含选项数据的字典:

options = {
    'port': {'action': 'store', 'help': 'TCP port', 'type': int},
    'ip': {'action': 'store', 'help': 'IP address', 'type': str},
}

我们可以在没有帮助和输入的情况下进行操作,但是稍后它将具有一定的实用性.

然后,您可以使用以下选项创建pytest选项:

def pytest_addoption(parser):
    for option, config in options.items():
        parser.addoption(f'--{option}', **config)

此时,pytest –help提供了这一点(请注意提供方便文档的帮助数据用法):

usage: pytest [options] [file_or_dir] [file_or_dir] [...]
...
custom options:
  --port=PORT           TCP port
  --ip=IP               IP address

最后,我们必须定义固定装置.为此,我提供了一个make_fixture函数,该函数在conftest.py读取时在循环中使用,以动态创建灯具并将其添加到模块的全局范围中:

def make_fixture(option, config):
    func = lambda request: request.config.getoption(option)
    func.__doc__ = config['help']
    globals()[option] = pytest.fixture()(func)


for option, config in options.items():
    make_fixture(option, config)

同样,“帮助”数据用于为创建的灯具建立文档字符串并对其进行记录.因此,调用pytest –fixtures将输出以下内容:

...
---- fixtures defined from conftest ----
ip
    IP address
port
    TCP port

调用pytest –port 80 –ip 127.0.0.1,并通过以下两个非常简单的测试,似乎验证了这一技巧(此处类型数据显示了其实用性,它使pytest将端口转换为int而不是字符串):

def test_ip(ip):
    assert ip == '127.0.0.1'


def test_ip_port(ip, port):
    assert ip == '127.0.0.1'
    assert port == 80

(非常有趣的问题,我想看到更多类似的问题)

上一篇:day42--new操作符具体干了什么?


下一篇:DRY原则和Shy原则