0

0

Python如何测试数据库_利用Fixture创建临时测试库(sqlite/:memory:)并在结束后销毁恢复

P粉602998670

P粉602998670

发布时间:2026-03-11 13:01:42

|

686人浏览过

|

来源于php中文网

原创

pytest中sqlite :memory: 测试报“no such table”是因为每个连接独享内存库,需在fixture中创建engine后立即执行base.metadata.create_all(),且fixture应返回session实例而非sessionmaker,确保测试使用同一连接。

python如何测试数据库_利用fixture创建临时测试库(sqlite/:memory:)并在结束后销毁恢复

pytest里用sqlite:///:memory:为什么测试总报“no such table”

因为:memory:数据库是每个连接独享的,create_engine建一次就开一个新内存库,而ORM模型没在那个连接里执行Base.metadata.create_all(),表根本不存在。Django的TestCase自动建表,但纯SQLAlchemy+pytest不会。

  • 每次sessionconnection新建,都得手动触发Base.metadata.create_all(engine)
  • 别在conftest.py顶层import后直接调用create_all——那时engine还没创建,或指向的是别的库
  • 如果用了scoped_session,确保create_all在fixture创建engine之后、session初始化之前执行

写fixture时scope="function"scope="session"怎么选

选错scope会导致测试间污染或重复建库拖慢速度。:memory:库无法跨进程/线程共享,所以哪怕设成session,pytest-xdist并行时每个worker仍会拿到独立内存库——这时session反而不如function直观。

  • scope="function":最安全,默认推荐。每个测试函数独占干净:memory:库,无需清理
  • scope="session":仅当所有测试串行运行(禁用-n)、且建库开销大(比如含大量初始数据)时考虑;但必须确认没有测试修改全局Baseengine状态
  • 别用scope="module":模块内测试共享一个库,但若某个测试commit了脏数据,后续测试可能失败,排查困难

如何让fixture返回的enginesession能被测试函数直接用

关键是把enginesession绑定到同一个:memory:实例,并确保测试中使用的session来自该engine。常见错误是fixture返回engine,测试里又自己sessionmaker(bind=engine)(),结果session连的是另一个内存库。

Text-To-Song
Text-To-Song

免费的实时语音转换器和调制器

下载
  • fixture应返回session实例(不是sessionmaker),并在内部完成create_all和session初始化
  • 测试函数参数名必须和fixture名一致,例如def test_user_create(db_session):,其中db_session是fixture名
  • 如果测试需要engine(比如执行raw SQL),在fixture里一并yield出来,如yield {"engine": engine, "session": session},但注意这会让测试代码变啰嗦
import pytest
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
<p>@pytest.fixture(scope="function")
def db_session():
engine = create_engine("sqlite:///:memory:")
Base.metadata.create_all(engine)  # 表在这里建好
Session = sessionmaker(bind=engine)
session = Session()
yield session
session.close()  # 内存库随session销毁自动丢弃,不用drop_all

为什么测试跑完数据库没清空,或者报OperationalError: database is locked

:memory:库本身不需要手动清理——Python对象销毁、连接关闭后,内存就释放了。所谓“没清空”,其实是上个测试没session.close()或没rollback(),导致连接还挂着锁;而database is locked通常发生在多线程/协程复用同一:memory:引擎时,但pytest默认单线程,所以大概率是fixture没正确关闭session。

立即学习Python免费学习笔记(深入)”;

  • 务必在fixture的yield后调用session.close(),不能只靠__del__或GC
  • 避免在测试里调用session.commit()后不close()——:memory:虽轻量,但未关闭的session会持锁阻塞下一个测试的同名表操作
  • 如果用了fastapi.testclient之类集成测试,确保依赖注入的db session也被正确reset,而不是复用fixture里的长生命周期session

真正容易被忽略的点是:你以为:memory:能自动隔离,其实隔离边界在连接层,不是进程层;而fixture的生命周期管理稍有松动,就会让锁或未提交事务跨测试泄漏。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
数据分析工具有哪些
数据分析工具有哪些

数据分析工具有Excel、SQL、Python、R、Tableau、Power BI、SAS、SPSS和MATLAB等。详细介绍:1、Excel,具有强大的计算和数据处理功能;2、SQL,可以进行数据查询、过滤、排序、聚合等操作;3、Python,拥有丰富的数据分析库;4、R,拥有丰富的统计分析库和图形库;5、Tableau,提供了直观易用的用户界面等等。

1133

2023.10.12

SQL中distinct的用法
SQL中distinct的用法

SQL中distinct的语法是“SELECT DISTINCT column1, column2,...,FROM table_name;”。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

340

2023.10.27

SQL中months_between使用方法
SQL中months_between使用方法

在SQL中,MONTHS_BETWEEN 是一个常见的函数,用于计算两个日期之间的月份差。想了解更多SQL的相关内容,可以阅读本专题下面的文章。

381

2024.02.23

SQL出现5120错误解决方法
SQL出现5120错误解决方法

SQL Server错误5120是由于没有足够的权限来访问或操作指定的数据库或文件引起的。想了解更多sql错误的相关内容,可以阅读本专题下面的文章。

2152

2024.03.06

sql procedure语法错误解决方法
sql procedure语法错误解决方法

sql procedure语法错误解决办法:1、仔细检查错误消息;2、检查语法规则;3、检查括号和引号;4、检查变量和参数;5、检查关键字和函数;6、逐步调试;7、参考文档和示例。想了解更多语法错误的相关内容,可以阅读本专题下面的文章。

380

2024.03.06

oracle数据库运行sql方法
oracle数据库运行sql方法

运行sql步骤包括:打开sql plus工具并连接到数据库。在提示符下输入sql语句。按enter键运行该语句。查看结果,错误消息或退出sql plus。想了解更多oracle数据库的相关内容,可以阅读本专题下面的文章。

1663

2024.04.07

sql中where的含义
sql中where的含义

sql中where子句用于从表中过滤数据,它基于指定条件选择特定的行。想了解更多where的相关内容,可以阅读本专题下面的文章。

585

2024.04.29

sql中删除表的语句是什么
sql中删除表的语句是什么

sql中用于删除表的语句是drop table。语法为drop table table_name;该语句将永久删除指定表的表和数据。想了解更多sql的相关内容,可以阅读本专题下面的文章。

440

2024.04.29

C# ASP.NET Core微服务架构与API网关实践
C# ASP.NET Core微服务架构与API网关实践

本专题围绕 C# 在现代后端架构中的微服务实践展开,系统讲解基于 ASP.NET Core 构建可扩展服务体系的核心方法。内容涵盖服务拆分策略、RESTful API 设计、服务间通信、API 网关统一入口管理以及服务治理机制。通过真实项目案例,帮助开发者掌握构建高可用微服务系统的关键技术,提高系统的可扩展性与维护效率。

3

2026.03.11

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.5万人学习

Django 教程
Django 教程

共28课时 | 4.9万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.9万人学习

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号