使用@DataJpaTest测试多租户Spring Boot应用Repository层的问题

@DataJpaTest是Spring Boot的切片测试注解，默认仅加载与JPA Repository、EntityManager相关的核心组件，不会自动加载多租户相关的配置（比如绑定CurrentTenantIdentifierResolver到Hibernate的EntityManagerFactory、多租户连接提供者等），这就是你的租户ID未生效的原因。而@SpringBootTest会加载完整应用上下文，所有多租户配置都被初始化，因此可以正常工作。

以下是在@DataJpaTest下实现正确行为的具体方案：

1. 导入多租户配置类 #

找到项目中配置多租户的类（比如配置了CurrentTenantIdentifierResolver、MultiTenantConnectionProvider，以及设置Hibernate多租户策略的配置类），在测试类上通过@Import注解导入该类，让@DataJpaTest的上下文加载这些配置：

@DataJpaTest
@Import(TenantConfiguration.class) // 替换为你的多租户配置类
public class VendorRepositoryTest implements BaseTest {

  @Autowired
  private VendorRepository vendorRepository;

  @Test
  public void testVendorRepository() {
    TenantIdentifierResolver.forceTenantId("123", () -> {
      vendorRepository.save(new Vendor().setName("name1"));
    });

    TenantIdentifierResolver.forceTenantId("456", () -> {
      vendorRepository.save(new Vendor().setName("name2"));
      Page<Vendor> a = vendorRepository.findAll(Pageable.ofSize(10));
      // 断言仅返回当前租户的数据
      Assertions.assertEquals(1, a.getTotalElements());
      Assertions.assertEquals("name2", a.getContent().get(0).getName());
    });
  }
}

2. 手动配置EntityManagerFactory的多租户属性 #

如果没有专门的多租户配置类，可以在测试中创建内部配置类，直接设置Hibernate的多租户相关属性：

@DataJpaTest
public class VendorRepositoryTest implements BaseTest {

  @Autowired
  private VendorRepository vendorRepository;

  @Configuration
  static class TestTenantConfig {
    @Bean
    public CurrentTenantIdentifierResolver tenantIdentifierResolver() {
      return YourTenantIdentifierResolverImpl.getInstance(); // 替换为你的实现类
    }

    @Bean
    public HibernatePropertiesCustomizer hibernatePropertiesCustomizer(CurrentTenantIdentifierResolver resolver) {
      return properties -> {
        properties.put("hibernate.multi_tenant", "DATABASE"); // 替换为你的多租户策略（如SCHEMA/DATABASE）
        properties.put("hibernate.tenant_identifier_resolver", resolver);
        // 若需要，添加多租户连接提供者配置
        // properties.put("hibernate.multi_tenant_connection_provider", yourConnectionProvider());
      };
    }
  }

  // 测试方法...
}

3. 验证租户解析器的线程安全性 #

确保你的CurrentTenantIdentifierResolver实现使用ThreadLocal存储租户ID（符合forceTenantId的线程绑定逻辑），且为单例Bean——@DataJpaTest默认会注册单例Bean，只要实现正确，就能在测试中正常生效。

通过以上步骤，无需加载完整应用上下文，就能让@DataJpaTest的切片上下文正确识别并应用多租户配置，实现Repository层的租户隔离测试。

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