处理 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) yielddef 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_dbdef mytest(django_capture_on_commit_callbacks): with django_capture_on_commit_callbacks(execute=True): # 测试逻辑 # on_commit hooks 将立即执行
总结
选择哪种方法取决于你的项目复杂性和需求。 如果简单性优先，方法一可能足够；如果需要更全面的测试覆盖率，则方法二或三更合适。 方法四提供了在特定场景下更优雅的解决方案。 请根据实际情况选择最合适的方法。
参考:
pytest-django 文档Stack Overflow: 测试非托管模型
希望这些信息对您有所帮助! 如果您还有其他方法或技巧，欢迎分享！
以上就是在 Pytest-Django 中处理非托管模型的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1355115.html
微信扫一扫 
支付宝扫一扫 
