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

Question

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

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

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

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

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

更新:Apparently there is a way在 Connection 而不是 Session 上使用嵌套事务，这可能有效，但需要对应用程序进行一些重组才能使用由测试代码创建的 Connection。我还没有尝试过这个。我最终会解决它，但与此同时我希望有另一种方法。有人说这种做法may not work with MySQL由于“real nested transactions ”和保存点之间的区别，但是 Postgres documentation还说要使用 SAVEPOINT 而不是试图嵌套事务。我想我们可以忽略这个警告。我看不出这两个数据库之间有任何区别，如果它适用于一个数据库，那么它也可能适用于另一个数据库。

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

在尝试了所有这些之后，我开始看到在测试之间删除和创建的智慧。但是，是否有我在上面尝试过的应该起作用的东西，但我只是做错了什么？还是有人知道我还没有尝试过另一种方法？

更新:Another method I just found on StackOverflow是截断所有表而不是删除和创建它们。这显然快了一倍，但它仍然显得笨拙，并且不如回滚方便(回滚不会删除测试用例之前放置在数据库中的任何样本数据)。

Answer 1

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

感谢 SQLAlchemy，您可以编写大量在生产数据库(可能是 MySQL、Postgres 等)上运行的通用数据库代码，同时它在 sqlite 上运行以进行测试。不可能每个应用程序都使用 100% 通用的 SQLAlchemy，因为 sqlite 与其他应用程序有一些重要的区别，但在许多情况下，它工作得很好。

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

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