gerrit.onap Code Review - sdc.git/blob

   1 package org.openecomp.sdc.vendorsoftwareproduct.impl;
   2
   3 import org.mockito.InjectMocks;
   4 import org.mockito.Mock;
   5 import org.mockito.MockitoAnnotations;
   6 import org.openecomp.core.enrichment.types.MonitoringUploadType;
   7 import org.openecomp.sdc.common.errors.CoreException;
   8 import org.openecomp.sdc.vendorsoftwareproduct.dao.ComponentArtifactDao;
   9 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ComponentMonitoringUploadEntity;
  10 import org.openecomp.sdc.vendorsoftwareproduct.types.schemagenerator.MonitoringUploadStatus;
  11 import org.openecomp.sdc.versioning.dao.types.Version;
  12 import org.testng.Assert;
  13 import org.testng.annotations.BeforeMethod;
  14 import org.testng.annotations.Test;
  15
  16 import java.io.IOException;
  17 import java.io.InputStream;
  18 import java.net.URL;
  19 import java.util.Arrays;
  20 import java.util.Optional;
  21 import java.util.function.Consumer;
  22
  23 import static org.mockito.Matchers.any;
  24 import static org.mockito.Matchers.anyObject;
  25 import static org.mockito.Mockito.doReturn;
  26 import static org.mockito.Mockito.never;
  27 import static org.mockito.Mockito.verify;
  28
  29 public class MonitoringUploadsManagerImplTest {
  30
  31   private static final String USER1 = "ComponentsUploadTestUser";
  32   private static final String COMPONENT_ID = "COMPONENT_ID";
  33   private static final String VSP_ID = "vspId";
  34   private static final Version VERSION = new Version(0, 1);
  35   private static final String TRAP_FILE_NAME = "MMSC.zip";
  36   private static final String POLL_FILE_NAME = "MNS OAM FW.zip";
  37   private static final String VES_FILE_NAME = "vesTest-yml_only.zip";
  38   private static final String INVALID_VES_FILE_NAME = "invalid_ves_file.zip";
  39   private static final String NOT_ZIP_FILE_NAME = "notZipFile";
  40   private static final String ZIP_WITH_FOLDERS_FILE_NAME = "zipFileWithFolder.zip";
  41   private static final String EMPTY_ZIP_FILE_NAME = "emptyZip.zip";
  42   private static final String ZIP_DIR = "/vspmanager/zips/";
  43
  44   @Mock
  45   private ComponentArtifactDao componentArtifactDaoMock;
  46   @InjectMocks
  47   private MonitoringUploadsManagerImpl monitoringUploadsManager;
  48
  49   @BeforeMethod
  50   public void setUp() throws Exception {
  51     MockitoAnnotations.initMocks(this);
  52   }
  53
  54   @Test(expectedExceptions = CoreException.class)
  55   public void testUploadEmptyZip() {
  56     processFile(ZIP_DIR + EMPTY_ZIP_FILE_NAME, inputStream ->
  57       monitoringUploadsManager.upload(inputStream, EMPTY_ZIP_FILE_NAME, VSP_ID, VERSION, COMPONENT_ID,
  58           MonitoringUploadType.SNMP_TRAP, USER1));
  59   }
  60
  61   @Test
  62   public void testUploadInvalidZip() {
  63
  64     try {
  65       processFile("/notZipFile", inputStream ->
  66         monitoringUploadsManager
  67             .upload(inputStream, NOT_ZIP_FILE_NAME, VSP_ID, VERSION, COMPONENT_ID,
  68                     MonitoringUploadType.VES_EVENTS, USER1));
  69       Assert.fail();
  70     } catch (Exception exception) {
  71       Assert.assertEquals(exception.getMessage(), "Invalid zip file");
  72     }
  73   }
  74
  75   @Test
  76   public void testUploadZipWithFolders() {
  77
  78     try {
  79       processFile(ZIP_DIR + ZIP_WITH_FOLDERS_FILE_NAME, inputStream -> monitoringUploadsManager
  80           .upload(inputStream, ZIP_WITH_FOLDERS_FILE_NAME, VSP_ID, VERSION, COMPONENT_ID,
  81               MonitoringUploadType.SNMP_TRAP, USER1));
  82       Assert.fail();
  83     } catch (Exception exception) {
  84       Assert.assertEquals(exception.getMessage(), "Zip file should not contain folders");
  85     }
  86   }
  87
  88   @Test
  89   public void testUploadVEsEventZipWithNonYamlFiles() {
  90
  91     try {
  92       processFile(ZIP_DIR + INVALID_VES_FILE_NAME, inputStream -> monitoringUploadsManager
  93           .upload(inputStream, INVALID_VES_FILE_NAME, VSP_ID, VERSION, COMPONENT_ID,
  94               MonitoringUploadType.VES_EVENTS, USER1));
  95       Assert.fail();
  96     } catch (Exception exception) {
  97       Assert.assertEquals(exception.getMessage(),
  98           "Wrong VES EVENT Artifact was uploaded - all files contained in Artifact must be YAML " +
  99               "files (using .yaml/.yml extensions)");
 100     }
 101   }
 102
 103
 104   @Test
 105   public void testListMonitoringFilenames() {
 106     ComponentMonitoringUploadEntity artifact1 =
 107         new ComponentMonitoringUploadEntity(VSP_ID, VERSION, COMPONENT_ID, "artifact1");
 108     artifact1.setType(MonitoringUploadType.SNMP_TRAP);
 109     artifact1.setArtifactName(TRAP_FILE_NAME);
 110
 111     ComponentMonitoringUploadEntity artifact2 =
 112         new ComponentMonitoringUploadEntity(VSP_ID, VERSION, COMPONENT_ID, "artifact2");
 113     artifact2.setType(MonitoringUploadType.SNMP_POLL);
 114     artifact2.setArtifactName(POLL_FILE_NAME);
 115
 116     ComponentMonitoringUploadEntity artifact3 =
 117         new ComponentMonitoringUploadEntity(VSP_ID, VERSION, COMPONENT_ID, "artifact3");
 118     artifact3.setType(MonitoringUploadType.VES_EVENTS);
 119     artifact3.setArtifactName(VES_FILE_NAME);
 120
 121     doReturn(Arrays.asList(artifact1, artifact2, artifact3))
 122         .when(componentArtifactDaoMock).list(anyObject());
 123
 124     MonitoringUploadStatus monitoringUploadStatus =
 125         monitoringUploadsManager.listFilenames(VSP_ID, VERSION, COMPONENT_ID, USER1);
 126
 127     Assert.assertEquals(monitoringUploadStatus.getSnmpTrap(), TRAP_FILE_NAME);
 128     Assert.assertEquals(monitoringUploadStatus.getSnmpPoll(), POLL_FILE_NAME);
 129     Assert.assertEquals(monitoringUploadStatus.getVesEvent(), VES_FILE_NAME);
 130   }
 131
 132   @Test (expectedExceptions = CoreException.class)
 133   public void testDeleteComponentMibWhenNone() {
 134     doReturn(Optional.empty()).when(componentArtifactDaoMock).getByType(any());
 135     monitoringUploadsManager
 136         .delete(VSP_ID, VERSION, COMPONENT_ID, MonitoringUploadType.SNMP_POLL, USER1);
 137
 138     verify(componentArtifactDaoMock, never()).delete(anyObject());
 139   }
 140
 141   @Test
 142   public void testDeleteComponentMonitoringUpload() {
 143     doReturn(Optional
 144         .of(new ComponentMonitoringUploadEntity(VSP_ID, VERSION, COMPONENT_ID, "artifactId")))
 145         .when
 146             (componentArtifactDaoMock).getByType(anyObject());
 147
 148     monitoringUploadsManager
 149         .delete(VSP_ID, VERSION, COMPONENT_ID, MonitoringUploadType.SNMP_POLL, USER1);
 150
 151     verify(componentArtifactDaoMock).delete(anyObject());
 152   }
 153
 154
 155   private void processFile(String fileName, Consumer<InputStream> processor) {
 156
 157     URL url = this.getClass().getResource(fileName);
 158     try (InputStream inputStream = url.openStream()) {
 159       processor.accept(inputStream);
 160     } catch (IOException e) {
 161       throw new RuntimeException("Failed to process file: " + fileName, e);
 162     }
 163   }
 164 }