如何在 pytest 中复用 TCP 服务器实例以提升测试效率

夜晨同学_2110

夜晨同学_2110

2026-09-01

721人浏览

原创

如何在 pytest 中复用 TCP 服务器实例以提升测试效率

本文详解如何通过正确配置 pytest-asyncio 的 loop_scope 与 fixture 作用域,实现跨测试复用同一个异步 TCP 服务器,避免重复启停开销,并解决因事件循环不匹配导致的 RuntimeError: Task got Future attached to a different loop 错误。

本文详解如何通过正确配置 `pytest-asyncio` 的 loop_scope 与 fixture 作用域,实现跨测试复用同一个异步 tcp 服务器,避免重复启停开销,并解决因事件循环不匹配导致的 `runtimeerror: task got future attached to a different loop` 错误。

在使用 pytest 编写异步网络测试时,一个常见需求是:让多个测试共享同一个 TCP 服务器实例,而非每个测试都启动/关闭一次——这不仅能显著提升测试执行速度,还能更真实地模拟长连接服务场景。但直接将 @pytest.fixture(scope="module") 应用于异步 fixture(如 async def tcp_server())会导致运行时错误:

RuntimeError: Task ... got Future attached to a different loop

该错误的根本原因是:pytest-asyncio 默认为每个 async fixture 创建独立的事件循环(loop),而 scope="module" 的 fixture 生命周期跨越多个测试函数,若其内部 asyncio.Event、asyncio.Task 等对象绑定到某个特定 loop,后续测试中若在另一个 loop 中 await 它们,就会触发跨 loop 引用异常。

✅ 正确解法是显式统一 fixture 的事件循环作用域(loop_scope) 与 fixture 自身的作用域(scope),确保二者协同一致。

✅ 推荐配置(稳定可靠)

# test_tcp.py
import pytest
import pytest_asyncio

# 全局设置:所有 async 测试使用 session 级别事件循环
pytestmark = pytest.mark.asyncio(loop_scope="session")

@pytest_asyncio.fixture(loop_scope="session", scope="module")
async def tcp_server():
    server = TCPServer('localhost', 3102)
    # 启动服务器任务(注意:不 await,仅创建 task)
    task = asyncio.create_task(server.start())
    # 确保服务器已就绪(可选:加简短等待或健康检查)
    await asyncio.sleep(0.05)
    yield server
    # 清理:取消任务并等待优雅退出
    task.cancel()
    try:
        await task
    except asyncio.CancelledError:
        pass

⚠️ 关键点说明:

  • 必须使用 @pytest_asyncio.fixture(而非原生 @pytest.fixture);
  • loop_scope="session" 表示整个测试会话共用同一个事件循环;
  • scope="module" 表示 fixture 在模块内只初始化一次,供该模块下所有测试复用;
  • loop_scope 必须 ≥ scope(即 "session" > "module" > "function"),否则 pytest-asyncio 无法保证 loop 复用一致性。

? 测试函数写法(保持简洁)

async def test_motor_identification_dataFL(tcp_server):
    tcp_server.set_header_function(lambda: create_motor_identification_header(110))

    try:
        await asyncio.wait_for(tcp_server.data_received_event.wait(), timeout=5.0)
    except asyncio.TimeoutError:
        pytest.fail("Timeout waiting for client data")

    assert tcp_server.last_received_data == b'\x00\x06'

无需额外装饰器(因已通过 pytestmark 全局声明),也无需手动管理 loop。

? 常见误区与规避建议

  • ❌ 错误:混合使用 @pytest.fixture 和 async def —— 必须改用 @pytest_asyncio.fixture;
  • ❌ 错误:loop_scope="function" + scope="module" —— loop 生命周期短于 fixture,必然报错;
  • ⚠️ 注意:TCPServer.start() 是无限 serve_forever(),务必通过 task.cancel() 主动终止,否则测试进程无法退出;
  • ? 提升健壮性:可在 yield 前添加简易连接探测(如 await asyncio.open_connection(...)),确保端口已监听再继续;
  • ? 环境要求:确保安装 pytest-asyncio>=0.24.0(支持 loop_scope 参数),并禁用旧版兼容警告(在 pyproject.toml 中添加):
    [tool.pytest.ini_options]
    asyncio_default_fixture_loop_scope = "session"

通过以上配置,你就能安全、高效地在多个 pytest 测试间复用同一个异步 TCP 服务器,兼顾性能与稳定性。

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
服务器是什么
服务器是什么

服务器是一种计算机硬件设备或软件程序,它具有强大的计算和存储能力,用请求、存储数据和提供服务。它在互联网中着关重要的作用,为用户提供各种服务和资源。本专题为大家提供服务器相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.15

417

5

连接apple id服务器时出错
连接apple id服务器时出错

连接apple id服务器时出错的原因包括网络连接问题、服务器问题、Apple ID账户问题、设备问题、防火墙或安全软件问题、时间和日期设置问题、Apple服务器维护等。本专题为大家提供apple id相关的文章、下载、课程内容,供大家免费下载体验。

2023.09.08

840

5

搭建互联网服务器
搭建互联网服务器

搭建互联网服务器需要:1、选择合适的硬件和操作系统,第一步是选择合适的硬件和操作系统;2、安装和配置操作系统,是搭建互联网服务器的关键步骤;3、安装和配置服务器软件,是搭建互联网服务器的下一步,常见的服务器软件包括Apache、Nginx、Tomcat等;4、配置防火墙和安全性,是搭建互联网服务器的重要步骤;5、域名解析和配置,是搭建互联网服务器的最后一步。

2023.09.19

2432

5

如何查看服务器状态
如何查看服务器状态

查看服务器状态的方法有使用命令行工具、图形界面工具、监控工具、日志文件和远程管理工具等。本专题为大家提供服务器状态相关的文章、下载、课程内容,供大家免费下载体验。

2023.10.09

816

5

服务器域名转接慢怎么解决
服务器域名转接慢怎么解决

服务器域名转接慢的解决办法有DNS优化、服务器优化、CDN加速、前端优化和网络优化等。本专题为大家提供服务器相关的文章、下载、课程内容,供大家免费下载体验。

2023.10.17

709

5

服务器评测软件
服务器评测软件

服务器评测软件有PassMark Software、CPU-Z、GPU-Z、CrystalDiskMark、IOmeter、JMeter、LoadRunner、Apache Bench等等。详细介绍:1、PassMark Software是一款综合性的服务器性能测试软件,可以评估服务器在各种负载条件下的性能;2、CPU-Z是一款可以提供服务器CPU详细信息的软件等等。

2023.10.17

374

3

如何开启TFTP服务器
如何开启TFTP服务器

开启TFTP服务器的步骤包括选择TFTP服务器软件、下载和安装软件、配置TFTP服务器以及启动和测试服务器等。本专题为大家提供服务器相关的文章、下载、课程内容,供大家免费下载体验。

2023.10.18

2176

4

服务器负载不兼容怎么解决
服务器负载不兼容怎么解决

解决方法:1、增加服务器资源;2、负载均衡;3、优化应用程序;4、增加缓存机制;5、分布式架构;6、限流和熔断;7、自动化扩容。想知道更详细服务器负载不兼容的解决方法,可以访问本专题下面的文章。

2023.10.20

4012

4

宽带如何接入服务器
宽带如何接入服务器

宽带接入服务器的方法有ADSL宽带接入服务器、光纤接入服务器、无线接入服务器和以太网接入服务器等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2023.10.20

647

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.6万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 133.2万人学习