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 ImageFromCloudRegionNfTypeTest extends QueryTest {
34 public ImageFromCloudRegionNfTypeTest() throws AAIException, NoEdgeRuleFoundException {
44 protected void createGraph() throws AAIException, NoEdgeRuleFoundException {
45 Vertex cloudregion1 = graph.addVertex(T.label, "cloud-region", T.id, "10", "aai-node-type", "cloud-region", "cloud-region-id", "cloudreg1", "cloud-region-owner", "cloudOwnername00");
46 Vertex cloudregion2 = graph.addVertex(T.label, "cloud-region", T.id, "11", "aai-node-type", "cloud-region", "cloud-region-id", "cloudregWrong", "cloud-region-owner", "cloudOwnername01");
48 Vertex genericvnf1 = graph.addVertex(T.label, "generic-vnf", T.id, "20", "aai-node-type", "generic-vnf", "vnf-id", "genvnf1", "vnf-name", "genvnfname1", "nf-type", "sample-nf-type");
49 Vertex genericvnf2 = graph.addVertex(T.label, "generic-vnf", T.id, "21", "aai-node-type", "generic-vnf", "vnf-id", "genvnf2", "vnf-name", "genvnfname2", "nf-type", "wrong-nf-type");
50 Vertex genericvnf3 = graph.addVertex(T.label, "generic-vnf", T.id, "22", "aai-node-type", "generic-vnf", "vnf-id", "genvnf3", "vnf-name", "genvnfname3", "nf-type", "sample-nf-type");
52 Vertex tenant1 = graph.addVertex(T.label, "tenant", T.id, "30", "aai-node-type", "tenant", "tenant-id", "tenantid01", "tenant-name", "tenantName01");
53 Vertex tenant2 = graph.addVertex(T.label, "tenant", T.id, "31", "aai-node-type", "tenant", "tenant-id", "tenantid02", "tenant-name", "tenantName02");
55 Vertex vserver1 = graph.addVertex(T.label, "vserver", T.id, "40", "aai-node-type", "vserver", "vserver-id", "vserverid01");
56 Vertex vserver2 = graph.addVertex(T.label, "vserver", T.id, "41", "aai-node-type", "vserver", "vserver-id", "vserverid02");
57 Vertex vserver3 = graph.addVertex(T.label, "vserver", T.id, "42", "aai-node-type", "vserver", "vserver-id", "vserverid03");
58 Vertex vserver4 = graph.addVertex(T.label, "vserver", T.id, "43", "aai-node-type", "vserver", "vserver-id", "vserverid04");
59 Vertex vserver5 = graph.addVertex(T.label, "vserver", T.id, "44", "aai-node-type", "vserver", "vserver-id", "vserverid05");
61 Vertex image1 = graph.addVertex(T.label, "image", T.id, "50", "aai-node-type", "image",
62 "image-id", "image1", "image-name", "imageName1", "image-os-distro", "boopOS", "image-os-version", "1.0", "image-selflink", "self/link");
63 Vertex image2 = graph.addVertex(T.label, "image", T.id, "51", "aai-node-type", "image",
64 "image-id", "image2", "image-name", "imageName2", "image-os-distro", "beepOS", "image-os-version", "1.0", "image-selflink", "self/link");
65 Vertex image3 = graph.addVertex(T.label, "image", T.id, "52", "aai-node-type", "image",
66 "image-id", "image3", "image-name", "imageName3", "image-os-distro", "beepOS", "image-os-version", "1.1", "image-selflink", "self/link");
67 Vertex image4 = graph.addVertex(T.label, "image", T.id, "53", "aai-node-type", "image",
68 "image-id", "image4", "image-name", "imageName4", "image-os-distro", "beepOS", "image-os-version", "1.1", "image-selflink", "self/link");
70 GraphTraversalSource g = graph.traversal();
72 rules.addTreeEdge(g, cloudregion1, tenant1);
73 rules.addTreeEdge(g, cloudregion2, tenant2);
74 rules.addTreeEdge(g, tenant1, vserver1);
75 rules.addTreeEdge(g, tenant1, vserver2);
76 rules.addTreeEdge(g, tenant2, vserver3);
77 rules.addTreeEdge(g, tenant1, vserver4);
78 rules.addTreeEdge(g, tenant1, vserver5);
79 rules.addEdge(g, vserver1, image1);
80 rules.addEdge(g, vserver2, image2);
81 rules.addEdge(g, vserver3, image3);
82 rules.addEdge(g, vserver4, image4);
83 rules.addEdge(g, vserver5, image1);
84 rules.addEdge(g, genericvnf1, vserver1);
85 rules.addEdge(g, genericvnf2, vserver2);
86 rules.addEdge(g, genericvnf1, vserver3);
87 rules.addEdge(g, genericvnf3, vserver4);
89 expectedResult.add(image1);
90 expectedResult.add(image4);
94 protected String getQueryName() {
95 return "images-fromCloudRegionNfType";
99 protected void addStartNode(GraphTraversal<Vertex, Vertex> g) {
100 g.has("aai-node-type","cloud-region").has("cloud-region-id","cloudreg1");
104 protected void addParam(Map<String, Object> params) {
105 params.put("nfType", "sample-nf-type");