处理 Pytest-Django 中的非托管模型的策略
在 Django 项目中测试非托管模型(managed = False)可能很棘手,尤其是在测试环境包含托管和非托管模型混合,或涉及多个数据库时。本文探讨几种使用 pytest-django 测试非托管模型的方法,并分析其优缺点。
方法一:临时将所有模型标记为托管
最简单的解决方法是,在测试期间暂时将所有非托管模型标记为托管。 这可以通过修改 conftest.py 文件实现:
@pytest.hookimpl(tryfirst=True) def pytest_runtestloop(): from django.apps import apps for app in apps.get_app_configs(): for model in app.get_models(): if not model._meta.managed: model._meta.managed = True
登录后复制
重要:此方法需要在 pytest 命令中添加 --no-migrations 选项。
优点: 简单易行。
缺点: 绕过了迁移测试,这在多人协作开发时可能导致问题。
方法二:手动创建和删除非托管模型
更稳妥的方法是在测试设置中手动创建和删除非托管模型,确保迁移得到测试。 可以使用以下 pytest fixture:
@pytest.fixture(scope="session", autouse=True) def django_db_setup(django_db_blocker, django_db_setup): with django_db_blocker.unblock(): for connection in connections.all(): with connection.schema_editor() as schema_editor: setup_unmanaged_models(connection, schema_editor) yield def setup_unmanaged_models(connection, schema_editor): from django.apps import apps for model in apps.get_models(): if not model._meta.managed: if connection.introspection.table_names(cursor=connection.cursor()): schema_editor.delete_model(model) schema_editor.create_model(model)
登录后复制
优点: 包含迁移测试。
缺点: 实现较为复杂;transaction=True 不适用(详见下文)。
事务测试与非托管模型
pytest-django 提供了 django_db 和 django_db(transaction=True) 两个 fixture。
- django_db:在测试结束时回滚数据库更改。
- django_db(transaction=True):在每个测试用例后提交更改并截断数据库表。 只有托管模型会被截断。
方法三:在事务测试中临时将非托管模型标记为托管
为了使 transaction=True 与非托管模型兼容,可以在测试运行期间将非托管模型临时标记为托管,使其包含在截断过程中:
def setup_unmanaged_models(connection, schema_editor): # ... (代码同上) ... for model in unmanaged_models: # ... (代码同上) ... model._meta.managed = True # 关键:临时标记为托管
登录后复制
方法四:使用 on_commit hooks 避免 transaction=True (如果适用)
如果你的测试涉及 on_commit hooks,可以使用 pytest-django (>= v.4.4) 的 django_capture_on_commit_callbacks fixture 来避免使用 transaction=True:
@pytest.mark.django_db def mytest(django_capture_on_commit_callbacks): with django_capture_on_commit_callbacks(execute=True): # 测试逻辑 # on_commit hooks 将立即执行
登录后复制
总结
选择哪种方法取决于你的项目复杂性和需求。 如果简单性优先,方法一可能足够;如果需要更全面的测试覆盖率,则方法二或三更合适。 方法四提供了在特定场景下更优雅的解决方案。 请根据实际情况选择最合适的方法。
参考:
- pytest-django 文档
- Stack Overflow: 测试非托管模型
希望这些信息对您有所帮助! 如果您还有其他方法或技巧,欢迎分享!
以上就是在 Pytest-Django 中处理非托管模型的详细内容,更多请关注抖狐科技其它相关文章!
-
iphone跟支付宝绑定怎么取消自动续费
要取消 iphone 上支付宝自动续费,请依次进入“支付宝应用”→“我的”→“管理自动续费”→“找到要取消的订阅”→“取消订阅”→“确认取消”。您的支付宝自动续费将被取消,您将不再收到续费通知。如何取...
-
笔记本的显卡是什么牌子的
笔记本显卡主要品牌为 nvidia 和 amd。nvidia 提供 geforce gtx 和 rtx 系列显卡,而 amd 提供 radeon 系列显卡。选择显卡时需考虑用途、预算、笔记本型号和性能...
-
QQ怎么设置添加我的方式 修改设置添加我的方式图文教程
如何设置 qq 的添加方式?想知道的网友们快来看看吧!php小编子墨为大家带来 qq 设置添加我的方式图文教程,详细介绍了如何修改 qq 的添加好友方式,帮助大家解决添加好友时的不便。 如何设置QQ好...
-
如何修复u盘误删文件
误删 u 盘文件时,请立即停止使用 u 盘以防数据覆盖。使用文件恢复软件(如 recuva)进行“深度扫描”搜索已删除的文件。预览并选择要恢复的文件,选择与 u 盘不同的恢复路径,以避免数据覆盖。若无...
-
笔记本死机的主要原因
笔记本死机最常见的原因有:内存不足、过热、软件问题、硬件故障、电源问题和系统文件损坏。可以通过检查内存使用情况、清理灰尘、更新软件、诊断硬件故障、确保电源充足和修复系统文件来解决这些问题。笔记本死机的...