Manualy使用回滚执行测试中事务中的代码

Question

我们使用的是spring Boot2，在集成测试中，我们需要在事务中手动执行一些代码，在事务结束时和断言之后，我们希望回滚该事务。

我们使用显式定义的事务而不是@Transactional，因为有时我们需要在Test2事务中执行。

@Test
public void fooTest() {

    // transaction 1
    transactionTemplate.execute(new TransactionCallbackWithoutResult() {
        @Override
            protected void doInTransactionWithoutResult(TransactionStatus status) {
                // some code in transaction
            }

    // transaction 2
    transactionTemplate.execute(new TransactionCallbackWithoutResult() {
        @Override
            protected void doInTransactionWithoutResult(TransactionStatus status) {
                // some code in transaction
            }

    // here I need rollback these transactions for clean db for another tests
}

你能告诉我如何在我们的例子中使用回滚来回滚这两个事务吗？这是我们维护的较老的代码，所以如果有可能在引导2中做得更好，我将非常感谢任何建议。我们只需要在一个测试中执行2个事务。

Answer 1

将对每个TransactionStatus的引用存储在AtomicReference中，并在测试后用事务管理器回滚它们。

@Test
void test() {
  final AtomicReference<TransactionStatus> first = new AtomicReference<>();
  final AtomicReference<TransactionStatus> second = new AtomicReference<>();
  transactionTemplate.execute(new TransactionCallbackWithoutResult() {
    @Override
    protected void doInTransactionWithoutResult(TransactionStatus status) {
      // some code in transaction
      first.set(status);
    }
  });

  // transaction 2
  transactionTemplate.execute(new TransactionCallbackWithoutResult() {
    @Override
    protected void doInTransactionWithoutResult(TransactionStatus status) {
      // some code in transaction
      second.set(status);
    }
  });
  transactionTemplate.getTransactionManager().rollback(first.get());
  transactionTemplate.getTransactionManager().rollback(second.get());
}