多租户架构下Spring Boot单元测试回滚异常问题求助

1. 确认事务管理器与EntityManagerFactory的绑定关系 #

你配置了两个JpaTransactionManager和两个LocalContainerEntityManagerFactoryBean，首先要确保标记为@Primary的事务管理器，确实关联了带有MultiTenantConnectionProvider的EntityManagerFactory：

• 在测试类的@Transactional注解里显式指定事务管理器的bean名称，比如@Transactional(transactionManager = "multiTenantTxManager")，避免默认逻辑选错管理器。
• 检查配置类中，事务管理器的setEntityManagerFactory方法是否正确指向多租户的EMF实例：

  @Primary
  @Bean
  public JpaTransactionManager multiTenantTxManager(EntityManagerFactory multiTenantEmf) {
      JpaTransactionManager txManager = new JpaTransactionManager();
      txManager.setEntityManagerFactory(multiTenantEmf);
      return txManager;
  }
  

2. 自定义测试执行监听器适配多租户上下文 #

默认的TransactionalTestExecutionListener可能没处理多租户的上下文清理，导致测试间租户状态残留，或者回滚时未关联正确的租户连接。可以继承它扩展逻辑：

public class MultiTenantTransactionalTestExecutionListener extends TransactionalTestExecutionListener {
    @Override
    protected void beforeTestMethod(TestContext testContext) throws Exception {
        // 测试前清理旧上下文并设置默认租户ID
        TenantContext.clear();
        TenantContext.setCurrentTenantId("test-tenant");
        super.beforeTestMethod(testContext);
    }

    @Override
    protected void afterTestMethod(TestContext testContext) throws Exception {
        super.afterTestMethod(testContext);
        // 测试后强制清除租户上下文，避免影响下一个测试
        TenantContext.clear();
    }
}

然后在测试类上替换默认监听器：

@SpringBootTest
@AutoConfigureTestDatabase
@Sql
@Transactional
@TestExecutionListeners(
    listeners = MultiTenantTransactionalTestExecutionListener.class,
    mergeMode = TestExecutionListeners.MergeMode.MERGE_WITH_DEFAULTS
)
public class YourTestClass {
    // 测试方法
}

3. 调整@Sql注解的事务执行模式 #

默认情况下@Sql可能在事务启动前执行SQL，导致这部分数据不在测试事务的回滚范围内，从而出现唯一键冲突。可以强制让SQL在当前事务内执行：

@Sql(
    scripts = "/init-test-data.sql",
    config = @SqlConfig(transactionMode = SqlConfig.TransactionMode.INFERRED)
)

INFERRED模式会让SQL脚本使用测试方法的事务上下文，这样脚本执行的数据会被测试后的回滚操作覆盖。

4. 排查多租户连接提供者的事务逻辑 #

检查你的MultiTenantConnectionProvider实现，确保：

• 获取连接时正确切换到当前租户的Schema，且该连接被事务管理器正确管理。
• 连接关闭或事务结束时，正确重置Schema或清理租户绑定，避免后续测试复用带有旧租户状态的连接。
  比如，在getConnection(String tenantId)方法中，要确保执行了SET SCHEMA '${tenantId}'这类切换语句，且事务回滚时针对该Schema生效。

内容的提问来源于stack exchange，提问作者smotron