2 * ============LICENSE_START=======================================================
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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
20 package org.onap.aai.rest.search;
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 import org.onap.aai.exceptions.AAIException;
30 import org.onap.aai.serialization.db.exceptions.NoEdgeRuleFoundException;
32 public class CloudRegionFromNfTypeVendorVersionTest extends QueryTest {
34 public CloudRegionFromNfTypeVendorVersionTest() throws AAIException, NoEdgeRuleFoundException {
44 protected void createGraph() throws AAIException, NoEdgeRuleFoundException {
45 Vertex image1 = graph.addVertex(T.label, "image", T.id, "10", "aai-node-type", "image",
46 "image-id", "image1", "image-name", "imageName1", "image-os-distro", "boopOS", "image-os-version", "1.0", "image-selflink", "self/link",
47 "application-vendor","vendor1");
49 Vertex vserver1 = graph.addVertex(T.label, "vserver", T.id, "20", "aai-node-type", "vserver", "vserver-id", "vserverid01");
50 Vertex vserver2 = graph.addVertex(T.label, "vserver", T.id, "21", "aai-node-type", "vserver", "vserver-id", "vserverid02");
51 Vertex vserver3 = graph.addVertex(T.label, "vserver", T.id, "22", "aai-node-type", "vserver", "vserver-id", "vserverid03");
53 Vertex tenant1 = graph.addVertex(T.label, "tenant", T.id, "30", "aai-node-type", "tenant", "tenant-id", "tenantid01", "tenant-name", "tenantName01");
54 Vertex tenant2 = graph.addVertex(T.label, "tenant", T.id, "31", "aai-node-type", "tenant", "tenant-id", "tenantid02", "tenant-name", "tenantName02");
55 Vertex tenant3 = graph.addVertex(T.label, "tenant", T.id, "32", "aai-node-type", "tenant", "tenant-id", "tenantid03", "tenant-name", "tenantName03");
57 Vertex genericvnf1 = graph.addVertex(T.label, "generic-vnf", T.id, "40", "aai-node-type", "generic-vnf", "vnf-id", "genvnf1", "vnf-name", "genvnfname1", "nf-type", "sample-nf-type");
58 Vertex genericvnf2 = graph.addVertex(T.label, "generic-vnf", T.id, "41", "aai-node-type", "generic-vnf", "vnf-id", "genvnf2", "vnf-name", "genvnfname2", "nf-type", "wrong-nf-type");
60 Vertex cloudregion1 = graph.addVertex(T.label, "cloud-region", T.id, "50", "aai-node-type", "cloud-region", "cloud-region-id", "cloudreg1", "cloud-region-owner", "cloudOwnername00");
61 Vertex cloudregion2 = graph.addVertex(T.label, "cloud-region", T.id, "51", "aai-node-type", "cloud-region", "cloud-region-id", "cloudreg2", "cloud-region-owner", "cloudOwnername01");
62 Vertex cloudregion3 = graph.addVertex(T.label, "cloud-region", T.id, "52", "aai-node-type", "cloud-region", "cloud-region-id", "cloudreg3", "cloud-region-owner", "cloudOwnername02");
64 GraphTraversalSource g = graph.traversal();
66 rules.addTreeEdge(g, cloudregion1, tenant1);
67 rules.addTreeEdge(g, cloudregion2, tenant2);
68 rules.addTreeEdge(g, cloudregion3, tenant3);
69 rules.addTreeEdge(g, tenant1, vserver1);
70 rules.addTreeEdge(g, tenant2, vserver2);
71 rules.addTreeEdge(g, tenant3, vserver3);
72 rules.addEdge(g, genericvnf1, vserver1);
73 rules.addEdge(g, genericvnf2, vserver2);
74 rules.addEdge(g, genericvnf1, vserver3);
75 rules.addEdge(g, vserver1, image1);
76 rules.addEdge(g, vserver2, image1);
77 rules.addEdge(g, vserver3, image1);
79 expectedResult.add(cloudregion1);
80 expectedResult.add(cloudregion3);
84 protected String getQueryName() {
85 return "cloudRegion-fromNfTypeVendorVersion";
89 protected void addStartNode(GraphTraversal<Vertex, Vertex> g) {
90 g.has("aai-node-type","image").has("application-vendor","vendor1"); //TODO another test w the optional param too
94 protected void addParam(Map<String, Object> params) {
95 params.put("nfType", "sample-nf-type");