如何使用pytest-testinfra模块验证运行中Docker容器的内容

我正在用testinfra测试CI/CD构建的Docker镜像，目标是连接到运行中的容器验证内容。已经写好pytest fixture并确认能成功连接容器，但现在遇到两个问题：

1. 想用testinfra的File模块测试容器内文件属性，但无法正常使用，也没有智能提示
2. 验证容器内是否安装Python3.7的测试代码无法正常工作

已实现的Fixture代码 #

# scope='session' 表示所有测试共用同一个容器；
# scope='function' 表示每个测试函数用新容器。
@pytest.fixture(scope='session')
def host(request):
    # 运行容器
    docker_id = subprocess.check_output(
        ['docker', 'run', '-d', 'gitlab.CHANGECOMPANYNAME.net:8443/mhs/SOMEURLNAE/cicd/platform-infrastructure/gitlab-tools/gitlab-runners/docker-in-docker:1.021']).decode().strip()
    # 返回连接到容器的testinfra实例
    yield testinfra.get_host("docker://" + docker_id)
    print("printing  dockerid" + docker_id)

（注：原代码中构建镜像的部分已注释，当前直接拉取指定镜像运行容器）

有问题的Python版本测试代码 #

def test_python_3_7(host):
"""
验证运行python命令时使用的是3.7版本
"""
python_version = host.command('python --version').stdout
#assert host.('3.7', python_version)

一、修复Python3.7验证测试 #

你的测试代码存在语法错误，且python --version的输出在多数环境下会输出到stderr而非stdout，断言逻辑也不完整。修改后的代码如下：

def test_python_3_7(host):
    """验证运行python命令时使用的是3.7版本"""
    # 同时捕获stdout和stderr，避免版本信息丢失
    result = host.command('python --version')
    version_output = result.stdout.strip() + result.stderr.strip()
    
    # 验证版本包含3.7标识
    assert '3.7' in version_output, f"Python版本不符，输出内容: {version_output}"
    
    # 可选：直接验证python3.7命令是否存在
    assert host.exists('python3.7'), "容器内未安装python3.7"

如果容器内python命令未指向3.7版本，可以直接用python3.7 --version来执行验证。

二、解决File模块使用及智能提示问题 #

1. 添加类型注解：给fixture的返回值添加类型注解，让IDE识别testinfra的Host对象类型，从而提供智能提示：

from testinfra.host import Host

@pytest.fixture(scope='session')
def host(request) -> Host:
    docker_id = subprocess.check_output(
        ['docker', 'run', '-d', 'gitlab.CHANGECOMPANYNAME.net:8443/mhs/SOMEURLNAE/cicd/platform-infrastructure/gitlab-tools/gitlab-runners/docker-in-docker:1.021']).decode().strip()
    yield testinfra.get_host("docker://" + docker_id)
    # 测试结束后自动清理容器，避免残留
    subprocess.check_call(['docker', 'stop', docker_id])
    subprocess.check_call(['docker', 'rm', docker_id])

2. 正确调用File模块：通过host.file()方法获取文件对象，示例如下：

def test_file_attributes(host):
    # 测试/etc/passwd文件的属性
    passwd_file = host.file('/etc/passwd')
    assert passwd_file.exists, "/etc/passwd文件不存在"
    assert passwd_file.is_file, "/etc/passwd不是普通文件"
    assert passwd_file.mode == 0o644, "/etc/passwd权限不符合预期"
    assert passwd_file.user == 'root', "/etc/passwd所有者不是root"
    assert passwd_file.group == 'root', "/etc/passwd所属组不是root"

3. IDE环境检查：确保当前Python环境已安装testinfra（执行pip install testinfra），且IDE已正确识别该环境的依赖包，必要时重启IDE生效。

额外优化：容器自动清理 #

原fixture未添加容器清理逻辑，测试后会残留运行中的容器，建议在yield之后加入docker stop和docker rm命令，自动清理测试用容器。

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