How to use

ImageRepository

in

org.openbaton.nfvo.repositories

 @Override
 @Transactional
 public void deleteImage(String idVim, String idImage) {
  vimRepository.findFirstById(idVim).getImages().remove(imageRepository.findOne(idImage));
  imageRepository.delete(idImage);
 }
}

@Override
public NFVImage update(NFVImage nfvImage, String id) {
 nfvImage = imageRepository.save(nfvImage);
 nfvImage.setUpdated(new Date());
 return nfvImage;
}

@Override
public Iterable<BaseNfvImage> query() {
 return imageRepository.findAll();
}

@Test
public void nfvImageManagementQueryTest() {
 when(imageRepository.findAll()).thenReturn(new ArrayList<>());
 Assert.assertEquals(false, nfvImageManagement.query().iterator().hasNext());
 NFVImage nfvImage_exp = createNfvImage();
 when(imageRepository.findOne(nfvImage_exp.getId())).thenReturn(nfvImage_exp);
 BaseNfvImage nfvImage_new = nfvImageManagement.query(nfvImage_exp.getId());
 Assert.assertEquals(nfvImage_exp.getId(), nfvImage_new.getId());
 //    Assert.assertEquals(nfvImage_exp.getName(), nfvImage_new.getName());
 Assert.assertEquals(nfvImage_exp.getExtId(), nfvImage_new.getExtId());
 //    Assert.assertEquals(nfvImage_exp.getMinRam(), nfvImage_new.getMinRam());
}

@Override
public BaseNfvImage query(String id) {
 return imageRepository.findOne(id);
}

@Override
public void delete(String id) {
 log.debug("Removing image with id " + id);
 imageRepository.delete(id);
}

@Test
public void nfvImageManagementDeleteTest() {
 BaseNfvImage nfvImage_exp = createNfvImage();
 when(imageRepository.findOne(nfvImage_exp.getId())).thenReturn(nfvImage_exp);
 nfvImageManagement.delete(nfvImage_exp.getId());
 when(imageRepository.findOne(nfvImage_exp.getId())).thenReturn(null);
 BaseNfvImage nfvImage_new = nfvImageManagement.query(nfvImage_exp.getId());
 Assert.assertNull(nfvImage_new);
}

private void initMocks() throws VimException, PluginException {
 Vim vim = mock(Vim.class);
 when(vim.queryImages(any(BaseVimInstance.class))).thenReturn(new ArrayList<>());
 when(vimBroker.getVim(anyString())).thenReturn(vim);
 //    try {
 //      Future<Set<DeploymentFlavour>> futureFlavours = mock(AsyncResult.class);
 //      Future<Set<BaseNetwork>> futureNetworks = mock(AsyncResult.class);
 //      Future<Set<BaseNfvImage>> futureImages = mock(AsyncResult.class);
 //
 // when(asyncVimManagement.updateFlavors(any(BaseVimInstance.class))).thenReturn(futureFlavours);
 //      when(asyncVimManagement.updateFlavors(any(BaseVimInstance.class)).get())
 //          .thenReturn(new HashSet<>());
 //
 // when(asyncVimManagement.updateImages(any(BaseVimInstance.class))).thenReturn(futureImages);
 //      when(asyncVimManagement.updateImages(any(BaseVimInstance.class)).get())
 //          .thenReturn(new HashSet<>());
 //      when(asyncVimManagement.updateNetworks(any(BaseVimInstance.class)))
 //          .thenReturn(futureNetworks);
 //      when(asyncVimManagement.updateNetworks(any(BaseVimInstance.class)).get())
 //          .thenReturn(new HashSet<>());
 //    } catch (InterruptedException | ExecutionException | BadRequestException e) {
 //      e.printStackTrace();
 //    }
 doNothing().when(imageRepository).delete(any(NFVImage.class));
 doNothing().when(networkRepository).delete(any(Network.class));
}

@Override
public NFVImage add(NFVImage NFVImage) {
 log.trace("Adding image " + NFVImage);
 log.debug("Adding image with name " + NFVImage.getName());
 return imageRepository.save(NFVImage);
}

public BaseNfvImage queryImage(String idVim, String idImage, String projectId)
  throws EntityUnreachableException, NotFoundException {
 BaseVimInstance vimInstance = vimRepository.findFirstByIdAndProjectId(idVim, projectId);
 if (vimInstance == null)
  throw new NotFoundException("VIM Instance with ID " + idVim + " not found.");
 if (!vimInstance.isActive()) {
  throw new EntityUnreachableException(
    "VimInstance " + vimInstance.getName() + " is not reachable");
 }
 try {
  refresh(vimInstance, true);
 } catch (Exception e) {
  log.error(
    "Unable to refresh the VIM instance with ID "
      + idVim
      + " before querying the image with ID "
      + idImage);
  e.printStackTrace();
 }
 for (BaseNfvImage image : vimInstance.getImages()) {
  if (image.getId().equals(idImage)) return imageRepository.findOne(idImage);
 }
 throw new NotFoundException(
   "Did not find image with ID "
     + idImage
     + " for the VIM instance "
     + vimInstance.getName()
     + " with ID "
     + idVim);
}

@Override
@Transactional
public BaseNfvImage addImage(String id, BaseNfvImage image) {
 image = imageRepository.save(image);
 vimRepository.findFirstById(id).addImage(image);
 return image;
}

@Test
public void nfvImageManagementUpdateTest() {
 NFVImage nfvImage_exp = createNfvImage();
 NFVImage nfvImage_new = createNfvImage();
 nfvImage_new.setName("UpdatedName");
 nfvImage_new.setMinRam(2046);
 when(imageRepository.save(any(NFVImage.class))).thenReturn(nfvImage_new);
 nfvImage_exp = nfvImageManagement.update(nfvImage_new, nfvImage_exp.getId());
 Assert.assertEquals(nfvImage_exp.getName(), nfvImage_new.getName());
 Assert.assertEquals(nfvImage_exp.getExtId(), nfvImage_new.getExtId());
 Assert.assertEquals(nfvImage_exp.getMinRam(), nfvImage_new.getMinRam());
}

@Test
public void nfvImageManagementAddTest() {
 NFVImage nfvImage_exp = createNfvImage();
 when(imageRepository.save(any(NFVImage.class))).thenReturn(nfvImage_exp);
 NFVImage nfvImage_new = nfvImageManagement.add(nfvImage_exp);
 Assert.assertEquals(nfvImage_exp.getId(), nfvImage_new.getId());
 Assert.assertEquals(nfvImage_exp.getName(), nfvImage_new.getName());
 Assert.assertEquals(nfvImage_exp.getExtId(), nfvImage_new.getExtId());
 Assert.assertEquals(nfvImage_exp.getMinRam(), nfvImage_new.getMinRam());
}