1 files changed, 128 insertions, 0 deletions
diff --git a/openecomp-be/lib/openecomp-sdc-enrichment-lib/openecomp-sdc-enrichment-impl/src/test/java/org/openecomp/sdc/enrichment/impl/external/artifact/ProcessArtifactEnricherTest.java b/openecomp-be/lib/openecomp-sdc-enrichment-lib/openecomp-sdc-enrichment-impl/src/test/java/org/openecomp/sdc/enrichment/impl/external/artifact/ProcessArtifactEnricherTest.java
new file mode 100644
index 0000000000..cfb241483a
--- /dev/null
+++ b/openecomp-be/lib/openecomp-sdc-enrichment-lib/openecomp-sdc-enrichment-impl/src/test/java/org/openecomp/sdc/enrichment/impl/external/artifact/ProcessArtifactEnricherTest.java
@@ -0,0 +1,128 @@
+package org.openecomp.sdc.enrichment.impl.external.artifact;
+
+import org.mockito.ArgumentCaptor;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.mockito.MockitoAnnotations;
+import org.openecomp.core.enrichment.types.ArtifactCategory;
+import org.openecomp.core.model.dao.EnrichedServiceModelDao;
+import org.openecomp.core.model.types.ServiceArtifact;
+import org.openecomp.core.utilities.file.FileUtils;
+import org.openecomp.sdc.enrichment.EnrichmentInfo;
+import org.openecomp.sdc.vendorsoftwareproduct.dao.ProcessDao;
+import org.openecomp.sdc.vendorsoftwareproduct.dao.VendorSoftwareProductDao;
+import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ComponentEntity;
+import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ProcessEntity;
+import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ProcessType;
+import org.openecomp.sdc.versioning.dao.types.Version;
+import org.testng.Assert;
+import org.testng.annotations.BeforeMethod;
+import org.testng.annotations.Test;
+
+import java.io.File;
+import java.io.InputStream;
+import java.nio.ByteBuffer;
+import java.util.ArrayList;
+import java.util.Collection;
+
+import static org.mockito.Matchers.anyObject;
+import static org.mockito.Mockito.atLeastOnce;
+import static org.mockito.Mockito.when;
+
+public class ProcessArtifactEnricherTest {
+  @Mock
+  ProcessDao processDaoMock;
+  @Mock
+  EnrichedServiceModelDao enrichedServiceModelDaoMock;
+  @Mock
+  VendorSoftwareProductDao vendorSoftwareProductDaoMock;
+
+  @InjectMocks
+  ProcessArtifactEnricher processArtifactEnricher;
+
+
+  @BeforeMethod(alwaysRun = true)
+  public void injectDoubles() {
+    MockitoAnnotations.initMocks(this);
+  }
+
+  @Test
+  public void testEnrichComponent() throws Exception {
+    String vspId = "123";
+    String componentId = "1111111111";
+    Version version = new Version();
+    version.setMajor(1);
+    version.setMinor(0);
+
+    ComponentEntity componentEntity = getComponentEntity(vspId, version, componentId);
+    setMockToEnrichComponent(vspId, componentId, version);
+
+    ProcessEntity entity = new ProcessEntity(vspId, version, componentId, null);
+    ProcessEntity processEntity = new ProcessEntity();
+    processEntity.setType(ProcessType.Lifecycle_Operations);
+    processEntity.setVspId(vspId);
+    processEntity.setVersion(version);
+    processEntity.setComponentId(componentId);
+
+    Collection<ComponentEntity> componentList = new ArrayList<ComponentEntity>();
+    componentList.add(componentEntity);
+    when(vendorSoftwareProductDaoMock.listComponents(vspId, version)).thenReturn(componentList);
+
+    Collection<ProcessEntity> list = new ArrayList<ProcessEntity>();
+    list.add(processEntity);
+    when(processDaoMock.list(entity)).thenReturn(list);
+
+    EnrichmentInfo info = new EnrichmentInfo();
+    info.setVersion(version);
+    info.setKey(vspId);
+    processArtifactEnricher.enrich(info);
+
+    String componentName = componentEntity.getComponentCompositionData().getName();
+
+    ArgumentCaptor<ServiceArtifact> expectedServiceArtifact =
+        ArgumentCaptor.forClass(ServiceArtifact.class);
+    Mockito.verify(enrichedServiceModelDaoMock, atLeastOnce())
+        .storeExternalArtifact(expectedServiceArtifact.capture());
+    Assert
+        .assertEquals(expectedServiceArtifact.getValue().getName().startsWith(componentName), true);
+    Assert.assertEquals(expectedServiceArtifact.getValue().getName(),
+        componentName + File.separator + ArtifactCategory.DEPLOYMENT.getDisplayName() +
+            File.separator + "Lifecycle Operations" + File.separator + "artifact_1kb.txt");
+
+  }
+
+  private void setMockToEnrichComponent(String vspId, String componentId, Version version) {
+    ProcessEntity returnedArtifact = new ProcessEntity();
+    returnedArtifact.setVspId(vspId);
+    returnedArtifact.setVersion(version);
+    returnedArtifact.setComponentId(componentId);
+    returnedArtifact.setArtifactName("artifact_1kb.txt");
+    returnedArtifact.setArtifact(getMibByteBuffer("/mock/enrichProcess/artifact_1kb.txt"));
+
+    Mockito.when(processDaoMock.get(anyObject()))
+        .thenReturn(returnedArtifact);
+    Mockito.doNothing().when(enrichedServiceModelDaoMock).storeExternalArtifact(anyObject());
+  }
+
+  private ComponentEntity getComponentEntity(String vspId, Version version, String componentId) {
+    ComponentEntity componentEntity = new ComponentEntity();
+    componentEntity.setId(componentId);
+    componentEntity.setVspId(vspId);
+    componentEntity.setVersion(version);
+
+    String componentName = vspId + "enrichMib_server";
+    String compositionData = "{\n" +
+        "  \"name\": \"org.openecomp.resource.vfc.nodes.heat." + componentName + "\",\n" +
+        "  \"displayName\": \"" + componentName + "\"\n" +
+        "}";
+    componentEntity.setCompositionData(compositionData);
+    return componentEntity;
+  }
+
+  private ByteBuffer getMibByteBuffer(String fileName) {
+    InputStream mibFile = FileUtils.getFileInputStream(this.getClass().getResource(fileName));
+    byte[] mibBytes = FileUtils.toByteArray(mibFile);
+    return ByteBuffer.wrap(mibBytes);
+  }
+}