如何用pytest fixture销毁Docker容器？解决参数化测试异常问题

我来帮你搞定这个问题，你当前遇到两个核心问题：一是pytest的fixture默认作用域是function，导致每个参数化测试用例都会重新执行fixture代码，反复创建销毁容器；二是你只调用了containers.create创建容器，但没有启动它，所以容器列表是空的，服务根本没运行起来。下面一步步给你修正：

1. 调整Fixture的作用域 #

把fixture的scope参数设置为module（整个测试模块只执行一次）或者session（整个测试会话只执行一次），这样所有参数化测试就能共用同一个容器，不用反复启停。

2. 正确启动并配置Docker容器 #

调用containers.create后，必须调用start()方法启动容器。另外，要配置端口映射，把容器内的5000端口映射到主机的5000端口，这样你的测试才能通过localhost:5000访问服务。

3. 安全销毁容器 #

在fixture的teardown阶段（yield之后的代码），先停止容器（可选，但更稳妥），然后用remove(force=True)确保即使容器还在运行也能被移除。

4. 修正解析函数和测试逻辑 #

你的parse_ingredient函数逻辑有问题，现在的循环完全没必要，而且没有返回结果。需要修正它来正确发送请求并返回解析后的对象，然后在测试里添加断言验证结果是否符合预期。

完整修正后的代码 #

import pprint
import json
import requests
import pytest
import time

local_testdata = [
    ("1/2 of an apple", {"product": "apple", "unit": "item", "quantity": "1/2"}),
    ("12 oz frozen spinach", {"product": "spinach", "unit": "oz", "quantity": "12"}),
    ("24 oz frozen strawberries", {"product": "strawberries", "unit": "oz", "quantity": "24"})
]

def parse_ingredient(ingredient_string):
    ingreedy_path = "http://localhost:5000/query"
    payload = {"query": ingredient_string}
    # 设置正确的Content-Type头，否则服务可能无法解析请求
    headers = {"Content-Type": "application/json"}
    result = requests.post(ingreedy_path, data=json.dumps(payload), headers=headers)
    result.raise_for_status()  # 如果请求失败，抛出异常，方便调试
    return json.loads(result.text)

@pytest.fixture(scope="module")
def ingreedy_service():
    import docker
    client = docker.from_env()
    # 创建容器时配置端口映射，启动后调用start()
    ingreedy = client.containers.create(
        "openfoodbroker/ingreedy.service",
        name="ingreedy_service",
        detach=True,
        ports={"5000/tcp": 5000}  # 端口映射：容器5000 -> 主机5000
    )
    ingreedy.start()
    print("启动Docker容器中的Ingreedy服务")
    # 等待服务初始化，时间可以根据实际情况调整
    time.sleep(3)
    # 打印容器列表，确认容器已启动
    pprint.pprint(client.containers.list())
    yield ingreedy  # 把容器对象yield出去，方便测试中如果需要可以操作
    print("销毁Docker容器")
    # 停止并移除容器，force=True确保强制移除
    ingreedy.stop()
    ingreedy.remove(force=True)

@pytest.mark.parametrize("ingredient_string, expected_result", local_testdata)
def test_ingredient_parser_localdata(ingredient_string, expected_result, ingreedy_service):
    print(f"\n执行测试：[{ingredient_string}]")
    result = parse_ingredient(ingredient_string)
    # 断言解析结果是否符合预期，注意可能需要根据实际返回结构调整
    assert result["product"] == expected_result["product"]
    assert result["unit"] == expected_result["unit"]
    assert result["quantity"] == expected_result["quantity"]

关键说明 #

• Fixture作用域：scope="module"确保容器在整个测试模块启动一次，所有参数化测试共用这个容器，避免重复创建销毁。
• 容器启动与端口映射：ports={"5000/tcp": 5000}把容器的5000端口映射到主机，测试才能正常访问；ingreedy.start()真正启动容器。
• 请求头设置：添加Content-Type: application/json头，确保服务能正确解析JSON payload。
• 异常处理：result.raise_for_status()会在请求失败（比如4xx、5xx）时抛出异常，帮助你快速定位问题。
• 容器销毁：先stop()再remove(force=True)，确保容器被彻底清理。

这样修改后，你的测试会只启动一次容器，所有参数化测试复用这个容器，测试完毕后自动销毁容器，而且服务能正常响应请求啦。

内容的提问来源于stack exchange，提问作者Open Food Broker