Catalog alignment
[sdc.git] / catalog-model / src / test / java / org / openecomp / sdc / be / model / jsonjanusgraph / operations / ToscaElementOperationCatalogTest.java
index ef820ed..15653c0 100644 (file)
@@ -21,6 +21,8 @@
 package org.openecomp.sdc.be.model.jsonjanusgraph.operations;
 
 import fj.data.Either;
+import org.apache.tinkerpop.gremlin.structure.Direction;
+import org.apache.tinkerpop.gremlin.structure.Edge;
 import org.apache.tinkerpop.gremlin.structure.Vertex;
 import org.apache.tinkerpop.gremlin.structure.VertexProperty;
 import org.junit.Before;
@@ -37,6 +39,7 @@ import org.openecomp.sdc.be.dao.neo4j.GraphPropertiesDictionary;
 import org.openecomp.sdc.be.model.catalog.CatalogComponent;
 
 import java.util.ArrayList;
+import java.util.Iterator;
 import java.util.List;
 
 import static org.junit.Assert.assertEquals;
@@ -46,14 +49,23 @@ import static org.mockito.Mockito.when;
 @RunWith(MockitoJUnitRunner.class)
 public class ToscaElementOperationCatalogTest {
 
+    private static final String UPDATER_ID = "m08740";
     private ArrayList<Vertex> vertexList = new ArrayList<>();
 
     @Mock
     Vertex vertex;
     @Mock
+    Edge edge;
+    @Mock
+    Vertex outVertex;
+    @Mock
+    Iterator<Edge> edges;
+    @Mock
     JanusGraphDao janusGraphDao;
     @Mock
     VertexProperty<Object> property;
+    @Mock
+    VertexProperty<Object> updaterProperty;
 
     @InjectMocks
     private ToscaElementOperation toscaOperation = new TopologyTemplateOperation();
@@ -67,6 +79,20 @@ public class ToscaElementOperationCatalogTest {
         when(vertex.property(GraphPropertiesDictionary.METADATA.getProperty())).thenReturn(property);
     }
 
+    private void stubEmptyUpdater() {
+        when(vertex.edges(Direction.IN, EdgeLabelEnum.STATE.name())).thenReturn(edges);
+        when(edges.hasNext()).thenReturn(false);
+    }
+
+    private void stubExistingUpdater() {
+        when(vertex.edges(Direction.IN, EdgeLabelEnum.STATE.name())).thenReturn(edges);
+        when(edges.hasNext()).thenReturn(true);
+        when(edges.next()).thenReturn(edge);
+        when(edge.outVertex()).thenReturn(outVertex);
+        when(updaterProperty.value()).thenReturn(UPDATER_ID);
+        when(outVertex.property(GraphPropertiesDictionary.USERID.getProperty())).thenReturn(updaterProperty);
+    }
+
     @Test
     public void getComponentFromCatalogWhenDeleteIsTrue() {
         final String vertexJsonIsDeletedTrue = "{\"lifecycleState\":\"CERTIFIED\",\"componentType\":\"RESOURCE\",\"vendorRelease\":\"1\",\"contactId\":\"ah7840\",\"lastUpdateDate\":1496119811038,\"icon\":\"att\",\"description\":\"Cloud\",\"creationDate\":1459432094781,\"vendorName\":\"AT&T\",\"mandatory\":false,\"version\":\"1.0\",\"tags\":[\"Cloud\"],\"highestVersion\":true,\"systemName\":\"Cloud\",\"name\":\"Cloud\",\"isDeleted\":true,\"invariantUuid\":\"ab5263fd-115c-41f2-8d0c-8b9279bce2b2\",\"UUID\":\"208cf7df-68a1-4ae7-afc9-409ea8012332\",\"normalizedName\":\"cloud\",\"toscaResourceName\":\"org.openecomp.resource.Endpoint.Cloud\",\"uniqueId\":\"9674e7e1-bc1a-41fe-b503-fbe996801475\",\"resourceType\":\"VFC\"}";
@@ -77,6 +103,7 @@ public class ToscaElementOperationCatalogTest {
 
     @Test
     public void getComponentFromCatalogWhenDeleteNotFound() {
+        stubEmptyUpdater();
         final String vertexJsonIsDeletedNotFound = "{\"lifecycleState\":\"CERTIFIED\",\"componentType\":\"RESOURCE\",\"vendorRelease\":\"1\",\"contactId\":\"ah7840\",\"lastUpdateDate\":1496119811038,\"icon\":\"att\",\"description\":\"Cloud\",\"creationDate\":1459432094781,\"vendorName\":\"AT&T\",\"mandatory\":false,\"version\":\"1.0\",\"tags\":[\"Cloud\"],\"highestVersion\":true,\"systemName\":\"Cloud\",\"name\":\"Cloud\",\"invariantUuid\":\"ab5263fd-115c-41f2-8d0c-8b9279bce2b2\",\"UUID\":\"208cf7df-68a1-4ae7-afc9-409ea8012332\",\"normalizedName\":\"cloud\",\"toscaResourceName\":\"org.openecomp.resource.Endpoint.Cloud\",\"uniqueId\":\"9674e7e1-bc1a-41fe-b503-fbe996801475\",\"resourceType\":\"VFC\"}";
         when(property.value()).thenReturn(vertexJsonIsDeletedNotFound);
         List<CatalogComponent> componentList = toscaOperation.getElementCatalogData(true, null).left().value();
@@ -85,9 +112,12 @@ public class ToscaElementOperationCatalogTest {
 
     @Test
     public void getComponentFromCatalogWhenDeleteIsFalse() {
+        stubExistingUpdater();
         final String vertexJsonIsDeletedFalse = "{\"lifecycleState\":\"CERTIFIED\",\"componentType\":\"RESOURCE\",\"vendorRelease\":\"1\",\"contactId\":\"ah7840\",\"lastUpdateDate\":1496119811038,\"icon\":\"att\",\"description\":\"Cloud\",\"creationDate\":1459432094781,\"vendorName\":\"AT&T\",\"mandatory\":false,\"version\":\"1.0\",\"tags\":[\"Cloud\"],\"highestVersion\":true,\"systemName\":\"Cloud\",\"name\":\"Cloud\",\"isDeleted\":false,\"invariantUuid\":\"ab5263fd-115c-41f2-8d0c-8b9279bce2b2\",\"UUID\":\"208cf7df-68a1-4ae7-afc9-409ea8012332\",\"normalizedName\":\"cloud\",\"toscaResourceName\":\"org.openecomp.resource.Endpoint.Cloud\",\"uniqueId\":\"9674e7e1-bc1a-41fe-b503-fbe996801475\",\"resourceType\":\"VFC\"}";
         when(property.value()).thenReturn(vertexJsonIsDeletedFalse);
         List<CatalogComponent> componentList = toscaOperation.getElementCatalogData(true, null).left().value();
         assertEquals(1, componentList.size());
+        assertEquals(UPDATER_ID, componentList.get(0).getLastUpdaterUserId());
     }
+
 }