aai-traversal/src/test/java/org/onap/aai/rest/search/VnfTopologyFromVfModuleTest.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
  30 import org.onap.aai.exceptions.AAIException;
  31 import org.onap.aai.serialization.db.exceptions.NoEdgeRuleFoundException;
  32
  33 public class VnfTopologyFromVfModuleTest extends QueryTest {
  34     public VnfTopologyFromVfModuleTest() throws AAIException, NoEdgeRuleFoundException {
  35         super();
  36     }
  37
  38     @Test
  39     public void run() {
  40         super.run();
  41     }
  42
  43     @Override
  44     protected void createGraph() throws AAIException, NoEdgeRuleFoundException {
  45         //Set up the test graph
  46
  47
  48         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");
  49         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");
  50         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");
  51         Vertex customer = graph.addVertex(T.label, "customer", T.id, "3", "aai-node-type", "customer", "customer-id", "customer-id-1", "customer-name", "customer-name1");
  52         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");
  53         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");
  54         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");
  55         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");
  56         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");
  57         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");
  58         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");
  59         Vertex vserver = graph.addVertex(T.label, "vserver", T.id, "11", "aai-node-type", "vserver", "vserver-name1", "vservername1");
  60         Vertex tenant = graph.addVertex(T.label, "tenant", T.id, "12", "aai-node-type", "tenant", "tenant-name1", "tenant-name-1","tenant-id", "tenant-id-1");
  61         Vertex region1 = graph.addVertex(T.label, "cloud-region", T.id, "13", "aai-node-type", "cloud-region", "cloud-owner", "cloudOwner1");
  62         Vertex pserver = graph.addVertex(T.label, "pserver", T.id, "14", "aai-node-type", "pserver", "hostname", "pservername");
  63         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");
  64         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");
  65         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");
  66         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");
  67         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");
  68         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");
  69         Vertex vfmoduleStart = graph.addVertex(T.label, "vf-module", T.id, "21", "aai-node-type", "vf-module", "vf-module-id", "vf-module-id-2", "vf-module-name", "vf-module-name2");
  70
  71
  72
  73         GraphTraversalSource g = graph.traversal();
  74         rules.addTreeEdge(g, gnvf1, vfmoduleStart);//true
  75         rules.addEdge(g, gnvf1, serviceinstance);//false
  76         rules.addTreeEdge(g, serviceinstance, servicesubscription);//true
  77         rules.addTreeEdge(g, servicesubscription, customer);//true
  78         rules.addTreeEdge(g, serviceinstance, allottedresource);//true
  79         rules.addTreeEdge(g, gnvf1, vfmodule);//true
  80         rules.addEdge(g, gnvf1, volumegroup);//false
  81         rules.addTreeEdge(g, gnvf1, linter1);//true
  82         rules.addTreeEdge(g, linter1, l3inter1ipv4addresslist);//true
  83         rules.addEdge(g, l3inter1ipv4addresslist, l3network1);//false
  84         rules.addTreeEdge(g, linter1, l3inter1ipv6addresslist);//true
  85         rules.addEdge(g, l3inter1ipv6addresslist, l3network2);//false
  86         rules.addEdge(g, gnvf1, vserver);//false
  87         rules.addTreeEdge(g, vserver, tenant);//true
  88         rules.addTreeEdge(g, tenant, region1);//true
  89         rules.addEdge(g, vserver, pserver);//false
  90         rules.addTreeEdge(g, vserver, linter2);//false
  91         rules.addTreeEdge(g, linter2, l3inter2ipv4addresslist);//false
  92         rules.addEdge(g, l3inter2ipv4addresslist, l3network3);//false
  93         rules.addTreeEdge(g, linter2, l3inter2ipv6addresslist);//true
  94         rules.addEdge(g, l3inter2ipv6addresslist, l3network4);//true
  95
  96
  97         expectedResult.add(gnvf1);
  98         expectedResult.add(serviceinstance);
  99         expectedResult.add(customer);
 100         expectedResult.add(allottedresource);
 101         expectedResult.add(vfmodule);
 102         expectedResult.add(vfmoduleStart);
 103         expectedResult.add(volumegroup);
 104         expectedResult.add(l3inter1ipv4addresslist);
 105         expectedResult.add(l3network1);
 106         expectedResult.add(l3inter1ipv6addresslist);
 107         expectedResult.add(l3network2);
 108         expectedResult.add(vserver);
 109         expectedResult.add(tenant);
 110         expectedResult.add(region1);
 111         expectedResult.add(pserver);
 112         expectedResult.add(l3inter2ipv4addresslist);
 113         expectedResult.add(l3network3);
 114         expectedResult.add(l3inter2ipv6addresslist);
 115         expectedResult.add(l3network4);
 116
 117     }
 118
 119     @Override
 120     protected String getQueryName() {
 121         return  "vnf-topology-fromVfModule";
 122     }
 123     @Override
 124     protected void addStartNode(GraphTraversal<Vertex, Vertex> g) {
 125         g.has("vf-module-name", "vf-module-name2");
 126
 127     }
 128     @Override
 129     protected void addParam(Map<String, Object> params) {
 130         return;
 131     }
 132 }