aai-traversal/src/test/java/org/onap/aai/rest/search/VnfTopologyFromServiceInstanceTest.java

   1 /**
   2  * ============LICENSE_START=======================================================
   3  * org.onap.aai
   4  * ================================================================================
   5  * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *    http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20 package org.onap.aai.rest.search;
  21
  22 import java.util.Map;
  23
  24 import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversal;
  25 import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversalSource;
  26 import org.apache.tinkerpop.gremlin.structure.T;
  27 import org.apache.tinkerpop.gremlin.structure.Vertex;
  28 import org.junit.Test;
  29 import org.onap.aai.exceptions.AAIException;
  30 import org.onap.aai.serialization.db.exceptions.NoEdgeRuleFoundException;
  31
  32 public class VnfTopologyFromServiceInstanceTest extends QueryTest {
  33     public VnfTopologyFromServiceInstanceTest() throws AAIException, NoEdgeRuleFoundException {
  34         super();
  35     }
  36
  37     @Test
  38     public void run() {
  39         super.run();
  40     }
  41
  42     @Override
  43     protected void createGraph() throws AAIException, NoEdgeRuleFoundException {
  44         //Set up the test graph
  45         Vertex gnvf1 = graph.addVertex(T.label, "generic-vnf", T.id, "0", "aai-node-type", "generic-vnf", "vnf-id", "vnf-id-1", "vnf-name", "vnf-name-1");
  46         Vertex serviceinstance = graph.addVertex(T.label, "service-instance", T.id, "1", "aai-node-type", "service-instance", "service-instance-id", "service-instance-id-1", "service-instance-name", "service-instance-name-1");
  47         Vertex servicesubscription = graph.addVertex(T.label, "service-subscription", T.id, "2", "aai-node-type", "service-subscription", "service-subscription-id", "service-subscription-id-1","service-subscription-name","service-subscription-name1");
  48         Vertex customer = graph.addVertex(T.label, "customer", T.id, "3", "aai-node-type", "customer", "customer-id", "customer-id-1", "customer-name", "customer-name1");
  49         Vertex allottedresource = graph.addVertex(T.label, "allotted-resource", T.id, "4", "aai-node-type", "allotted-resource", "allotted-resource-id", "allotted-resource-id-1", "allotted-resource-name", "allotted-resource-name1");
  50         Vertex vfmodule = graph.addVertex(T.label, "vf-module", T.id, "5", "aai-node-type", "vf-module", "vf-module-id", "vf-module-id-1", "vf-module-name", "vf-module-name1");
  51         Vertex volumegroup = graph.addVertex(T.label, "volume-group", T.id, "6", "aai-node-type", "volume-group", "volume-group-id", "volume-group-id-1", "volume-group-name", "volume-group-name1");
  52         Vertex linter1 = graph.addVertex(T.label, "l-interface", T.id, "7", "aai-node-type", "l-interface", "l-interface-id", "l-interface-id-1", "l-interface-name", "l-interface-name1");
  53         Vertex l3inter1ipv4addresslist = graph.addVertex(T.label, "interface-ipv4-address-list", T.id, "8", "aai-node-type", "l3-interface-ipv4-address-list", "l3-interface-ipv4-address-list-id", "l3-interface-ipv4-address-list-id-1", "l3-interface-ipv6-address-list-name", "l3-interface-ipv6-address-list-name1");
  54         Vertex l3network1 = graph.addVertex(T.label, "l3-network", T.id, "9", "aai-node-type", "l3-network", "ll3-network-id", "l3-network-id-1", "l3-network-name", "l3-network-name1");
  55         Vertex l3inter1ipv6addresslist = graph.addVertex(T.label, "l3-interface-ipv6-address-list", T.id, "10", "aai-node-type", "l3-interface-ipv6-address-list", "l3-interface-ipv6-address-list-id", "l3-interface-ipv6-address-list-id-1", "l3-interface-ipv6-address-list-name", "l3-interface-ipv6-address-list-name1");
  56         Vertex vserver = graph.addVertex(T.label, "vserver", T.id, "11", "aai-node-type", "vserver", "vserver-name1", "vservername1");
  57         Vertex tenant = graph.addVertex(T.label, "tenant", T.id, "12", "aai-node-type", "tenant", "tenant-name1", "tenant-name-1","tenant-id", "tenant-id-1");
  58         Vertex region1 = graph.addVertex(T.label, "cloud-region", T.id, "13", "aai-node-type", "cloud-region", "cloud-owner", "cloudOwner1");
  59         Vertex pserver = graph.addVertex(T.label, "pserver", T.id, "14", "aai-node-type", "pserver", "hostname", "pservername");
  60         Vertex linter2 = graph.addVertex(T.label, "l-interface", T.id, "15", "aai-node-type", "l-interface", "l-interface-id", "l-interface-id-2", "l-interface-name", "l-interface-name2");
  61         Vertex l3inter2ipv4addresslist = graph.addVertex(T.label, "interface-ipv6-address-list", T.id, "16", "aai-node-type", "l3-interface-ipv6-address-list", "l3-interface-ipv6-address-list-id", "l3-interface-ipv6-address-list-id-2", "l3-interface-ipv6-address-list-name", "l3-interface-ipv6-address-list-name2");
  62         Vertex l3network2 = graph.addVertex(T.label, "l3-network", T.id, "17", "aai-node-type", "l3-network", "ll3-network-id", "l3-network-id-2", "l3-network-name", "l3-network-name2");
  63         Vertex l3inter2ipv6addresslist = graph.addVertex(T.label, "l3-interface-ipv6-address-list", T.id, "18", "aai-node-type", "l3-interface-ipv6-address-list", "l3-interface-ipv6-address-list-id", "l3-interface-ipv6-address-list-id-2", "l3-interface-ipv6-address-list-name", "l3-interface-ipv6-address-list-name2");
  64         Vertex l3network3 = graph.addVertex(T.label, "l3-network", T.id, "19", "aai-node-type", "l3-network", "ll3-network-id", "l3-network-id-3", "l3-network-name", "l3-network-name3");
  65         Vertex l3network4 = graph.addVertex(T.label, "l3-network", T.id, "20", "aai-node-type", "l3-network", "ll3-network-id", "l3-network-id-4", "l3-network-name", "l3-network-name4");
  66
  67
  68
  69         GraphTraversalSource g = graph.traversal();
  70         rules.addEdge(g, gnvf1, serviceinstance);//false
  71         rules.addTreeEdge(g, serviceinstance, servicesubscription);//true
  72         rules.addTreeEdge(g, servicesubscription, customer);//true
  73         rules.addTreeEdge(g, serviceinstance, allottedresource);//true
  74         rules.addTreeEdge(g, gnvf1, vfmodule);//true
  75         rules.addEdge(g, gnvf1, volumegroup);//false
  76         rules.addTreeEdge(g, gnvf1, linter1);//true
  77         rules.addTreeEdge(g, linter1, l3inter1ipv4addresslist);//true
  78         rules.addEdge(g, l3inter1ipv4addresslist, l3network1);//false
  79         rules.addTreeEdge(g, linter1, l3inter1ipv6addresslist);//true
  80         rules.addEdge(g, l3inter1ipv6addresslist, l3network2);//false
  81         rules.addEdge(g, gnvf1, vserver);//false
  82         rules.addTreeEdge(g, vserver, tenant);//true
  83         rules.addTreeEdge(g, tenant, region1);//true
  84         rules.addEdge(g, vserver, pserver);//false
  85         rules.addTreeEdge(g, vserver, linter2);//false
  86         rules.addTreeEdge(g, linter2, l3inter2ipv4addresslist);//false
  87         rules.addEdge(g, l3inter2ipv4addresslist, l3network3);//false
  88         rules.addTreeEdge(g, linter2, l3inter2ipv6addresslist);//true
  89         rules.addEdge(g, l3inter2ipv6addresslist, l3network4);//true
  90
  91
  92         expectedResult.add(gnvf1);
  93         expectedResult.add(serviceinstance);
  94         expectedResult.add(customer);
  95         expectedResult.add(allottedresource);
  96         expectedResult.add(vfmodule);
  97         expectedResult.add(volumegroup);
  98         expectedResult.add(l3inter1ipv4addresslist);
  99         expectedResult.add(l3network1);
 100         expectedResult.add(l3inter1ipv6addresslist);
 101         expectedResult.add(l3network2);
 102         expectedResult.add(vserver);
 103         expectedResult.add(tenant);
 104         expectedResult.add(region1);
 105         expectedResult.add(pserver);
 106         expectedResult.add(l3inter2ipv4addresslist);
 107         expectedResult.add(l3network3);
 108         expectedResult.add(l3inter2ipv6addresslist);
 109         expectedResult.add(l3network4);
 110
 111     }
 112
 113     @Override
 114     protected String getQueryName() {
 115         return  "vnf-topology-fromServiceInstance";
 116     }
 117     @Override
 118     protected void addStartNode(GraphTraversal<Vertex, Vertex> g) {
 119         g.has("service-instance-name", "service-instance-name-1");
 120
 121     }
 122     @Override
 123     protected void addParam(Map<String, Object> params) {
 124         return;
 125     }
 126 }