Snippet

Verifying Custom Query Logic via @DataJpaTest and TestEntityManager

The @DataJpaTest annotation disables full Spring context initialization and focuses strictly on JPA components. TestEntityManager provides an alternative to the standard EntityManager specifically configured for slice testing, ensuring test entities are persisted and flushed to an in-memory database before executing repository query assertions.

snippet.java

java

@DataJpaTest
class UserRepositoryTest {
    @Autowired
    private TestEntityManager entityManager;
 
    @Autowired
    private UserRepository userRepository;
 
    @Test
    void shouldFindActiveUsersByDepartment() {
        User user = new User("Alice", "Engineering", true);
        entityManager.persistAndFlush(user);
 
        List<User> result = userRepository.findByDepartmentAndActiveTrue("Engineering");
 
        assertThat(result).hasSize(1).contains(user);
    }
}

spring

Breakdown

@DataJpaTest

Configures a focused test slice for JPA repositories and in-memory database configuration.

private TestEntityManager entityManager;

Injects a helper for setting up entity states and flushing database changes during tests.

entityManager.persistAndFlush(user);

Synchronizes the transient test entity directly into the underlying persistence context.

Previous snippet Next snippet

More Intermediate Java exercises →

Verifying Custom Query Logic via @DataJpaTest and TestEntityManager

Related