如何在pytest中以后台线程运行完整Flask服务器，支持端点回调？

问题场景 #

常规Flask测试使用官方推荐的test_client模拟请求，但当端点业务逻辑需要向自身发起真实HTTP回调时（比如依赖的第三方库只能指定URL而非注入请求处理逻辑），test_client的模拟机制就无法满足需求，此时需要启动真实的Flask服务器在后台运行。

实现方案 #

1. 编写后台服务器Fixture #

利用Python的threading模块在后台线程启动真实的Flask服务器，并在测试完成后自动关闭服务。

import pytest
import threading
from my_project import create_app
from werkzeug.serving import make_server

@pytest.fixture(scope="module")
def app():
    # 创建测试用的Flask应用实例
    app = create_app()
    app.config['TESTING'] = True
    return app

@pytest.fixture(scope="module")
def live_server(app):
    # 创建服务器，指定localhost和随机可用端口（port=0让系统分配）
    server = make_server('localhost', 0, app)
    # 获取系统分配的实际端口
    port = server.socket.getsockname()[1]
    
    # 启动后台线程运行服务器
    server_thread = threading.Thread(target=server.serve_forever)
    server_thread.daemon = True
    server_thread.start()
    
    # 返回服务器实例和端口，供测试用例使用
    yield {"server": server, "port": port}
    
    # 测试结束后关闭服务器
    server.shutdown()

2. 编写带回调逻辑的测试用例 #

使用真实HTTP客户端（如requests）向后台服务器发起请求，验证回调逻辑是否正常：

import requests

def test_foobar_with_callback(live_server):
    # 构造请求URL
    request_url = f"http://localhost:{live_server['port']}/foobar"
    # 发起真实HTTP请求
    response = requests.get(request_url)
    # 验证响应状态码
    assert response.status_code == 200

关键注意点 #

• Fixture作用域：使用scope="module"可以让服务器在整个测试模块中只启动一次，避免重复启停带来的性能损耗。
• 端口冲突规避：通过port=0让系统分配随机可用端口，避免本地已有服务占用固定端口导致的启动失败。

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