re base code
[sdc.git] / catalog-model / src / test / java / org / openecomp / sdc / be / model / jsontitan / operations / TopologyTemplateOperationTest.java
index e8655b5..6193c5d 100644 (file)
@@ -1,13 +1,7 @@
 package org.openecomp.sdc.be.model.jsontitan.operations;
 
-import static org.assertj.core.api.Assertions.assertThat;
-import static org.mockito.ArgumentMatchers.any;
-import static org.mockito.ArgumentMatchers.eq;
-import static org.mockito.Mockito.when;
-
-import java.util.Collections;
-import java.util.HashMap;
-
+import com.thinkaurelius.titan.core.TitanVertex;
+import fj.data.Either;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.InjectMocks;
@@ -19,15 +13,19 @@ import org.openecomp.sdc.be.dao.jsongraph.TitanDao;
 import org.openecomp.sdc.be.dao.jsongraph.types.EdgeLabelEnum;
 import org.openecomp.sdc.be.dao.jsongraph.types.JsonParseFlagEnum;
 import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
-import org.openecomp.sdc.be.datatypes.elements.MapCapabiltyProperty;
+import org.openecomp.sdc.be.datatypes.elements.MapCapabilityProperty;
 import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
 import org.openecomp.sdc.be.model.PolicyDefinition;
 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
 import org.openecomp.sdc.be.model.operations.impl.UniqueIdBuilder;
 
-import com.thinkaurelius.titan.core.TitanVertex;
+import java.util.Collections;
+import java.util.HashMap;
 
-import fj.data.Either;
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.when;
 
 @RunWith(MockitoJUnitRunner.class)
 //error scenarios
