在 Pytest 这个大家最喜欢的 Python 测试框架中，fixture 是一段可重用的代码，它在测试进入之前安排something，并在测试退出后进行清理。例如，临时文件或文件夹、设置环境、启动 Web 服务器等。在这篇文章中，我们将了解如何创建 Pytest 夹具，该夹具创建一个可以清理的测试数据库（空或已知状态） ，允许每个测试在完全干净的数据库上运行.
目标
我们将使用 Psycopg 3 创建一个 Pytest 夹具来准备和清理测试数据库。因为空数据库对测试几乎没有帮助，所以我们将选择应用 Yoyo 迁移（在撰写本文时网站已关闭，请转到 archive.org 快照）来填充它。
因此，对本博文中创建的名为 test_db 的 Pytest 夹具的要求是：
- 删除测试数据库如果在测试之前存在
- 测试前创建一个空数据库可选地
在测试之前应用迁移或创建测试数据
- 提供到测试数据库
的连接到测试
- 测试后删除测试数据库
（即使失败）
- 任何通过列出测试方法参数来请求它的测试方法：
雷雷
将收到连接到测试数据库的常规 Psycopg Connection 实例。测试可以做任何需要做的事情，就像普通的 Psycopg 常见用法一样，例如：
雷雷
动机和替代方案
看起来有一些 Pytest 插件承诺为依赖数据库的测试提供 PostgreSQL 固定装置。它们可能很适合你。
我尝试过 pytest-postgresql ，它也有同样的承诺。在编写自己的装置之前我已经尝试过它，但我无法让它为我工作。也许是因为他们的文档让我很困惑。
另一个，pytest-dbt-postgres，我根本没有尝试过。
项目文件布局
在经典的Python项目中，源代码位于src/中，测试位于tests/中：
雷雷
如果你使用像 Fantasyal Yoyo 这样的迁移库，迁移脚本很可能在migrations/:
雷雷
配置
我们的测试数据库夹具将需要很少的配置：
连接 URL
- （无数据库）
- 测试数据库名称
- 将为每个测试重新创建
- （可选）迁移文件夹
- 应用于每个测试的迁移脚本
- Pytest 有一个天然的地方 conftest.py 用于跨多个文件共享固定装置。灯具配置也会在那里：
雷雷
您可以从环境变量或任何适合您情况的值来设置这些值。
创建 test_db 夹具
有了
PostgreSQL和Psycopg库的知识
，在conftest.py中编写fixture：
雷雷创建迁移固定装置
在我们的例子中，我们使用
Yoyo 迁移
。将应用迁移编写为另一个名为 yoyo 的固定装置：
雷雷如果你想
将迁移应用到每个测试数据库
，需要 yoyo 夹具用于 test_db 夹具：
雷雷要
仅将迁移应用于某些测试
，需要单独使用yoyo：
雷雷结论
构建自己的装置来为您的测试提供一个干净的数据库对我来说是一次有益的经历，让我能够更深入地研究 Pytest 和 Postgres。
我希望这篇文章对您自己的数据库测试套件有所帮助。请随时在评论中留下您的问题，祝您编码愉快！
以上是Pytest 和 PostgreSQL：每次测试的新数据库的详细内容。更多信息请关注PHP中文网其他相关文章！
