openecomp-be/api/openecomp-sdc-rest-webapp/vendor-software-products-rest/vendor-software-products-rest-services/src/test/java/org/openecomp/sdcrests/vsp/rest/services/ImagesImplTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.openecomp.sdcrests.vsp.rest.services;
  22
  23 import org.apache.http.HttpStatus;
  24 import org.junit.Assert;
  25 import org.junit.Before;
  26 import org.junit.Test;
  27 import org.junit.runner.RunWith;
  28 import org.mockito.ArgumentMatchers;
  29 import org.mockito.Mock;
  30 import org.openecomp.sdc.logging.api.Logger;
  31 import org.openecomp.sdc.logging.api.LoggerFactory;
  32 import org.openecomp.sdc.vendorsoftwareproduct.ComponentManager;
  33 import org.openecomp.sdc.vendorsoftwareproduct.ComponentManagerFactory;
  34 import org.openecomp.sdc.vendorsoftwareproduct.ImageManager;
  35 import org.openecomp.sdc.vendorsoftwareproduct.ImageManagerFactory;
  36 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ImageEntity;
  37 import org.openecomp.sdc.vendorsoftwareproduct.types.CompositionEntityResponse;
  38 import org.openecomp.sdc.vendorsoftwareproduct.types.QuestionnaireResponse;
  39 import org.openecomp.sdc.vendorsoftwareproduct.types.composition.CompositionEntityType;
  40 import org.openecomp.sdc.vendorsoftwareproduct.types.composition.CompositionEntityValidationData;
  41 import org.openecomp.sdc.vendorsoftwareproduct.types.composition.Image;
  42 import org.openecomp.sdcrests.vendorsoftwareproducts.types.ImageCreationDto;
  43 import org.openecomp.sdcrests.vendorsoftwareproducts.types.ImageDto;
  44 import org.openecomp.sdcrests.vendorsoftwareproducts.types.ImageRequestDto;
  45 import org.openecomp.sdcrests.vendorsoftwareproducts.types.QuestionnaireResponseDto;
  46 import org.openecomp.sdcrests.wrappers.GenericCollectionWrapper;
  47 import org.powermock.core.classloader.annotations.PrepareForTest;
  48 import org.powermock.modules.junit4.PowerMockRunner;
  49
  50 import javax.ws.rs.core.Response;
  51 import java.util.Collection;
  52 import java.util.Collections;
  53 import java.util.UUID;
  54
  55 import static org.mockito.MockitoAnnotations.initMocks;
  56 import static org.powermock.api.mockito.PowerMockito.mockStatic;
  57 import static org.powermock.api.mockito.PowerMockito.when;
  58
  59 @RunWith(PowerMockRunner.class)
  60 @PrepareForTest({ComputeImpl.class, ComponentManagerFactory.class, ImageManagerFactory.class})
  61 public class ImagesImplTest {
  62
  63   private Logger logger = LoggerFactory.getLogger(ImagesImplTest.class);
  64
  65   @Mock
  66   private ImageManagerFactory imageManagerFactory;
  67
  68   @Mock
  69   private ImageManager imageManager;
  70
  71   @Mock
  72   private ComponentManagerFactory componentManagerFactory;
  73
  74   @Mock
  75   private ComponentManager componentManager;
  76
  77   private final String vspId = UUID.randomUUID().toString();
  78   private final String versionId = UUID.randomUUID().toString();
  79   private final String componentId = "" + System.currentTimeMillis();
  80   private final String imageId = "" + System.currentTimeMillis();
  81   private final String user = "cs0008";
  82
  83   @Before
  84   public void setUp() {
  85     try {
  86       initMocks(this);
  87
  88       mockStatic(ComponentManagerFactory.class);
  89       when(ComponentManagerFactory.getInstance()).thenReturn(componentManagerFactory);
  90       when(componentManagerFactory.createInterface()).thenReturn(componentManager);
  91
  92       mockStatic(ImageManagerFactory.class);
  93       when(ImageManagerFactory.getInstance()).thenReturn(imageManagerFactory);
  94       when(imageManagerFactory.createInterface()).thenReturn(imageManager);
  95
  96
  97
  98       ImageEntity ie = new ImageEntity();
  99       ie.setComponentId(componentId);
 100       ie.setId(imageId);
 101       ie.setCompositionData("{\"name\":\"nm\",\"description\":\"d\"}");
 102
 103
 104       Collection<ImageEntity> cList = Collections.singletonList(ie);
 105       when(imageManager.listImages(
 106               ArgumentMatchers.eq(vspId),
 107               ArgumentMatchers.any(),
 108               ArgumentMatchers.eq(componentId))).thenReturn(cList);
 109
 110       when(imageManager.createImage(
 111               ArgumentMatchers.any())).thenReturn(ie);
 112
 113       CompositionEntityResponse<Image> r = new CompositionEntityResponse<>();
 114       r.setId(vspId);
 115       when(imageManager.getImage(
 116               ArgumentMatchers.eq(vspId),
 117               ArgumentMatchers.any(),
 118               ArgumentMatchers.eq(componentId),
 119               ArgumentMatchers.eq(imageId))).thenReturn(r);
 120
 121       CompositionEntityType tpe = CompositionEntityType.component;
 122       CompositionEntityValidationData data = new CompositionEntityValidationData(tpe, vspId);
 123       when(imageManager.updateImage(
 124               ArgumentMatchers.any())).thenReturn(data);
 125
 126
 127       QuestionnaireResponse qr = new QuestionnaireResponse();
 128       qr.setData("helloworld");
 129       when(imageManager.getImageQuestionnaire(
 130               ArgumentMatchers.eq(vspId),
 131               ArgumentMatchers.any(),
 132               ArgumentMatchers.eq(componentId),
 133               ArgumentMatchers.eq(imageId))).thenReturn(qr);
 134
 135
 136     } catch (Exception e) {
 137       logger.error(e.getMessage(), e);
 138     }
 139   }
 140
 141   @Test
 142   public void testList() {
 143     ImagesImpl ii = new ImagesImpl();
 144
 145     Response rsp = ii.list(vspId, versionId, componentId, user);
 146     Assert.assertEquals("Response should be 200", HttpStatus.SC_OK, rsp.getStatus());
 147     Object e = rsp.getEntity();
 148     Assert.assertNotNull(e);
 149     @SuppressWarnings("unchecked")
 150     GenericCollectionWrapper<ImageDto> results = (GenericCollectionWrapper<ImageDto>)e;
 151     Assert.assertEquals("result length", 1, results.getListCount());
 152   }
 153
 154
 155   @Test
 156   public void testCreate() {
 157
 158     ImageRequestDto dto = new ImageRequestDto();
 159     dto.setDescription("hello");
 160     dto.setFileName("name");
 161
 162     ImagesImpl ii = new ImagesImpl();
 163     Response rsp = ii.create(dto, vspId, versionId, componentId, user);
 164     Assert.assertEquals("Response should be 200", HttpStatus.SC_OK, rsp.getStatus());
 165     Object e = rsp.getEntity();
 166     Assert.assertNotNull(e);
 167     try {
 168       ImageCreationDto creationDto = (ImageCreationDto)e;
 169       Assert.assertEquals(imageId, creationDto.getId());
 170     } catch (ClassCastException ex) {
 171       Assert.fail("unexpected class for DTO " + e.getClass().getName());
 172     }
 173   }
 174
 175
 176   @Test
 177   public void testDelete() {
 178     ImagesImpl ii = new ImagesImpl();
 179     Response rsp = ii.delete(vspId, versionId, componentId, imageId, user);
 180     Assert.assertEquals("Response should be 200", HttpStatus.SC_OK, rsp.getStatus());
 181     Assert.assertNull(rsp.getEntity());
 182   }
 183
 184
 185   @Test
 186   public void testGet() {
 187     ImagesImpl ii = new ImagesImpl();
 188     Response rsp = ii.get(vspId, versionId, componentId, imageId, user);
 189     Assert.assertEquals("Response should be 200", HttpStatus.SC_OK, rsp.getStatus());
 190     Assert.assertNotNull(rsp.getEntity());
 191   }
 192
 193   @Test
 194   public void testUpdate() {
 195     ImagesImpl ii = new ImagesImpl();
 196     ImageRequestDto dto = new ImageRequestDto();
 197     Response rsp = ii.update(dto, vspId, versionId, componentId, imageId, user);
 198     Assert.assertEquals("Response should be 200", HttpStatus.SC_OK, rsp.getStatus());
 199     Assert.assertNull(rsp.getEntity());
 200   }
 201
 202   @Test
 203   public void testGetQuestionaire() {
 204     ImagesImpl ii = new ImagesImpl();
 205     Response rsp = ii.getQuestionnaire(vspId, versionId, componentId, imageId, user);
 206     Assert.assertEquals("Response should be 200", HttpStatus.SC_OK, rsp.getStatus());
 207     try {
 208       QuestionnaireResponseDto dto = (QuestionnaireResponseDto)rsp.getEntity();
 209       Assert.assertEquals("helloworld", dto.getData());
 210     }
 211     catch (Exception ex) {
 212       logger.error("caught exception", ex);
 213       Assert.fail(ex.getMessage());
 214     }
 215   }
 216
 217
 218   @Test
 219   public void testUpdateQuestionaire() {
 220     ImagesImpl ii = new ImagesImpl();
 221     Response rsp = ii.updateQuestionnaire("helloworld", vspId, versionId, componentId, imageId, user);
 222     Assert.assertEquals("Response should be 200", HttpStatus.SC_OK, rsp.getStatus());
 223     Assert.assertNull(rsp.getEntity());
 224   }
 225 }