@Test public void testGetSocialRequests() { // Given SocialRequest socialRequest = mock(SocialRequest.class); when(socialRelationService.getUserRequests(any(User.class), anyBoolean())) .thenReturn(Arrays.asList(socialRequest)); // When User user = mock(User.class); List<SocialRequest> requests = notificationService.getSocialRequests(user); // Then assertEquals(1, requests.size()); }
@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()); }
@Test public void testGetSocialRequestsWithRespectiveUser() { // Given HashMap<SocialRequest, User> value = new HashMap<SocialRequest, User>(); SocialRequest socialRequest = mock(SocialRequest.class); User mock = mock(User.class); value.put(socialRequest, mock); when(socialRelationService.getUserRequestsWithUser(any(User.class), anyBoolean())) .thenReturn(value); // When User user = mock(User.class); Map<SocialRequest, User> socialRequestsWithRespectiveUser = notificationService.getSocialRequestsWithRespectiveUser(user); // Then assertEquals(1, socialRequestsWithRespectiveUser.size()); }