是否有更简单的方法来恢复 SqlAlchemy 集成测试之间的状态？

Question

Flask 示例应用程序Flasky和​​Flaskr在每次测试之间创建、删除和重新种子其整个数据库。即使这不会使测试套件运行缓慢，我想知道是否有一种方法可以完成同样的事情，同时又不那么“具有破坏性”。我很惊讶没有一种“更温和”的方式来回滚任何更改。我尝试了一些没有效果的事情。

对于上下文，我的测试使用 self.client.post('/things') 之类的东西通过 Flask test_client 调用端点，并在端点内调用 session.commit() 。

我尝试过制作自己的“提交”函数，该函数实际上只在测试期间刷新，但是如果我发出两个连续的请求，例如 self.client.post('/things') 和 self.client.get('/things') ，新创建的项目不会出现在结果集中，因为新请求具有新的请求上下文和新的数据库会话（和事务），该上下文不知道仅刷新而不是提交的更改。这似乎是这种方法不可避免的问题。

我尝试过将子事务与 db.session.begin(subtransactions=True) 一起使用，但后来遇到了更糟糕的问题。因为我有 autoflush=False，所以在提交外部事务之前，实际上没有任何内容被提交或刷新。同样，任何依赖于同一测试中早期请求修改的数据的请求都将失败。即使使用 autoflush=True，顺序请求也会出现前面的问题。

我尝试过嵌套事务，其结果与子事务相同，但显然它们没有做我希望它们做的事情。我看到嵌套事务向数据库发出 SAVEPOINT 命令。我希望这将允许提交发生，对其他会话可见，然后能够在任意时间回滚到该保存点，但这不是他们所做的。它们在事务中使用，并且与之前的方法具有相同的问题。

更新：显然有一种在连接而不是会话上使用嵌套事务的方法，这可能有效，但需要对应用程序进行一些重组才能使用测试代码创建的连接。我还没试过这个。我最终会解决这个问题，但同时我希望还有另一种方法。有人说，由于“真正的嵌套事务”和保存点之间的区别，这种方法可能不适用于 MySQL ，但Postgres 文档也说使用 SAVEPOINT 而不是尝试嵌套事务。我认为我们可以忽略这个警告。我看不出这两个数据库之间有任何区别，如果它适用于其中一个数据库，那么它可能也适用于另一个数据库。

避免 DB drop_all、create_all 和重新播种数据的另一个选项是手动撤消测试引入的更改。但是，在测试端点时，可以将许多行插入到许多表中，并且可靠地手动撤消此操作既费力又容易出现错误。

在尝试了所有这些事情之后，我开始看到在测试之间放弃并创建的智慧。但是，是否有一些我在上面尝试过的应该有效的方法，但我只是做错了一些事情？或者还有其他人知道我还没有尝试过的方法吗？

更新：我刚刚在 StackOverflow 上找到的另一种方法是截断所有表，而不是删除并创建它们。这显然是大约两倍的速度，但它仍然看起来很严厉，并且不如回滚那么方便（回滚不会删除测试用例之前放置在数据库中的任何示例数据）。

Answer 1

对于单元测试，我认为重新生成整个数据库的标准方法是最有意义的，正如您在我的示例和许多其他示例中看到的那样。但我同意，对于大型应用程序，这可能会在测试运行期间花费大量时间。

借助 SQLAlchemy，您可以编写大量在生产数据库上运行的通用数据库代码，这些数据库可能是 MySQL、Postgres 等，同时它还可以在 sqlite 上运行以进行测试。不可能每个应用程序都使用 100% 通用的 SQLAlchemy，因为 sqlite 与其他应用程序有一些重要的差异，但在许多情况下这效果很好。

因此，只要有可能，我都会为我的测试设置一个 sqlite 数据库。即使对于大型数据库，使用内存中的 sqlite 数据库也应该相当快。另一个非常快速的替代方法是生成一次表，使用所有空表备份 sqlite 文件，然后在每次测试之前恢复文件而不是执行create_all().

我还没有探索过使用空表对数据库进行初始备份，然后在 MySQL 或 Postgres 测试之间使用基于文件的恢复的想法，但理论上这也应该有效，所以我想这是您还没有的解决方案在你的清单中提到。不过，您需要在测试之间停止并重新启动数据库服务。