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