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