@Override public NFVImage add(NFVImage NFVImage) { log.trace("Adding image " + NFVImage); log.debug("Adding image with name " + NFVImage.getName()); return imageRepository.save(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 update(NFVImage nfvImage, String id) { nfvImage = imageRepository.save(nfvImage); nfvImage.setUpdated(new Date()); return 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()); }