事务管理器恢复/回滚上次提交

Question

我尝试通过将测试用例 setUp() 中的测试数据放入 setUpClass()/teardownClass 类方法来加速我的测试，因此它不会为测试用例中的每个测试重新创建相同的仅选择夹具。

@classmethod
def setUpClass(cls):
    plant.StuffFactory() #plant stuff with FactoryBoy
    transaction.commit()

@classmethod
def tearDownClass(cls):
    session.query(models.Stuff).delete() # delete planted stuff
    transaction.commit()

但我不喜欢自己用 session.delete 删除东西，因为我使用了很多模型并且不想跟踪我种植的东西。我想要类似的东西

@classmethod
def tearDownClass(cls):
    session.clear() #delete all
    transaction.commit()

但session.close() 或session.remove() 不会影响已提交的数据。 所以我想办法“取消”setUpClasstransaction.commit()，就像我什么都不种。

我尝试了嵌套事务和保存点，但它们仅在尚未提交数据时才有效。

任何指针？

Answer 1

如果您不想提交任何事情，请不要致电transaction.commit() :)

@classmethod
def setUpClass(cls):
    plant.StuffFactory() #plant stuff with FactoryBoy
    # you may possibly want to flush the session, so everything has proper IDs etc.
    DBSession.flush() 
    # let the transaction continue for the duration of the test

@classmethod
def tearDownClass(cls):
    # let the database make everything as if nothing happened
    transaction.rollback()

这将要求您的任何被测代码都不进行显式事务管理（应用程序代码中的transaction.commit() \ transaction.rollback()），但无论如何这是一个不好的做法：

作为一般规则，应用程序应在处理特定数据的函数外部管理会话的生命周期。这是一种基本的关注点分离，它使特定于数据的操作与它们访问和操作该数据的上下文无关。

会话基础、SQLAlchemy 文档 (https://docs.sqlalchemy.org/en/13/orm/session_basics.html)

Answer 2

在setUpClass() 中，您可以创建一个保存点：

sp = transaction.savepoint()

那么在tearDownClass() 你可以简单地使用：

sp.rollback()

希望这会有所帮助。