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

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