@Override public VirtualLinkRecord queryRecord(String id) { return virtualLinkRecordRepository.findOne(id); }
@Override public void delete(String id) { log.debug("Removing image with id " + id); VirtualLinkDescriptor vld = null; VirtualLinkRecord vlr = null; try { vld = virtualLinkDescriptorRepository.findOne(id); } catch (NoResultException ignored) { vlr = virtualLinkRecordRepository.findOne(id); } if (vld == null) virtualLinkRecordRepository.delete(vlr); else virtualLinkDescriptorRepository.delete(vld); }
@Test public void virtualLinkManagementDeleteRecordTest() { VirtualLinkRecord virtualLinkRecord_exp = createVirtualLinkRecord(); when(virtualLinkRecordRepository.findOne(virtualLinkRecord_exp.getId())) .thenReturn(virtualLinkRecord_exp); when(virtualLinkDescriptorRepository.findOne(virtualLinkRecord_exp.getId())) .thenThrow(NoResultException.class); virtualLinkManagement.delete(virtualLinkRecord_exp.getId()); when(virtualLinkRecordRepository.findOne(virtualLinkRecord_exp.getId())).thenReturn(null); VirtualLinkRecord virtualLinkRecord_new = virtualLinkManagement.queryRecord(virtualLinkRecord_exp.getId()); Assert.assertNull(virtualLinkRecord_new); } }
@Test public void virtualLinkManagementUpdateRecordTest() { VirtualLinkRecord virtualLinkRecord_exp = createVirtualLinkRecord(); when(virtualLinkRecordRepository.findOne(virtualLinkRecord_exp.getId())) .thenReturn(virtualLinkRecord_exp); when(virtualLinkRecordRepository.save(virtualLinkRecord_exp)).thenReturn(virtualLinkRecord_exp); VirtualLinkRecord virtualLinkRecord_new = createVirtualLinkRecord(); virtualLinkRecord_new.setRoot_requirement("root_requirement_updated"); when(virtualLinkRecordRepository.save(virtualLinkRecord_new)).thenReturn(virtualLinkRecord_new); virtualLinkRecord_exp = virtualLinkManagement.update(virtualLinkRecord_new, virtualLinkRecord_exp.getId()); assertEquals(virtualLinkRecord_exp, virtualLinkRecord_new); }
@Test public void virtualLinkManagementQueryTest() { when(virtualLinkDescriptorRepository.findAll()) .thenReturn(new ArrayList<VirtualLinkDescriptor>()); when(virtualLinkRecordRepository.findAll()).thenReturn(new ArrayList<VirtualLinkRecord>()); Assert.assertEquals(false, virtualLinkManagement.queryDescriptors().iterator().hasNext()); Assert.assertEquals(false, virtualLinkManagement.queryRecords().iterator().hasNext()); VirtualLinkDescriptor virtualLinkDescriptor_exp = createVirtualLinkDescriptor(); when(virtualLinkDescriptorRepository.findOne(virtualLinkDescriptor_exp.getId())) .thenReturn(virtualLinkDescriptor_exp); VirtualLinkDescriptor virtualLinkDescriptor_new = virtualLinkManagement.queryDescriptor(virtualLinkDescriptor_exp.getId()); assertEquals(virtualLinkDescriptor_exp, virtualLinkDescriptor_new); VirtualLinkRecord virtualLinkRecord_exp = createVirtualLinkRecord(); when(virtualLinkRecordRepository.findOne(virtualLinkRecord_exp.getId())) .thenReturn(virtualLinkRecord_exp); VirtualLinkRecord virtualLinkRecord_new = virtualLinkManagement.queryRecord(virtualLinkRecord_exp.getId()); assertEquals(virtualLinkRecord_exp, virtualLinkRecord_new); }