1 files changed, 101 insertions, 0 deletions
diff --git a/catalog-be/src/test/java/org/openecomp/sdc/be/components/merge/instance/ComponentInstanceHeatEnvMergeTest.java b/catalog-be/src/test/java/org/openecomp/sdc/be/components/merge/instance/ComponentInstanceHeatEnvMergeTest.java
new file mode 100644
index 0000000000..2101ea49ab
--- /dev/null
+++ b/catalog-be/src/test/java/org/openecomp/sdc/be/components/merge/instance/ComponentInstanceHeatEnvMergeTest.java
@@ -0,0 +1,101 @@
+package org.openecomp.sdc.be.components.merge.instance;
+
+import static org.mockito.Mockito.when;
+
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.mockito.MockitoAnnotations;
+import org.openecomp.sdc.be.components.impl.ArtifactsBusinessLogic;
+import org.openecomp.sdc.be.components.merge.heat.HeatEnvArtifactsMergeBusinessLogic;
+import org.openecomp.sdc.be.components.utils.ArtifactBuilder;
+import org.openecomp.sdc.be.components.utils.ComponentInstanceBuilder;
+import org.openecomp.sdc.be.components.utils.ResourceBuilder;
+import org.openecomp.sdc.be.impl.ComponentsUtils;
+import org.openecomp.sdc.be.model.ArtifactDefinition;
+import org.openecomp.sdc.be.model.Resource;
+import org.openecomp.sdc.be.model.User;
+import org.openecomp.sdc.common.api.ArtifactGroupTypeEnum;
+import org.openecomp.sdc.common.api.ArtifactTypeEnum;
+
+import fj.data.Either;
+
+public class ComponentInstanceHeatEnvMergeTest {
+
+    @InjectMocks
+    private ComponentInstanceHeatEnvMerge testInstance;
+
+    @Mock
+    private ArtifactsBusinessLogic artifactsBusinessLogicMock;
+
+    @Mock
+    private HeatEnvArtifactsMergeBusinessLogic heatEnvArtifactsMergeBusinessLogicMock;
+
+    @Mock
+    private ComponentsUtils componentsUtils;
+
+    private static final User USER = new User();
+
+    @Before
+    public void setUp() throws Exception {
+        MockitoAnnotations.initMocks(this);
+    }
+
+    @Test
+    public void mergeDataAfterCreate_mergeAndPersistArtifacts() throws Exception {
+        Map<String, ArtifactDefinition> nodeTypeArtifactsByName = buildMapOfHeatArtifacts("artifact1", "artifact2");
+        DataForMergeHolder dataHolder = new DataForMergeHolder();
+        dataHolder.setOrigComponentDeploymentArtifactsCreatedOnTheInstance(nodeTypeArtifactsByName);
+        String instanceId = "instance1";
+        Resource resource = buildResourceWithHeatArtifacts(instanceId, "heatArtifact1", "heatArtifact2");
+        List<ArtifactDefinition> mergedArtifacts = buildListOfArtifacts("artifact1, heatArtifact1");
+        when(heatEnvArtifactsMergeBusinessLogicMock.mergeInstanceHeatEnvArtifacts(dataHolder.getOrigComponentInstanceHeatEnvArtifacts(), resource.safeGetComponentInstanceHeatArtifacts(instanceId)))
+                                                   .thenReturn(mergedArtifacts);
+        expectMergedArtifactsToBePersisted(mergedArtifacts, instanceId, resource);
+        testInstance.mergeDataAfterCreate(USER, dataHolder, resource, instanceId);
+    }
+
+    private void expectMergedArtifactsToBePersisted(List<ArtifactDefinition> mergedArtifacts, String instanceId, Resource resource) {
+        for (ArtifactDefinition mergedArtifact : mergedArtifacts) {
+            Map<String, Object> json = new HashMap<>();
+            when(artifactsBusinessLogicMock.buildJsonForUpdateArtifact(mergedArtifact, ArtifactGroupTypeEnum.DEPLOYMENT, null)).thenReturn(json);
+            ArtifactsBusinessLogic.ArtifactOperationInfo artifactUpdateOperation = artifactsBusinessLogicMock.new ArtifactOperationInfo(false, false, ArtifactsBusinessLogic.ArtifactOperationEnum.Update);
+            when(artifactsBusinessLogicMock.updateResourceInstanceArtifactNoContent(Mockito.eq(instanceId), Mockito.eq(resource),
+                                                                                    Mockito.eq(USER), Mockito.eq(json),
+                                                                                    Mockito.refEq(artifactUpdateOperation),
+                                                                                    Mockito.isNull(ArtifactDefinition.class)))
+                                           .thenReturn(Either.left(Either.left(new ArtifactDefinition())));
+        }
+    }
+
+    private Resource buildResourceWithHeatArtifacts(String instanceId, String ... artifacts) {
+        ComponentInstanceBuilder componentInstanceBuilder = new ComponentInstanceBuilder().setId(instanceId);
+        for (String artifact : artifacts) {
+            ArtifactDefinition heatArtifact = new ArtifactBuilder().setType(ArtifactTypeEnum.HEAT_ARTIFACT.getType()).setName(artifact).build();
+            componentInstanceBuilder.addDeploymentArtifact(heatArtifact);
+        }
+        return new ResourceBuilder().addComponentInstance(componentInstanceBuilder.build()).build();
+    }
+
+    private Map<String, ArtifactDefinition> buildMapOfHeatArtifacts(String ... artifacts) {
+        Map<String, ArtifactDefinition> artifactsByName = new HashMap<>();
+        for (String artifact : artifacts) {
+            ArtifactDefinition heatArtifact = new ArtifactBuilder().setType(ArtifactTypeEnum.HEAT_ARTIFACT.getType()).setName(artifact).build();
+            artifactsByName.put(artifact, heatArtifact);
+        }
+        return artifactsByName;
+    }
+
+    private List<ArtifactDefinition> buildListOfArtifacts(String ... artifacts) {
+        return Stream.of(artifacts).map(artifact -> new ArtifactBuilder().setName(artifact).build()).collect(Collectors.toList());
+    }
+
+}
+\ No newline at end of file