@@ -58,7 +56,7 @@ public class TopologyTemplateOperationTest {
     @Test
     public void updateToscaDataDeepElements_failedToFetchContainerVertex() {
         when(titanDao.getVertexById(CONTAINER_ID, JsonParseFlagEnum.NoParse)).thenReturn(Either.right(TitanOperationStatus.INVALID_ID));
-        StorageOperationStatus storageOperationStatus = topologyTemplateOperation.updateToscaDataDeepElementsBlockToToscaElement(CONTAINER_ID, EdgeLabelEnum.CALCULATED_CAP_PROPERTIES, new MapCapabiltyProperty(), "");
+        StorageOperationStatus storageOperationStatus = topologyTemplateOperation.updateToscaDataDeepElementsBlockToToscaElement(CONTAINER_ID, EdgeLabelEnum.CALCULATED_CAP_PROPERTIES, new MapCapabilityProperty(), "");
         assertThat(storageOperationStatus).isEqualTo(StorageOperationStatus.INVALID_ID);
     }
 
@@ -67,44 +65,44 @@ public class TopologyTemplateOperationTest {
         GraphVertex containerVertex = new GraphVertex();
         when(titanDao.getVertexById(CONTAINER_ID, JsonParseFlagEnum.NoParse)).thenReturn(Either.left(containerVertex));
         when(titanDao.getChildVertex(containerVertex, EdgeLabelEnum.CALCULATED_CAP_PROPERTIES, JsonParseFlagEnum.ParseJson)).thenReturn(Either.right(TitanOperationStatus.NOT_FOUND));
-        StorageOperationStatus storageOperationStatus = topologyTemplateOperation.updateToscaDataDeepElementsBlockToToscaElement(CONTAINER_ID, EdgeLabelEnum.CALCULATED_CAP_PROPERTIES, new MapCapabiltyProperty(), "");
+        StorageOperationStatus storageOperationStatus = topologyTemplateOperation.updateToscaDataDeepElementsBlockToToscaElement(CONTAINER_ID, EdgeLabelEnum.CALCULATED_CAP_PROPERTIES, new MapCapabilityProperty(), "");
         assertThat(storageOperationStatus).isEqualTo(StorageOperationStatus.NOT_FOUND);
     }
     
-       @Test
+    @Test
     public void addPolicyToToscaElementSuccessTest(){
-       TitanOperationStatus status = TitanOperationStatus.OK;
-       StorageOperationStatus result = addPolicyToToscaElementWithStatus(status);
-       assertThat(result).isEqualTo(StorageOperationStatus.OK);
+        TitanOperationStatus status = TitanOperationStatus.OK;
+        StorageOperationStatus result = addPolicyToToscaElementWithStatus(status);
+        assertThat(result).isEqualTo(StorageOperationStatus.OK);
     }
     
-       @Test
+    @Test
     public void addPolicyToToscaElementFailureTest(){
-       TitanOperationStatus status = TitanOperationStatus.ALREADY_EXIST;
-       StorageOperationStatus result = addPolicyToToscaElementWithStatus(status);
-       assertThat(result).isEqualTo(StorageOperationStatus.ENTITY_ALREADY_EXISTS);
+        TitanOperationStatus status = TitanOperationStatus.ALREADY_EXIST;
+        StorageOperationStatus result = addPolicyToToscaElementWithStatus(status);
+        assertThat(result).isEqualTo(StorageOperationStatus.ENTITY_ALREADY_EXISTS);
     }
 
-       @SuppressWarnings("unchecked")
-       private StorageOperationStatus addPolicyToToscaElementWithStatus(TitanOperationStatus status) {
-               GraphVertex componentV = new GraphVertex();
-       componentV.setVertex(Mockito.mock(TitanVertex.class));
-       GraphVertex dataV = new GraphVertex();
-       dataV.setVertex(Mockito.mock(TitanVertex.class));
-       String componentName = "componentName";
-       String componentId = UniqueIdBuilder.buildResourceUniqueId();
-       String policyTypeName = "org.openecomp.policies.placement.valet.Affinity";
-       componentV.setJsonMetadataField(JsonPresentationFields.NAME, componentName);
-       componentV.setUniqueId(componentId);
-       PolicyDefinition policy = new PolicyDefinition();
-       policy.setPolicyTypeName(policyTypeName);
-       int counter = 0;
-       Either<GraphVertex, TitanOperationStatus> toscaDataVertexRes = Either.right(TitanOperationStatus.NOT_FOUND);
-       when(titanDao.getChildVertex(eq(componentV), eq(EdgeLabelEnum.POLICIES), eq(JsonParseFlagEnum.ParseJson))).thenReturn(toscaDataVertexRes);
-               Either<GraphVertex, TitanOperationStatus> createVertex = Either.left(dataV);
-               when(titanDao.createVertex(any(GraphVertex.class))).thenReturn(createVertex);
-               when(titanDao.createEdge(any(TitanVertex.class), any(TitanVertex.class), any(EdgeLabelEnum.class), any(HashMap.class))).thenReturn(status);
-       return topologyTemplateOperation.addPolicyToToscaElement(componentV, policy, counter);
-       }
+    @SuppressWarnings("unchecked")
+    private StorageOperationStatus addPolicyToToscaElementWithStatus(TitanOperationStatus status) {
+        GraphVertex componentV = new GraphVertex();
+        componentV.setVertex(Mockito.mock(TitanVertex.class));
+        GraphVertex dataV = new GraphVertex();
+        dataV.setVertex(Mockito.mock(TitanVertex.class));
+        String componentName = "componentName";
+        String componentId = UniqueIdBuilder.buildResourceUniqueId();
+        String policyTypeName = "org.openecomp.policies.placement.valet.Affinity";
+        componentV.setJsonMetadataField(JsonPresentationFields.NAME, componentName);
+        componentV.setUniqueId(componentId);
+        PolicyDefinition policy = new PolicyDefinition();
+        policy.setPolicyTypeName(policyTypeName);
+        int counter = 0;
+        Either<GraphVertex, TitanOperationStatus> toscaDataVertexRes = Either.right(TitanOperationStatus.NOT_FOUND);
+        when(titanDao.getChildVertex(eq(componentV), eq(EdgeLabelEnum.POLICIES), eq(JsonParseFlagEnum.ParseJson))).thenReturn(toscaDataVertexRes);
+        Either<GraphVertex, TitanOperationStatus> createVertex = Either.left(dataV);
+        when(titanDao.createVertex(any(GraphVertex.class))).thenReturn(createVertex);
+        when(titanDao.createEdge(any(TitanVertex.class), any(TitanVertex.class), any(EdgeLabelEnum.class), any(HashMap.class))).thenReturn(status);
+        return topologyTemplateOperation.addPolicyToToscaElement(componentV, policy, counter);
+    }
     
 }