@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); }
@Before public void setUp() { Ticket ticket = null; MessageResponseServiceImpl messageResponseService = new MessageResponseServiceImpl(); messageResponseService.setMessageResponse(new MessageResponse()); ticketService.setMessageResponseService(messageResponseService); for (int i = 1; i <= 5; i++) { ticket = new Ticket(); ticket.setTicketId(i); ticket.setTitle( "Check long TITLE Check long TITLE Check long TITLE Check long TITLE Check long TITLE Ticket#" + i); ticket.setIsDeleted(false); ticket.setDescription("Ticket description #" + i); ticketList.add(ticket); } }