@Test
public void testRemoveArticle() throws Exception {
MessageResponseServiceImpl mrs = new MessageResponseServiceImpl();
// Successful case
mrs.setMessageResponse(new MessageResponse());
articleService.setMessageResponseService(mrs);
Article articleSuccess = new Article();
articleSuccess.setTitle("test success");
articleSuccess.setDeleted(true);
Mockito.when(articleDao.removeArticle(articleSuccess)).thenReturn(articleSuccess);
MessageResponse messageSuccess = articleService.removeArticle(articleSuccess);
// Fail case
mrs.setMessageResponse(new MessageResponse());
articleService.setMessageResponseService(mrs);
Article articleFail = new Article();
articleFail.setTitle("test fail");
articleFail.setDeleted(false);
Mockito.when(articleDao.removeArticle(articleFail)).thenReturn(articleFail);
MessageResponse messageFail = articleService.removeArticle(articleFail);
Assert.assertTrue(messageSuccess.getCode() == 1 && messageFail.getCode() == 0);
}
@Test
@PrepareForTest({UtilsHibernate.class})
public void testGetArticles() throws Exception {
PowerMockito.mockStatic(UtilsHibernate.class);
List<Article> articleList = new ArrayList<>();
Article article1 = new Article();
article1.setTitle("article1");
articleList.add(article1);
Article article2 = new Article();
article2.setTitle("article2");
articleList.add(article2);
Mockito.when(
articleDao.getAllArticles(
2, 1, UtilsHibernate.getOrderByString("asc", "title"), "title", 1))
.thenReturn(articleList);
List<Article> articles = articleService.getArticles(2, 1, "asc", "title", 1);
Assert.assertTrue(
articles.get(0).getTitle().equals("article1")
&& articles.get(1).getTitle().equals("article2"));
}
@Test
public void testUpdateArticle() throws Exception {
Article article = new Article();
article.setTitle("test");
Mockito.when(articleDao.updateArticle(article)).thenReturn(article);
Article articleResult = articleService.updateArticle(article);
Assert.assertEquals(articleResult, article);
}
@Test
public void testGetArticle() throws Exception {
Article article = new Article();
article.setArticleId(1);
article.setTitle("test title");
Mockito.when(articleDao.getArticle(article)).thenReturn(article);
Article responseArticle = articleService.getArticle(article);
Assert.assertEquals(article.getArticleId(), responseArticle.getArticleId());
}
@Test
public void testGetArticlesByUser() throws Exception {
Article article = new Article();
article.setArticleId(1);
article.setTitle("test title");
List<Article> articles = new ArrayList<>();
articles.add(article);
Mockito.when(articleDao.getArticlesByUserOwner(user)).thenReturn(articles);
List<Article> responseArticles = articleService.getArticlesByUser(user);
Assert.assertEquals(responseArticles.get(0).getArticleId(), 1);
}
/**
* Check creating new article with get user auth context
*
* @throws Exception
*/
@Test
@PrepareForTest({SecurityContextHolder.class})
public void testCreateArticle() throws Exception {
Article article = new Article();
PowerMockito.mockStatic(SecurityContextHolder.class);
PowerMockito.when(SecurityContextHolder.getContext()).thenReturn(securityContext);
Mockito.when(securityContext.getAuthentication()).thenReturn(authentication);
Mockito.when(authentication.getPrincipal()).thenReturn(userDetail);
Mockito.when(userDetail.getUsername()).thenReturn("userName");
Mockito.when(userService.getAuthorizedUser()).thenReturn(user);
articleService.createArticle(article);
ArgumentCaptor<Article> argument = ArgumentCaptor.forClass(Article.class);
Mockito.verify(articleDao, Mockito.times(1)).createArticle(argument.capture());
Assert.assertEquals(argument.getValue().getUserOwner(), user);
}