Update the aai-common with the latest code
[aai/aai-common.git] / aai-core / src / test / java / org / openecomp / aai / parsers / query / LegacyQueryTest.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * org.openecomp.aai
4  * ================================================================================
5  * Copyright (C) 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
21 package org.openecomp.aai.parsers.query;
22
23 import static org.junit.Assert.assertEquals;
24
25 import java.io.UnsupportedEncodingException;
26 import java.net.URI;
27
28 import javax.ws.rs.core.UriBuilder;
29 import javax.xml.bind.JAXBException;
30
31 import org.eclipse.persistence.jaxb.dynamic.DynamicJAXBContext;
32 import org.junit.BeforeClass;
33 import org.junit.Ignore;
34 import org.junit.Test;
35
36 import org.openecomp.aai.exceptions.AAIException;
37 import org.openecomp.aai.introspection.LoaderFactory;
38 import org.openecomp.aai.introspection.ModelInjestor;
39 import org.openecomp.aai.introspection.ModelType;
40 import org.openecomp.aai.introspection.Version;
41 import org.openecomp.aai.serialization.engines.QueryStyle;
42 import org.openecomp.aai.serialization.engines.TitanDBEngine;
43 import org.openecomp.aai.serialization.engines.TransactionalGraphEngine;
44
45
46 public class LegacyQueryTest {
47         private ModelInjestor injestor = ModelInjestor.getInstance();
48         private TransactionalGraphEngine dbEngine =
49                         new TitanDBEngine(QueryStyle.GREMLIN_TRAVERSAL,
50                                 LoaderFactory.createLoaderForVersion(ModelType.MOXY, Version.v8),
51                                 false); 
52         private final Version version = Version.v8;
53         private DynamicJAXBContext context = injestor.getContextForVersion(version);
54         
55         
56         /**
57          * Configure.
58          */
59         @BeforeClass
60         public static void configure() {
61                 System.setProperty("AJSC_HOME", ".");
62                 System.setProperty("BUNDLECONFIG_DIR", "src/test/resources/bundleconfig-local");
63         }
64         
65         /**
66          * Parent query.
67          *
68          * @throws JAXBException the JAXB exception
69          * @throws UnsupportedEncodingException the unsupported encoding exception
70          * @throws AAIException the AAI exception
71          */
72         @Test
73     public void parentQuery() throws JAXBException, UnsupportedEncodingException, AAIException {
74                 
75                 URI uri = UriBuilder.fromPath("cloud-infrastructure/pservers/pserver/key1").build();
76
77                 QueryParser query = dbEngine.getQueryBuilder().createQueryFromURI(uri);
78
79                 String expected = 
80                                 ".has('hostname', 'key1').has('aai-node-type', 'pserver')";
81                 assertEquals(
82                                 "gremlin query should be " + expected,
83                                 expected,
84                                 query.getQueryBuilder().getQuery());
85                 assertEquals(
86                                 "parent gremlin query should be equal to normal query",
87                                 expected,
88                                 query.getQueryBuilder().getParentQuery().getQuery());
89                 assertEquals(
90                                 "result type should be pserver",
91                                 "pserver",
92                                 query.getResultType());
93                 
94     }
95
96         /**
97          * Child query.
98          *
99          * @throws JAXBException the JAXB exception
100          * @throws UnsupportedEncodingException the unsupported encoding exception
101          * @throws AAIException the AAI exception
102          */
103         @Test
104     public void childQuery() throws JAXBException, UnsupportedEncodingException, AAIException {
105                 URI uri = UriBuilder.fromPath("cloud-infrastructure/pservers/pserver/key1/lag-interfaces/lag-interface/key2").build();
106                 QueryParser query = dbEngine.getQueryBuilder().createQueryFromURI(uri);
107
108                 String expected =
109                                 ".has('hostname', 'key1').has('aai-node-type', 'pserver')"
110                                 + ".out('hasLAGInterface').has('aai-node-type', 'lag-interface')"
111                                 + ".has('interface-name', 'key2')";
112                 String parentExpected = 
113                                 ".has('hostname', 'key1').has('aai-node-type', 'pserver')";
114                 assertEquals(
115                                 "gremlin query should be for node",
116                                 expected,
117                                 query.getQueryBuilder().getQuery());
118                 assertEquals(
119                                 "parent gremlin query should be for parent",
120                                 parentExpected,
121                                 query.getQueryBuilder().getParentQuery().getQuery());
122                 assertEquals(
123                                 "result type should be lag-interface",
124                                 "lag-interface",
125                                 query.getResultType());
126     }
127         
128         /**
129          * Naming exceptions.
130          *
131          * @throws JAXBException the JAXB exception
132          * @throws UnsupportedEncodingException the unsupported encoding exception
133          * @throws AAIException the AAI exception
134          */
135         @Ignore
136         @Test
137     public void namingExceptions() throws JAXBException, UnsupportedEncodingException, AAIException {
138                 URI uri = UriBuilder.fromPath("network/vces/vce/key1/port-groups/port-group/key2/cvlan-tags/cvlan-tag/655").build();
139         
140                 QueryParser query = dbEngine.getQueryBuilder().createQueryFromURI(uri);
141                 String expected = 
142                                 ".has('vnf-id', 'key1').has('aai-node-type', 'vce')"
143                                 + ".in('org.onap.relationships.inventory.BelongsTo').has('aai-node-type', 'port-group')"
144                                 + ".has('interface-id', 'key2')"
145                                 + ".in('org.onap.relationships.inventory.BelongsTo').has('aai-node-type', 'cvlan-tag')"
146                                 + ".has('cvlan-tag', 655)";
147                 String expectedParent = 
148                                                 ".has('vnf-id', 'key1').has('aai-node-type', 'vce')"
149                                                 + ".in('org.onap.relationships.inventory.BelongsTo').has('aai-node-type', 'port-group')"
150                                                 + ".has('interface-id', 'key2')";
151                 assertEquals(
152                                 "gremlin query should be " + expected,
153                                 expected,
154                                 query.getQueryBuilder().getQuery());
155                 assertEquals(
156                                 "parent gremlin query should be equal the query for port group",
157                                 expectedParent,
158                                 query.getQueryBuilder().getParentQuery().getQuery());
159                 assertEquals(
160                                 "result type should be cvlan-tag",
161                                 "cvlan-tag",
162                                 query.getResultType());
163                 
164     }
165         
166 }