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