Update traversal from AJSC 2 to Spring Boot
[aai/traversal.git] / aai-traversal / src / test / java / org / onap / aai / rest / search / TopologyDetailFromVserverQueryTest.java
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.structure.T;
29 import org.apache.tinkerpop.gremlin.structure.Vertex;
30 import org.junit.Test;
31
32 import org.onap.aai.exceptions.AAIException;
33 import org.onap.aai.serialization.db.exceptions.NoEdgeRuleFoundException;
34
35 public class TopologyDetailFromVserverQueryTest extends QueryTest {
36         public TopologyDetailFromVserverQueryTest() throws AAIException, NoEdgeRuleFoundException {
37                 super();
38         }
39         @Test
40         public void run() {
41                 super.run();
42         }
43         @Override
44         protected void createGraph() throws AAIException, NoEdgeRuleFoundException {
45                 //set up test graph
46                 Vertex genericvnf = graph.addVertex(T.label, "generic-vnf", T.id, "0", "aai-node-type", "generic-vnf", "vnf-id", "vnfid0", "nf-type", "sample-nf-type");
47                 Vertex cloudregion = graph.addVertex(T.label, "cloud-region", T.id, "1", "aai-node-type", "cloud-region", "cloud-region-id", "regionid0", "cloud-owner", "cloudOwnername0");
48                 Vertex tenant = graph.addVertex(T.label, "tenant", T.id, "2", "aai-node-type", "tenant", "tenant-id", "tenantid0", "tenant-name", "tenantName0");
49                 Vertex vserver = graph.addVertex(T.label, "vserver", T.id, "3", "aai-node-type", "vserver", "vserver-id", "vserverid0");
50                 Vertex linterface = graph.addVertex(T.label, "l-interface", T.id, "4", "aai-node-type", "l-interface", "l-interface-id", "l-interface-id0", "l-interface-name", "l-interface-name0");
51                 Vertex l3inter1ipv4addresslist = graph.addVertex(T.label, "interface-ipv4-address-list", T.id, "5", "aai-node-type", "l3-interface-ipv4-address-list", "l3-interface-ipv4-address", "l3-interface-ipv4-address-0");
52                 Vertex subnet4 = graph.addVertex(T.label, "subnet", T.id, "6", "aai-node-type", "subnet", "subnet-id", "subnet4-id-0");
53                 Vertex l3network4 = graph.addVertex(T.label, "l3-network", T.id, "7", "aai-node-type", "l3-network", "network-id", "network4-id-0", "network-name", "network4-name0");          
54                 Vertex l3inter1ipv6addresslist = graph.addVertex(T.label, "l3-interface-ipv6-address-list", T.id, "8", "aai-node-type", "l3-interface-ipv6-address-list", "l3-interface-ipv6-address", "l3-interface-ipv6-address-0");
55                 Vertex subnet6 = graph.addVertex(T.label, "subnet", T.id, "9", "aai-node-type", "subnet", "subnet-id", "subnet6-id-0"); 
56                 Vertex l3network6 = graph.addVertex(T.label, "l3-network", T.id, "10", "aai-node-type", "l3-network", "network-id", "network6-id-0", "network-name", "network6-name0");
57                 Vertex platform = graph.addVertex(T.label, "platform", T.id, "11", "aai-node-type", "platform", "platform-name", "platform0");
58                 Vertex lineofbusiness = graph.addVertex(T.label, "line-of-business", T.id, "12", "aai-node-type", "line-of-business", "line-of-business-name", "business0");
59                 Vertex servinst = graph.addVertex(T.label, "service-instance", T.id, "13", "aai-node-type", "service-instance", "service-instance-id", "servInstId0", "service-type", "servType0");
60                 Vertex owningentity = graph.addVertex(T.label, "owning-entity", T.id, "14", "aai-node-type", "owning-entity", "owning-entity-id", "entityId0", "owning-entity-name", "entityName0");
61                 Vertex project = graph.addVertex(T.label, "project", T.id, "15", "aai-node-type", "project", "project-name", "project0");
62                 Vertex vnfc = graph.addVertex(T.label, "vnfc", T.id, "16", "aai-node-type", "vnfc", "vnfc-name", "vnfc0", "nfc-naming-code", "namingCode0", "nfc-function", "function0");
63                 Vertex pserver = graph.addVertex(T.label, "pserver", T.id, "17", "aai-node-type", "pserver", "hostname", "pservername1");
64                 Vertex complex = graph.addVertex(T.label, "pserver", T.id, "18", "aai-node-type", "complex", "physical-location-id", "locationId", "physical-location-type", "locationType", "physical-location-id", "locationId", 
65                                 "city", "cityName", "state", "stateName", "postal-code", "zip", "country", "countryName");
66                 
67                 GraphTraversalSource g = graph.traversal();
68         
69                 rules.addTreeEdge(g, tenant, cloudregion);
70                 rules.addTreeEdge(g, vserver, tenant);
71                 rules.addEdge(g, pserver, vserver);
72                 rules.addEdge(g, complex, pserver);
73                 rules.addEdge(g, genericvnf, vserver);
74                 rules.addEdge(g, genericvnf, platform);
75                 rules.addEdge(g, genericvnf, lineofbusiness);
76                 rules.addEdge(g, genericvnf, vnfc);
77                 rules.addEdge(g, genericvnf, servinst);
78                 rules.addEdge(g, owningentity, servinst);
79                 rules.addEdge(g, project, servinst);
80                 rules.addTreeEdge(g, linterface, vserver);
81                 rules.addTreeEdge(g, l3inter1ipv4addresslist, linterface);
82         rules.addEdge(g, l3inter1ipv4addresslist, subnet4);
83         rules.addTreeEdge(g, l3network4, subnet4);
84         rules.addTreeEdge(g, l3inter1ipv6addresslist, linterface);
85         rules.addEdge(g, l3inter1ipv6addresslist, subnet6);
86         rules.addTreeEdge(g, l3network6, subnet6);
87                                 
88                 expectedResult.add(vserver);
89                 expectedResult.add(linterface);
90                 expectedResult.add(l3inter1ipv4addresslist);
91                 expectedResult.add(subnet4);
92                 expectedResult.add(l3network4);
93                 expectedResult.add(l3inter1ipv6addresslist);
94                 expectedResult.add(subnet6);
95                 expectedResult.add(l3network6);
96                 expectedResult.add(cloudregion);
97                 expectedResult.add(complex);
98                 expectedResult.add(genericvnf);
99                 expectedResult.add(platform);
100                 expectedResult.add(lineofbusiness);
101                 expectedResult.add(owningentity);
102                 expectedResult.add(project);
103                 expectedResult.add(vnfc);
104         }
105         @Override
106         protected String getQueryName() {
107                 return  "topology-detail-fromVserver";
108         }
109         @Override
110         protected void addStartNode(GraphTraversal<Vertex, Vertex> g) {
111                 g.has("aai-node-type", "cloud-region")
112                 .has("cloud-region-id", "regionid0")
113                 .has("cloud-owner", "cloudOwnername0")
114                 .in("org.onap.relationships.inventory.BelongsTo")
115                 .has("aai-node-type","tenant")
116                 .has("tenant-id", "tenantid0")
117                 .in("org.onap.relationships.inventory.BelongsTo")
118                 .has("aai-node-type","vserver")
119                 .has("vserver-id", "vserverid0");
120         }
121
122         @Override
123         protected void addParam(Map<String, Object> params) {
124                 return;
125         }
126 }