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