Integration test asserting the number of queries generated

NotificationResourceIntTest.java

...
import com.yannbriancon.interceptor.HibernateQueryCountInterceptor;


@RunWith(SpringRunner.class)
@SpringBootTest
@Transactional
public class NotificationResourceIntTest {
    @Autowired
    private HibernateQueryCountInterceptor hibernateQueryCountInterceptor;

    @Test
    public void saveFile_isOk() throws Exception {
        // Initialize the query to 0 and allow the counting
        hibernateQueryCountInterceptor.startCounter();

        // Call the resource that we want to test
        MvcResult result = mvc.perform(get("/rest/notifications"))
                .andExpect(status().isOk())
                .andReturn();

        // Get the query count for this thread and check that it is equal to the number of query you expect, let's say 4.
        // The count is checked and we detect potential n+1 queries.
        Assertions.assertThat(hibernateQueryCountInterceptor.getQueryCount()).isEqualTo(4);
    }
}