如何在pytest/testinfra中复用hosts参数实现自定义fixture？

你的需求完全可行，Testinfra基于pytest生态，能轻松实现自定义fixture复用命令行参数、对接配置管理工具，结合pytest.mark.parametrize做批量验证。以下是具体实现步骤：

1. 核心思路 #

• 通过pytest的config fixture读取命令行传入的--hosts参数
• 编写函数对接你的配置管理工具（比如Ansible/SaltStack），根据主机名获取对应配置数据
• 自定义fixture结合Testinfra的host fixture与配置数据，供测试函数调用
• 用pytest.mark.parametrize批量验证配置项

2. 代码实现示例 #

第一步：编写配置数据获取函数 #

替换成你实际用的配置管理工具逻辑，这里以模拟数据+Ansible为例：

def get_server_config(hostname):
    # 示例1：模拟配置数据（替换为你的真实逻辑）
    # configs = {
    #     "web01": {"service": "nginx", "port": 80},
    #     "db01": {"service": "mysql", "port": 3306}
    # }
    # return configs.get(hostname, {})

    # 示例2：从Ansible Inventory获取主机变量
    from testinfra.utils.ansible_runner import AnsibleRunner
    runner = AnsibleRunner("inventory.yml")
    host_vars = runner.get_host_vars(hostname)
    return {
        "service": host_vars.get("service_name"),
        "listen_port": host_vars.get("listen_port")
    }

第二步：自定义Fixture结合主机与配置 #

这个fixture会自动关联当前测试的主机，并拉取对应配置：

import pytest

@pytest.fixture
def host_with_config(host):
    # 从Testinfra的host对象拿到当前主机名
    hostname = host.backend.hostname
    # 获取该主机的配置数据
    config = get_server_config(hostname)
    # 返回主机对象+配置的组合
    return {"host": host, "config": config}

第三步：结合parametrize做批量验证 #

用pytest.mark.parametrize批量验证不同的配置项，不用写重复代码：

@pytest.mark.parametrize("check_type", ["service_running", "port_listening"])
def test_server_deployment(host_with_config, check_type):
    host = host_with_config["host"]
    config = host_with_config["config"]
    
    if check_type == "service_running":
        service = host.service(config["service"])
        assert service.is_running, f"{config['service']} 未运行"
        assert service.is_enabled, f"{config['service']} 未设置开机自启"
    elif check_type == "port_listening":
        socket = host.socket(f"tcp://0.0.0.0:{config['listen_port']}")
        assert socket.is_listening, f"端口 {config['listen_port']} 未监听"

3. 复用--hosts参数到外部函数 #

如果需要在全局批量获取所有主机的配置，可通过config fixture读取命令行的--hosts参数：

@pytest.fixture(scope="session")
def all_target_hosts(config):
    # 读取并拆分命令行传入的hosts列表
    hosts_str = config.getoption("--hosts")
    return [h.strip() for h in hosts_str.split(",") if h.strip()]

@pytest.fixture(scope="session")
def all_hosts_config(all_target_hosts):
    # 批量获取所有目标主机的配置
    return {host: get_server_config(host) for host in all_target_hosts}

4. 运行测试 #

和正常Testinfra命令一样传入--hosts参数即可：

pytest --hosts=web01,db01 test_deployment.py

内容的提问来源于stack exchange，提问作者NicoW