django - 将存储过程加载到 Django 单元测试数据库中的最佳方法是什么？

Question

我有一些我的 selenium 测试将依赖的 Postgres 存储过程。在开发中，我用脚本中的一行加载它们:

cat stored_procedures.sql | python manage.py dbshel​​l

这在单元测试时不起作用，因为新的数据库是从头开始创建的。如何在运行单元测试之前将保存在文件中的存储过程加载到测试数据库中？

Answer 1

我认为您几乎没有办法做到这一点。在我看来，最好的解决方案是使用自定义 SQL 添加迁移。将来，您不仅需要在开发阶段进行迁移，还需要在生产阶段进行迁移。因此，如果您将更改存储在少数几个地方的数据库中，部署过程将不清楚。

其他方式 - 只需将 SQL 的执行添加到测试用例的 setUp 方法。

额外迁移

您应该创建一个新的空迁移 ./manage.py makemigrations --empty myApp

将您的 SQL 代码添加到 operations 列表

操作 = [
migrations.RunSQL('原始 SQL 代码')
]