gerrit.onap Code Review - sdc.git/blob

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 2017 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.sdc.vendorsoftwareproduct.tree;
  22
  23
  24 import static org.junit.Assert.assertEquals;
  25 import static org.mockito.ArgumentMatchers.any;
  26 import static org.mockito.Mockito.doReturn;
  27
  28 import java.io.ByteArrayInputStream;
  29 import java.io.ByteArrayOutputStream;
  30 import java.io.File;
  31 import java.io.IOException;
  32 import java.io.InputStream;
  33 import java.net.URL;
  34 import java.nio.ByteBuffer;
  35 import java.util.zip.ZipOutputStream;
  36 import org.apache.commons.io.IOUtils;
  37 import org.junit.Before;
  38 import org.junit.Test;
  39 import org.mockito.InjectMocks;
  40 import org.mockito.Mock;
  41 import org.mockito.MockitoAnnotations;
  42 import org.mockito.Spy;
  43 import org.openecomp.core.model.dao.ServiceModelDao;
  44 import org.openecomp.core.model.types.ServiceElement;
  45 import org.openecomp.core.utilities.orchestration.OnboardingTypesEnum;
  46 import org.openecomp.sdc.healing.api.HealingManager;
  47 import org.openecomp.sdc.logging.api.Logger;
  48 import org.openecomp.sdc.logging.api.LoggerFactory;
  49 import org.openecomp.sdc.tosca.datatypes.ToscaServiceModel;
  50 import org.openecomp.sdc.vendorsoftwareproduct.CompositionEntityDataManager;
  51 import org.openecomp.sdc.vendorsoftwareproduct.dao.OrchestrationTemplateDao;
  52 import org.openecomp.sdc.vendorsoftwareproduct.dao.VendorSoftwareProductInfoDao;
  53 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.OrchestrationTemplateEntity;
  54 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.VspDetails;
  55 import org.openecomp.sdc.vendorsoftwareproduct.impl.OrchestrationTemplateCandidateManagerImpl;
  56 import org.openecomp.sdc.vendorsoftwareproduct.services.composition.CompositionDataExtractor;
  57 import org.openecomp.sdc.vendorsoftwareproduct.services.impl.filedatastructuremodule.CandidateServiceImpl;
  58 import org.openecomp.sdc.vendorsoftwareproduct.types.OnboardPackageInfo;
  59 import org.openecomp.sdc.vendorsoftwareproduct.types.UploadFileResponse;
  60 import org.openecomp.sdc.vendorsoftwareproduct.utils.VSPCommon;
  61 import org.openecomp.sdc.versioning.dao.types.Version;
  62
  63 public class UploadFileTest {
  64   private static final Logger LOGGER = LoggerFactory.getLogger(UploadFileTest.class);
  65
  66   private static final String USER1 = "vspTestUser1";
  67
  68   public static final Version VERSION01 = new Version(0, 1);
  69
  70   @Mock
  71   private OrchestrationTemplateDao orchestrationTemplateDataDaoMock;
  72   @Spy
  73   private CandidateServiceImpl candidateService;
  74   @Mock
  75   private HealingManager healingManagerMock;
  76   @Mock
  77   private CompositionDataExtractor compositionDataExtractorMock;
  78   @Mock
  79   private ServiceModelDao<ToscaServiceModel, ServiceElement> serviceModelDaoMock;
  80   @Mock
  81   private CompositionEntityDataManager compositionEntityDataManagerMock;
  82   @Mock
  83   private VendorSoftwareProductInfoDao vspInfoDaoMock;
  84
  85   private OnboardPackageInfo onboardPackageInfo;
  86
  87   @InjectMocks
  88   private OrchestrationTemplateCandidateManagerImpl candidateManager;
  89
  90   public static String id001 = "dummyId";
  91   public static Version activeVersion002 = new Version(1, 0);
  92
  93   private final VspDetails vspDetails = new VspDetails(id001, activeVersion002);
  94
  95   @Before
  96   public void setUp() throws Exception {
  97     MockitoAnnotations.initMocks(this);
  98   }
  99
 100   @Test
 101   public void testUploadFile() throws IOException {
 102     doReturn(vspDetails).when(vspInfoDaoMock).get(any(VspDetails.class));
 103     try (final InputStream inputStream = getZipInputStream("/legalUpload")) {
 104       onboardPackageInfo = new OnboardPackageInfo("legalUpload", OnboardingTypesEnum.ZIP.toString(),
 105               convertFileInputStream(inputStream));
 106       candidateManager.upload(vspDetails, onboardPackageInfo);
 107
 108     }
 109   }
 110
 111   private void testLegalUpload(String vspId, Version version, InputStream upload, String user) {
 112     onboardPackageInfo = new OnboardPackageInfo("file", OnboardingTypesEnum.ZIP.toString(),
 113             convertFileInputStream(upload));
 114     final UploadFileResponse uploadFileResponse = candidateManager.upload(vspDetails, onboardPackageInfo);
 115     assertEquals(uploadFileResponse.getOnboardingType(), OnboardingTypesEnum.ZIP);
 116     OrchestrationTemplateEntity uploadData = orchestrationTemplateDataDaoMock.get(vspId, version);
 117
 118   }
 119
 120   public InputStream getZipInputStream(String name) {
 121     URL url = this.getClass().getResource(name);
 122     File templateDir = new File(url.getFile());
 123
 124     ByteArrayOutputStream baos = new ByteArrayOutputStream();
 125     try (ZipOutputStream zos = new ZipOutputStream(baos)) {
 126       VSPCommon.zipDir(templateDir, "", zos, true);
 127     } catch (IOException e) {
 128       e.printStackTrace();
 129     }
 130     return new ByteArrayInputStream(baos.toByteArray());
 131   }
 132
 133   private ByteBuffer convertFileInputStream(final InputStream fileInputStream) {
 134     byte[] fileContent = new byte[0];
 135     try {
 136       fileContent = IOUtils.toByteArray(fileInputStream);
 137     } catch (final IOException e) {
 138       LOGGER.error(String.format("Could not convert %s into byte[]", fileInputStream), e);
 139     }
 140     return ByteBuffer.wrap(fileContent);
 141   }
 142
 143 }