讲师中心 微信公众号
AI工具推荐 视频效率加速

Pytest 中 FastAPI 测试下内存数据库状态不持久的解决方案

冬杰同学_5656

冬杰同学_5656

发布时间:2026-05-13 15:53:35

|

297人浏览过

|

来源于php中文网

原创

Pytest 中 FastAPI 测试下内存数据库状态不持久的解决方案

在 FastAPI 单元测试中,若使用 @fixture(scope="function") 注入依赖类(如 UserRepo),其内部状态(如 self.database)不会跨请求保留——因为每个 HTTP 请求都会触发新实例化,导致内存数据丢失。

在 fastapi 单元测试中,若使用 `@fixture(scope="function")` 注入依赖类(如 `userrepo`),其内部状态(如 `self.database`)不会跨请求保留——因为每个 http 请求都会触发新实例化,导致内存数据丢失。

FastAPI 的依赖注入系统在每次请求时都会重新构建被 Depends() 调用的依赖对象。这意味着:即使你将 user_repo fixture 设置为 function 作用域,并在其中注册了 MockUserRepo 类作为依赖覆盖(app.dependency_overrides[UserRepo] = MockUserRepo),FastAPI 仍会在每个请求中调用 MockUserRepo() 构造函数,创建一个全新的实例 —— 因此 self.database = [] 在每次请求中都被重置,导致 test_get_user 中前序 POST 创建的用户无法在后续 GET 请求中查到。

✅ 正确做法:将共享状态提升至 fixture 作用域内(而非实例属性)

关键在于:把可变状态(如 database 列表)定义在 fixture 函数作用域中,而非 MockUserRepo 实例内部。这样,所有由该 fixture 创建的 MockUserRepo 实例都共享同一份数据:

@fixture(scope="function")
def user_repo(app: FastAPI):
    @dataclass
    class User:
        id: int
        username: str
        password: str

    # ✅ 共享状态:定义在 fixture 函数体内,生命周期与 fixture 一致
    database = []

    class MockUserRepo:
        def create_user(self, user_create_model: UserCreateModel, session: Session) -> User | None:
            if session:
                user = User(
                    id=len(database) + 1,
                    username=user_create_model.username,
                    password=user_create_model.password,
                )
                database.append(user)
                return user

        def get_user_by_id(self, id: int, session: Session) -> User | None:
            if session:
                return next((u for u in database if u.id == id), None)

    app.dependency_overrides[UserRepo] = MockUserRepo
    yield  # fixture cleanup 不需要显式操作;database 随 fixture 结束自动回收

? 提示:yield 后无需 app.dependency_overrides.pop(UserRepo),因为 scope="function" 确保每次测试后 fixture 自动销毁,且 dependency_overrides 是字典引用,下次测试会重新赋值,无残留风险。

⚠️ 注意事项

  • ❌ 错误写法:self.database = [] 在 __init__ 中初始化 → 每个请求新建实例,状态隔离;
  • ✅ 正确写法:database = [] 在 fixture 函数体中定义 → 所有同 fixture 生命周期内的 MockUserRepo 实例共享;
  • 若需跨测试用例共享(如 session scope),需额外确保线程安全(本例中单线程 pytest 默认安全);
  • 此方案适用于轻量级内存模拟,真实集成测试仍应使用 pytest-asyncio + 真实 SQLite 内存数据库(sqlite:///:memory:)并配合 engine.begin() 管理事务。

通过将状态上提至 fixture 作用域,即可在单个测试函数内实现多请求间的数据一致性,精准模拟“内存数据库”行为,大幅提升 FastAPI 单元测试的可靠性与可维护性。

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

热门AI工具

更多
切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

相关专题

更多
Python FastAPI异步API开发_Python怎么用FastAPI构建异步API
Python FastAPI异步API开发_Python怎么用FastAPI构建异步API

Python FastAPI 异步开发利用 async/await 关键字,通过定义异步视图函数、使用异步数据库库 (如 databases)、异步 HTTP 客户端 (如 httpx),并结合后台任务队列(如 Celery)和异步依赖项,实现高效的 I/O 密集型 API,显著提升吞吐量和响应速度,尤其适用于处理数据库查询、网络请求等耗时操作,无需阻塞主线程。

119

2025.12.22

Python 微服务架构与 FastAPI 框架
Python 微服务架构与 FastAPI 框架

本专题系统讲解 Python 微服务架构设计与 FastAPI 框架应用,涵盖 FastAPI 的快速开发、路由与依赖注入、数据模型验证、API 文档自动生成、OAuth2 与 JWT 身份验证、异步支持、部署与扩展等。通过实际案例,帮助学习者掌握 使用 FastAPI 构建高效、可扩展的微服务应用,提高服务响应速度与系统可维护性。

534

2026.02.06

Python Web框架FastAPI 全栈开发教程合集
Python Web框架FastAPI 全栈开发教程合集

以 FastAPI 为核心,讲解现代 Python Web API 的高效开发方式,涵盖路由定义与路径参数/查询参数/请求体绑定、Pydantic 模型的数据校验与序列化、依赖注入(Depends)系统的分层设计、中间件与 CORS 配置、OAuth2 + JWT 认证流程、后台任务(BackgroundTasks)、WebSocket 实时通信、SQLAlchemy 异步 ORM 集成、自动生成 OpenAPI/Swagger 交互文

536

2026.05.09

Python FastAPI异步微服务与高性能接口设计
Python FastAPI异步微服务与高性能接口设计

本专题聚焦 Python FastAPI 框架在高性能接口与微服务开发中的应用,讲解异步请求处理、依赖注入机制、路由设计、数据库异步操作以及接口性能优化策略。结合实际项目案例,帮助开发者构建高并发、低延迟的现代化后端服务架构。

439

2026.06.16

C++运算符基础入门
C++运算符基础入门

本专题详细讲解了C++运算符的类型、语法与使用方法,涵盖算术运算符、关系运算符、逻辑运算符、位运算符、赋值运算符、条件运算符及其他特殊运算符,并通过代码示例解析优先级与结合性。

0

2026.10.09

PixPix官网入口合集
PixPix官网入口合集

本专题汇总了PixPix官网在线使用入口及平台功能详解,涵盖文生图、图生图、AI图片编辑、AI视频创作等核心能力,并整理了AI爆款图片复刻、商品套图、详情页生成、视频变清晰与去水印等电商专项工具的使用教程。同时收录了PixPix MCP接入Codex、Claude Code等主流Agent的操作指南,助您一站式完成AI图片与视频创作。

0

2026.10.09

FrankenPHP集成Laravel详细教程
FrankenPHP集成Laravel详细教程

本专题提供FrankenPHP集成Laravel的详细配置指南,全面解析运行原理、开发环境搭建、Caddyfile配置、Octane工作模式、数据库连接、队列任务、定时任务和生产环境优化,解决部署过程中常见的报错与兼容性问题。

60

2026.10.08

LLVM自定义Pass怎么写
LLVM自定义Pass怎么写

本专题聚焦LLVM自定义Pass开发,整理Pass类结构、run()方法、PreservedAnalyses、CMake构建、插件注册、-load-pass-plugin加载和测试用例编写流程。

160

2026.09.30

LLVM RISC-V参数配置教程
LLVM RISC-V参数配置教程

本专题介绍LLVM对RISC-V基础ISA和扩展的支持方式,涵盖RV32、RV64、标准扩展、实验性扩展、厂商扩展、-menable-experimental-extensions和版本差异。

140

2026.09.30

热门下载

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

精品课程

更多
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn