@Test public void testGetSocialRequestCount() throws Exception { // Given SocialRequest socialRequest = mock(SocialRequest.class); when(socialRelationService.getUserRequests(any(User.class), anyBoolean())) .thenReturn(Arrays.asList(socialRequest)); // When User user = mock(User.class); Future<CountResult> count = notificationService.getSocialRequestCount(user); // Then assertEquals(1, (int) count.get().getCount()); }