/** Test of findOne method, of class ChartCollectionService. */ @Test public void testFindOne() { System.out.println("findOne"); Long id = 1L; ChartCollection expResult = new ChartCollection(null, "burp", null, null); Mockito.when(chartCollectionRepository.findOne(id)).thenReturn(expResult); ChartCollection result = service.findOne(id); assertEquals(expResult, result); }
/** Test of save method, of class ChartCollectionService. */ @Test public void testSave() { System.out.println("save"); ChartCollection collection = new ChartCollection(); Mockito.when(chartCollectionRepository.save(Mockito.any(ChartCollection.class))) .thenAnswer( new Answer<ChartCollection>() { @Override public ChartCollection answer(InvocationOnMock invocation) throws Throwable { ChartCollection cc = (ChartCollection) invocation.getArguments()[0]; cc.setId(1); return cc; } }); ChartCollection result = service.save(collection); assertTrue(result.getId() > 0); }