@DataJpaTest how to specify repositories and entities to scan

Question

By default @DataJpaTest scans all jpa repositories and @Entity. In my case I have 5 repositories package and 5 entities package. E.g

com.acme.product.entity is associated with com.acme.product.repository com.acme.users.entity is associated with com.acme.users.repository com.acme.client.entity is associated with com.acme.client.repository

And so on....

I would like to test each part in a separate classe. E.g.

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

Note that I have configured 5 different EntityManager I would like to import them and use for example the productEntityManager in the TestProductRepository instead of the default TestEntityManager which loads all repositories/entites.

Thanks a lot

@SimonMartinelli To override the default one: `TestEntityManager`. I just need a way to tell the `TestEntityManager` to use those repositories and entities. — akuma8, Sep 10 '18 at 13:57
But if you want test the ProductRepository you can inject it and don't need the entitymanager because this is configured — Simon Martinelli, Sep 10 '18 at 14:01
@SimonMartinelli You’re right but the problem is it slows the test because it scans all repositories and all entities. — akuma8, Sep 10 '18 at 14:04
I finally found a way to override the default `TestEntityManager` in the same wave the test time of each `@DataJpaTest` case was divided by 4. — akuma8, Sep 10 '18 at 21:29
Can you please provide this as the answer, that other users can view how you solved it? Thank you — Simon Martinelli, Sep 11 '18 at 07:36
@Niccolò sorry, I was so busy that I forgot to post my answer. I'll do it tonight if I have time or this weekend. I'll let you know. — akuma8, Dec 11 '18 at 13:50

score 1 · Accepted Answer · answered Dec 26 '18 at 13:51

This is how I manage to achieve what I wanted:

@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; 

}

The important thing here is the @AutoConfigureTestDatabase(...) and @Import(...), as I replaced the autoconfigured beans and import my own ProductEntityManager the TestEntityManager uses that provided configuration. This also reduce the scope of @DataJpaTest which does not scan all entities and repositories in the classpath and that was what I wanted.

@DataJpaTest how to specify repositories and entities to scan

1 Answers1

Linked