src/test/java/org/onap/crud/parser/util/EdgePayloadUtilTest.java

   1 /**\r
   2  * ============LICENSE_START=======================================================\r
   3  * org.onap.aai\r
   4  * ================================================================================\r
   5  * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.\r
   6  * Copyright © 2017-2018 Amdocs\r
   7  * ================================================================================\r
   8  * Licensed under the Apache License, Version 2.0 (the "License");\r
   9  * you may not use this file except in compliance with the License.\r
  10  * You may obtain a copy of the License at\r
  11  *\r
  12  *       http://www.apache.org/licenses/LICENSE-2.0\r
  13  *\r
  14  * Unless required by applicable law or agreed to in writing, software\r
  15  * distributed under the License is distributed on an "AS IS" BASIS,\r
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  17  * See the License for the specific language governing permissions and\r
  18  * limitations under the License.\r
  19  * ============LICENSE_END=========================================================\r
  20  */\r
  21 package org.onap.crud.parser.util;\r
  22 \r
  23 import java.util.ArrayList;\r
  24 import java.util.List;\r
  25 import org.junit.Assert;\r
  26 import org.junit.Test;\r
  27 import org.onap.crud.entity.Edge;\r
  28 import org.onap.crud.entity.Vertex;\r
  29 import org.onap.crud.exception.CrudException;\r
  30 \r
  31 public class EdgePayloadUtilTest {\r
  32 \r
  33     @Test\r
  34     public void testGetVertexNodeType() throws CrudException {\r
  35         Assert.assertEquals("vserver", EdgePayloadUtil\r
  36                 .getVertexNodeType("services/inventory/v12/vserver/50bdab41-ad1c-4d00-952c-a0aa5d827811"));\r
  37     }\r
  38 \r
  39     @Test\r
  40     public void testGetVertexNodeId() throws CrudException {\r
  41         Assert.assertEquals("50bdab41-ad1c-4d00-952c-a0aa5d827811",\r
  42                 EdgePayloadUtil.getVertexNodeId("services/inventory/v12/vserver/50bdab41-ad1c-4d00-952c-a0aa5d827811"));\r
  43     }\r
  44 \r
  45     @Test\r
  46     public void testGenerateEdgeKey() throws CrudException {\r
  47         Assert.assertEquals("vserver:pserver:tosca.relationships.HostedOn",\r
  48                 EdgePayloadUtil.generateEdgeKey("services/inventory/v12/vserver/50bdab41-ad1c-4d00-952c-a0aa5d827811",\r
  49                         "services/inventory/v12/pserver/1d326bc7-b985-492b-9604-0d5d1f06f908",\r
  50                         "tosca.relationships.HostedOn"));\r
  51     }\r
  52 \r
  53     @Test\r
  54     public void testKeyGenerationEnforcesSourceAndTargetOrder() throws Exception {\r
  55         String generateEdgeKey1 = EdgePayloadUtil.generateEdgeKey(\r
  56                 "services/inventory/v12/vserver/50bdab41-ad1c-4d00-952c-a0aa5d827811",\r
  57                 "services/inventory/v12/pserver/1d326bc7-b985-492b-9604-0d5d1f06f908", "tosca.relationships.HostedOn");\r
  58 \r
  59         String generateEdgeKey2 = EdgePayloadUtil.generateEdgeKey(\r
  60                 "services/inventory/v12/pserver/1d326bc7-b985-492b-9604-0d5d1f06f908",\r
  61                 "services/inventory/v12/vserver/50bdab41-ad1c-4d00-952c-a0aa5d827811", "tosca.relationships.HostedOn");\r
  62 \r
  63         Assert.assertNotEquals(generateEdgeKey1, generateEdgeKey2);\r
  64     }\r
  65 \r
  66 \r
  67     @Test\r
  68     public void testGetBuilderFromEdgePayload() throws CrudException {\r
  69         Edge.Builder builder = EdgePayloadUtil.getBuilderFromEdgePayload(\r
  70                 "services/inventory/v12/vserver/50bdab41-ad1c-4d00-952c-a0aa5d827811",\r
  71                 "services/inventory/v12/pserver/1d326bc7-b985-492b-9604-0d5d1f06f908", "tosca.relationships.HostedOn");\r
  72 \r
  73         Edge edge = builder.build();\r
  74         Assert.assertEquals("tosca.relationships.HostedOn", edge.getType());\r
  75         Assert.assertEquals("vserver", edge.getSource().getType());\r
  76         Assert.assertEquals("50bdab41-ad1c-4d00-952c-a0aa5d827811", edge.getSource().getId().get());\r
  77         Assert.assertEquals("pserver", edge.getTarget().getType());\r
  78         Assert.assertEquals("1d326bc7-b985-492b-9604-0d5d1f06f908", edge.getTarget().getId().get());\r
  79     }\r
  80 \r
  81     @Test\r
  82     public void tesGetBuilderFromEdge() throws CrudException {\r
  83         Edge edge = createEdge("vserver", "pserver");\r
  84 \r
  85         Assert.assertNotNull(edge);\r
  86         Assert.assertNotNull(EdgePayloadUtil.getBuilderFromEdge(edge));\r
  87     }\r
  88 \r
  89     @Test\r
  90     public void testFilterEdgesByRelatedVertexAndType() throws CrudException {\r
  91         List<Edge> edges = new ArrayList<>();\r
  92         edges.add(createEdge("vserver", "pserver"));\r
  93         edges.add(createEdge("vce", "pserver"));\r
  94         edges.add(createEdge("snapshot", "pserver"));\r
  95         edges.add(createEdge("vserver", "pserver"));\r
  96 \r
  97         List<Edge> filteredEdges =\r
  98                 EdgePayloadUtil.filterEdgesByRelatedVertexAndType("vserver", "tosca.relationships.HostedOn", edges);\r
  99 \r
 100         Assert.assertNotNull(filteredEdges);\r
 101         Assert.assertEquals(2, filteredEdges.size());\r
 102     }\r
 103 \r
 104     private Edge createEdge(String sourceVertexType, String targetVertexType) {\r
 105         return new Edge.Builder("tosca.relationships.HostedOn").id("test")\r
 106                 .source(new Vertex.Builder(sourceVertexType).id("50bdab41-ad1c-4d00-952c-a0aa5d827811").build())\r
 107                 .target(new Vertex.Builder(targetVertexType).id("1d326bc7-b985-492b-9604-0d5d1f06f908").build())\r
 108                 .build();\r
 109     }\r
 110 }\r