spring-boot - @DataJpaTest 如何指定要扫描的存储库和实体

Question

默认 @DataJpaTest扫描所有 jpa 存储库和 @Entity .就我而言，我有 5 个存储库包和 5 个实体包。例如
com.acme.product.entity与....关联com.acme.product.repositorycom.acme.users.entity与....关联com.acme.users.repositorycom.acme.client.entity与....关联com.acme.client.repository
等等....

我想在一个单独的类中测试每个部分。例如。

@RunWith(SpringRunner.class)
@DataJpaTest
//Some configurations to import only product repositories and product entities
public class TestProductRepository {
  @Autowired
  TestEntityManager entityManager;
}

请注意，我配置了 5 个不同的 EntityManager我想导入它们并使用例如 productEntityManager在 TestProductRepository而不是默认 TestEntityManager加载所有存储库/实体。

非常感谢

Answer 1

这就是我设法实现我想要的:

@ActiveProfiles( "dev" )
@RunWith( SpringRunner.class )
@DataJpaTest
// Exclude the default test database + the default EntityManager in purpose to use my configurations instead.
@AutoConfigureTestDatabase( connection = H2, replace = AutoConfigureTestDatabase.Replace.AUTO_CONFIGURED )
@Import( {
    ProductDataBaseConfig.class,//Import ProductEntityManager and other beans related to DB operations like TransactionManager, etc...
    ProductRepositoryContainer.class //Custom bean containing all product repositories
} )
public class TestProductRepository {
  @Autowired
  private TestEntityManager entityManager; 

}

这里重要的是 @AutoConfigureTestDatabase(...)和 @Import(...) ，因为我替换了自动配置的 bean 并导入了我自己的 ProductEntityManager TestEntityManager使用提供的配置。
这也缩小了 @DataJpaTest的范围它不会扫描类路径中的所有实体和存储库，这就是我想要的。