gerrit.onap Code Review - aai/traversal.git/blob

   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 com.google.common.base.CaseFormat;
  23
  24 import java.util.Map;
  25
  26 import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversal;
  27 import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversalSource;
  28 import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.__;
  29 import org.apache.tinkerpop.gremlin.structure.T;
  30 import org.apache.tinkerpop.gremlin.structure.Vertex;
  31 import org.junit.Test;
  32 import org.onap.aai.exceptions.AAIException;
  33 import org.onap.aai.serialization.db.exceptions.NoEdgeRuleFoundException;
  34 import org.slf4j.Logger;
  35 import org.slf4j.LoggerFactory;
  36
  37 public class TopologyFromSubscriberNameAndServiceTypeTest extends QueryTest {
  38     private static final Logger LOGGER =
  39         LoggerFactory.getLogger(TopologyFromSubscriberNameAndServiceTypeTest.class);
  40
  41     public TopologyFromSubscriberNameAndServiceTypeTest()
  42         throws AAIException, NoEdgeRuleFoundException {
  43         super();
  44     }
  45
  46     @Test
  47     public void run() {
  48         super.run();
  49     }
  50
  51     @Override
  52     protected void createGraph() throws AAIException, NoEdgeRuleFoundException {
  53         // Set up the test graph
  54         Vertex customer1 = graph.addVertex(T.label, "customer", T.id, "1", "aai-node-type",
  55             "customer", "global-customer-id", "global-customer-id-1", "subscriber-name",
  56             "subscriber-name-1");
  57         // Customer has 2 subscriptions - we filter by subscription-type
  58         Vertex servicesubscription1 = graph.addVertex(T.label, "service-subscription", T.id, "2",
  59             "aai-node-type", "service-subscription", "service-type", "service-type-1");
  60         Vertex servicesubscription112 = graph.addVertex(T.label, "service-subscription", T.id,
  61             "112", "aai-node-type", "service-subscription", "service-type", "service-type-112");
  62
  63         Vertex serviceinstance1 = graph.addVertex(T.label, "service-instance", T.id, "3",
  64             "aai-node-type", "service-instance", "service-instance-id", "service-instance-id-1",
  65             "service-instance-name", "service-instance-name-1");
  66         Vertex serviceinstance113 = graph.addVertex(T.label, "service-instance", T.id, "113",
  67             "aai-node-type", "service-instance", "service-instance-id", "service-instance-id-113",
  68             "service-instance-name", "service-instance-name-113");
  69
  70         Vertex gnvf1 =
  71             graph.addVertex(T.label, "generic-vnf", T.id, "4", "aai-node-type", "generic-vnf",
  72                 "vnf-id", "vnf-id-1", "vnf-name", "vnf-name-1", "service-id", "service-id-1");
  73         Vertex gnvf114 =
  74             graph.addVertex(T.label, "generic-vnf", T.id, "114", "aai-node-type", "generic-vnf",
  75                 "vnf-id", "vnf-id-114", "vnf-name", "vnf-name-2", "service-id", "service-id-2");
  76
  77         Vertex pserver1 = graph.addVertex(T.label, "pserver", T.id, "5", "aai-node-type", "pserver",
  78             "hostname", "pservername1");
  79         Vertex pserver115 = graph.addVertex(T.label, "pserver", T.id, "115", "aai-node-type",
  80             "pserver", "hostname", "pservername115");
  81
  82         Vertex complex1 = graph.addVertex(T.label, "complex", T.id, "6", "aai-node-type", "complex",
  83             "physical-location-id", "physical-location-id-1", "country", "US");
  84
  85         Vertex pnfint1 = graph.addVertex(T.label, "p-interface", T.id, "7", "aai-node-type",
  86             "p-interface", "interface-name", "ge0/0/0");
  87
  88         Vertex vserver1 = graph.addVertex(T.label, "vserver", T.id, "8", "aai-node-type", "vserver",
  89             "vserver-id", "vserver-id-1", "vserver-name", "vserver-name-1", "vserver-selflink",
  90             "vserver-selflink-1");
  91
  92         Vertex vserver116 = graph.addVertex(T.label, "vserver", T.id, "116", "aai-node-type",
  93             "vserver", "vserver-id", "vserver-id-116", "vserver-name", "vserver-name-116",
  94             "vserver-selflink", "vserver-selflink-116");
  95
  96         Vertex flavor1 = graph.addVertex(T.label, "flavor", T.id, "9", "aai-node-type", "flavor",
  97             "flavor-id", "flavor-id-1", "flavor-name", "flavor-name-1", "flavor-selflink",
  98             "flavor-selflink-1");
  99         Vertex image1 = graph.addVertex(T.label, "image", T.id, "10", "aai-node-type", "image",
 100             "image-id", "image-id-1", "image-name", "image-name-1", "image-os-distro",
 101             "image-os-distro-1", "image-os-version", "image-os-version-1");
 102
 103         Vertex linter1 =
 104             graph.addVertex(T.label, "l-interface", T.id, "11", "aai-node-type", "l-interface",
 105                 "l-interface-id", "l-interface-id-1", "l-interface-name", "l-interface-name1");
 106         Vertex logicallink1 = graph.addVertex(T.label, "logical-link", T.id, "12", "aai-node-type",
 107             "logical-link", "link-name", "link-name-1", "l-interface-name", "l-interface-name1");
 108         Vertex l3inter1ipv4addresslist = graph.addVertex(T.label, "interface-ipv4-address-list",
 109             T.id, "13", "aai-node-type", "l3-interface-ipv4-address-list",
 110             "l3-interface-ipv4-address", "l3-interface-ipv4-address-1");
 111         Vertex subnet1 = graph.addVertex(T.label, "subnet", T.id, "14", "aai-node-type", "subnet",
 112             "subnet-id", "subnet-id-1");
 113         Vertex l3network1 = graph.addVertex(T.label, "l3-network", T.id, "15", "aai-node-type",
 114             "l3-network", "network-id", "network-id-1", "network-name", "network-name-1");
 115         Vertex l3inter1ipv6addresslist = graph.addVertex(T.label, "l3-interface-ipv6-address-list",
 116             T.id, "16", "aai-node-type", "l3-interface-ipv6-address-list",
 117             "l3-interface-ipv6-address", "l3-interface-ipv6-address-1");
 118         Vertex subnet2 = graph.addVertex(T.label, "subnet", T.id, "17", "aai-node-type", "subnet",
 119             "subnet-id", "subnet-id-2");
 120         Vertex l3network2 = graph.addVertex(T.label, "l3-network", T.id, "18", "aai-node-type",
 121             "l3-network", "network-id", "network-id-2", "network-name", "network-name2");
 122
 123         Vertex customer2 = graph.addVertex(T.label, "customer", T.id, "19", "aai-node-type",
 124             "customer", "global-customer-id", "global-customer-id-2", "subscriber-name",
 125             "subscriber-name-2");
 126         Vertex servicesubscription2 = graph.addVertex(T.label, "service-subscription", T.id, "20",
 127             "aai-node-type", "service-subscription", "service-type", "service-type-2");
 128         Vertex serviceinstance2 = graph.addVertex(T.label, "service-instance", T.id, "21",
 129             "aai-node-type", "service-instance", "service-instance-id", "service-instance-id-2",
 130             "service-instance-name", "service-instance-name-2");
 131         Vertex gnvf2 =
 132             graph.addVertex(T.label, "generic-vnf", T.id, "22", "aai-node-type", "generic-vnf",
 133                 "vnf-id", "vnf-id-2", "vnf-name", "vnf-name-2", "service-id", "service-id-2");
 134
 135         Vertex pserver2 = graph.addVertex(T.label, "pserver", T.id, "23", "aai-node-type",
 136             "pserver", "hostname", "pservername1");
 137         Vertex complex2 = graph.addVertex(T.label, "complex", T.id, "24", "aai-node-type",
 138             "complex", "physical-location-id", "physical-location-id-2", "country", "US");
 139         Vertex pnfint2 = graph.addVertex(T.label, "p-interface", T.id, "25", "aai-node-type",
 140             "p-interface", "interface-name", "ge0/0/0");
 141
 142         Vertex plink1 = graph.addVertex(T.label, "physical-link", T.id, "251", "aai-node-type",
 143             "physical-link", "link-name", "ge0/0/0-to-xe0/0/0");
 144
 145         Vertex vserver2 = graph.addVertex(T.label, "vserver", T.id, "26", "aai-node-type",
 146             "vserver", "vserver-id", "vserver-id-2", "vserver-name", "vserver-name-2",
 147             "vserver-selflink", "vserver-selflink-2");
 148         Vertex flavor2 = graph.addVertex(T.label, "flavor", T.id, "27", "aai-node-type", "flavor",
 149             "flavor-id", "flavor-id-2", "flavor-name", "flavor-name-2", "flavor-selflink",
 150             "flavor-selflink-2");
 151         Vertex image2 = graph.addVertex(T.label, "image", T.id, "28", "aai-node-type", "image",
 152             "image-id", "image-id-2", "image-name", "image-name-2", "image-os-distro",
 153             "image-os-distro-2", "image-os-version", "image-os-version-2");
 154
 155         Vertex linter2 =
 156             graph.addVertex(T.label, "l-interface", T.id, "29", "aai-node-type", "l-interface",
 157                 "l-interface-id", "l-interface-id-2", "l-interface-name", "l-interface-name2");
 158         Vertex logicallink2 = graph.addVertex(T.label, "logical-link", T.id, "30", "aai-node-type",
 159             "logical-link", "link-name", "link-name-2", "l-interface-name", "l-interface-name2");
 160         Vertex l3inter1ipv4addresslist2 = graph.addVertex(T.label, "interface-ipv4-address-list",
 161             T.id, "31", "aai-node-type", "l3-interface-ipv4-address-list",
 162             "l3-interface-ipv4-address", "l3-interface-ipv4-address-2");
 163         Vertex subnet3 = graph.addVertex(T.label, "subnet", T.id, "32", "aai-node-type", "subnet",
 164             "subnet-id", "subnet-id-3");
 165         Vertex l3network3 = graph.addVertex(T.label, "l3-network", T.id, "33", "aai-node-type",
 166             "l3-network", "network-id", "network-id-3", "network-name", "network-name-3");
 167         Vertex l3inter1ipv6addresslist2 = graph.addVertex(T.label, "l3-interface-ipv6-address-list",
 168             T.id, "34", "aai-node-type", "l3-interface-ipv6-address-list",
 169             "l3-interface-ipv6-address", "l3-interface-ipv6-address-2");
 170         Vertex subnet4 = graph.addVertex(T.label, "subnet", T.id, "35", "aai-node-type", "subnet",
 171             "subnet-id", "subnet-id-4");
 172         Vertex l3network4 = graph.addVertex(T.label, "l3-network", T.id, "36", "aai-node-type",
 173             "l3-network", "network-id", "network-id-4", "network-name", "network-name4");
 174
 175         GraphTraversalSource g = graph.traversal();
 176         rules.addTreeEdge(g, customer1, servicesubscription1);// true
 177
 178         rules.addTreeEdge(g, customer1, servicesubscription112);// true
 179
 180         rules.addTreeEdge(g, servicesubscription1, serviceinstance1);// true
 181         rules.addTreeEdge(g, servicesubscription1, serviceinstance113);// true
 182
 183         rules.addEdge(g, serviceinstance1, gnvf1);// true
 184         rules.addEdge(g, serviceinstance113, gnvf114);// true
 185
 186         rules.addEdge(g, gnvf1, vserver1);// true
 187         rules.addEdge(g, gnvf114, vserver116);// true
 188
 189         rules.addEdge(g, gnvf1, pserver1);// true
 190         rules.addEdge(g, gnvf114, pserver115);// true
 191
 192         rules.addEdge(g, pserver1, complex1);// true
 193         rules.addTreeEdge(g, pserver1, pnfint1);
 194         rules.addEdge(g, pnfint1, plink1);
 195
 196         rules.addEdge(g, vserver1, flavor1);// true
 197         rules.addEdge(g, vserver1, image1);// true
 198         rules.addEdge(g, vserver1, pserver1);// true
 199         rules.addEdge(g, linter1, logicallink1);// true
 200         rules.addTreeEdge(g, vserver1, linter1);// true
 201         rules.addTreeEdge(g, linter1, l3inter1ipv4addresslist);// true
 202         rules.addTreeEdge(g, linter1, l3inter1ipv6addresslist);// true
 203         rules.addEdge(g, l3inter1ipv4addresslist, subnet1);// true
 204         rules.addTreeEdge(g, subnet1, l3network1);// true
 205         rules.addEdge(g, l3inter1ipv6addresslist, subnet2);// true
 206         rules.addTreeEdge(g, subnet2, l3network2);// true
 207         // false
 208
 209         rules.addTreeEdge(g, customer2, servicesubscription2);// true
 210         rules.addTreeEdge(g, servicesubscription2, serviceinstance2);// true
 211         rules.addEdge(g, serviceinstance2, gnvf2);// true
 212         rules.addEdge(g, gnvf2, pserver2);// true
 213         rules.addEdge(g, pserver2, complex2);// true
 214         rules.addTreeEdge(g, pserver2, pnfint2);// true
 215
 216         rules.addEdge(g, gnvf2, vserver2);// true
 217
 218         rules.addEdge(g, vserver2, flavor2);// true
 219         rules.addEdge(g, vserver2, image2);// true
 220         rules.addEdge(g, linter2, logicallink2);// true
 221         rules.addTreeEdge(g, vserver2, linter2);// true
 222         rules.addTreeEdge(g, linter2, l3inter1ipv4addresslist2);// true
 223         rules.addTreeEdge(g, linter2, l3inter1ipv6addresslist2);// true
 224         rules.addEdge(g, l3inter1ipv4addresslist2, subnet3);// true
 225         rules.addTreeEdge(g, subnet3, l3network3);// true
 226         rules.addEdge(g, l3inter1ipv6addresslist2, subnet4);// true
 227         rules.addTreeEdge(g, subnet4, l3network4);// true
 228
 229         expectedResult.add(complex1);
 230         expectedResult.add(vserver1);
 231         expectedResult.add(vserver116);
 232         expectedResult.add(pserver1);
 233         expectedResult.add(flavor1);
 234         expectedResult.add(image1);
 235         expectedResult.add(linter1);
 236         expectedResult.add(logicallink1);
 237         expectedResult.add(plink1);
 238         expectedResult.add(l3inter1ipv4addresslist);
 239         expectedResult.add(subnet1);
 240         expectedResult.add(l3network1);
 241         expectedResult.add(l3inter1ipv6addresslist);
 242         expectedResult.add(subnet2);
 243         expectedResult.add(l3network2);
 244
 245     }
 246
 247     @Override
 248     protected String getQueryName() {
 249         return "spaas-topology-fromServiceInstance";
 250     }
 251
 252     @Override
 253     protected void addStartNode(GraphTraversal<Vertex, Vertex> g) {
 254         g.has("aai-node-type", "customer").has("global-customer-id", "global-customer-id-1")
 255             .in("org.onap.relationships.inventory.BelongsTo")
 256             .has("aai-node-type", "service-subscription").has("service-type", "service-type-1")
 257             .in("org.onap.relationships.inventory.BelongsTo")
 258             .has("aai-node-type", "service-instance");
 259     }
 260
 261     @Override
 262     protected void addParam(Map<String, Object> params) {
 263         // params.put("serviceType", "service-type-1");
 264         // params.put("subscriberName", "subscriber-name-1");
 265     }
 266 }