aai-core/src/test/java/org/onap/aai/parsers/query/RelationshipQueryTest.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.parsers.query;
  21
  22 import org.eclipse.persistence.dynamic.DynamicEntity;
  23 import org.eclipse.persistence.jaxb.UnmarshallerProperties;
  24 import org.eclipse.persistence.jaxb.dynamic.DynamicJAXBContext;
  25 import org.junit.Before;
  26 import org.junit.Ignore;
  27 import org.junit.Test;
  28 import org.onap.aai.AAISetup;
  29 import org.onap.aai.exceptions.AAIException;
  30 import org.onap.aai.introspection.*;
  31 import org.onap.aai.setup.SchemaVersion;
  32 import org.onap.aai.serialization.engines.QueryStyle;
  33 import org.onap.aai.serialization.engines.JanusGraphDBEngine;
  34 import org.onap.aai.serialization.engines.TransactionalGraphEngine;
  35
  36 import javax.xml.bind.JAXBException;
  37 import javax.xml.bind.Unmarshaller;
  38 import javax.xml.transform.stream.StreamSource;
  39 import java.io.StringReader;
  40 import java.io.UnsupportedEncodingException;
  41
  42 import static org.junit.Assert.assertEquals;
  43
  44 @Ignore
  45 public class RelationshipQueryTest extends AAISetup {
  46
  47
  48         private TransactionalGraphEngine dbEngine;
  49         private SchemaVersion version;
  50         private DynamicJAXBContext context = nodeIngestor.getContextForVersion(version);
  51
  52         @Before
  53         public void setup(){
  54             version = new SchemaVersion("v10");
  55                 dbEngine =
  56                         new JanusGraphDBEngine(QueryStyle.GREMLIN_TRAVERSAL,
  57                                 loaderFactory.createLoaderForVersion(ModelType.MOXY, version),
  58                                 false);
  59         }
  60         /**
  61          * Parent query.
  62          *
  63          * @throws JAXBException the JAXB exception
  64          * @throws UnsupportedEncodingException the unsupported encoding exception
  65          * @throws AAIException the AAI exception
  66          */
  67         @Test
  68     public void parentQuery() throws JAXBException, UnsupportedEncodingException, AAIException {
  69
  70                 String content =
  71                                 "{"
  72                                 + "\"related-to\" : \"pserver\","
  73                                 + "\"relationship-data\" : [{"
  74                                 + "\"relationship-key\" : \"pserver.hostname\","
  75                                 + "\"relationship-value\" : \"key1\""
  76                                 + "}]"
  77                                 + "}";
  78
  79                 Unmarshaller unmarshaller = context.createUnmarshaller();
  80             unmarshaller.setProperty(UnmarshallerProperties.MEDIA_TYPE, "application/json");
  81             unmarshaller.setProperty(UnmarshallerProperties.JSON_INCLUDE_ROOT, false);
  82                 unmarshaller.setProperty(UnmarshallerProperties.JSON_WRAPPER_AS_ARRAY_NAME, true);
  83                 Object obj = context.newDynamicEntity("Relationship");
  84
  85                 DynamicEntity entity = (DynamicEntity)unmarshaller.unmarshal(new StreamSource(new StringReader(content)), obj.getClass()).getValue();
  86
  87                 Introspector wrappedObj = IntrospectorFactory.newInstance(ModelType.MOXY, entity);
  88                 QueryParser query = dbEngine.getQueryBuilder().createQueryFromRelationship(wrappedObj);
  89
  90                 String expected =
  91                                 ".has('hostname', 'key1').has('aai-node-type', 'pserver')";
  92                 assertEquals(
  93                                 "gremlin query should be " + expected,
  94                                 expected,
  95                                 query.getQueryBuilder().getQuery());
  96                 assertEquals(
  97                                 "parent gremlin query should be equal to normal query",
  98                                 expected,
  99                                 query.getQueryBuilder().getParentQuery().getQuery());
 100                 assertEquals(
 101                                 "result type should be pserver",
 102                                 "pserver",
 103                                 query.getResultType());
 104
 105     }
 106
 107         /**
 108          * Child query.
 109          *
 110          * @throws JAXBException the JAXB exception
 111          * @throws UnsupportedEncodingException the unsupported encoding exception
 112          * @throws AAIException the AAI exception
 113          */
 114         @Ignore
 115         @Test
 116     public void childQuery() throws JAXBException, UnsupportedEncodingException, AAIException {
 117                 String content =
 118                                 "{"
 119                                 + "\"related-to\" : \"lag-interface\","
 120                                 + "\"relationship-data\" : [{"
 121                                 + "\"relationship-key\" : \"pserver.hostname\","
 122                                 + "\"relationship-value\" : \"key1\""
 123                                 + "}, {"
 124                                 + "\"relationship-key\" : \"lag-interface.interface-name\","
 125                                 + "\"relationship-value\" : \"key2\""
 126                                 + "}]"
 127                                 + "}";
 128
 129                 Unmarshaller unmarshaller = context.createUnmarshaller();
 130             unmarshaller.setProperty(UnmarshallerProperties.MEDIA_TYPE, "application/json");
 131             unmarshaller.setProperty(UnmarshallerProperties.JSON_INCLUDE_ROOT, false);
 132                 unmarshaller.setProperty(UnmarshallerProperties.JSON_WRAPPER_AS_ARRAY_NAME, true);
 133                 Object obj = context.newDynamicEntity("Relationship");
 134
 135                 DynamicEntity entity = (DynamicEntity)unmarshaller.unmarshal(new StreamSource(new StringReader(content)), obj.getClass()).getValue();
 136
 137                 Introspector wrappedObj = IntrospectorFactory.newInstance(ModelType.MOXY, entity);
 138                 QueryParser query = dbEngine.getQueryBuilder().createQueryFromRelationship(wrappedObj);
 139                 String expected =
 140                                 ".has('hostname', 'key1').has('aai-node-type', 'pserver').out('hasLAGInterface').has('aai-node-type', 'lag-interface')"
 141                                 + ".has('interface-name', 'key2')";
 142                 String parentExpected =
 143                                 ".has('hostname', 'key1').has('aai-node-type', 'pserver')";
 144                 assertEquals(
 145                                 "gremlin query should be for node",
 146                                 expected,
 147                                 query.getQueryBuilder().getQuery());
 148                 assertEquals(
 149                                 "parent gremlin query should be for parent",
 150                                 parentExpected,
 151                                 query.getQueryBuilder().getParentQuery().getQuery());
 152                 assertEquals(
 153                                 "result type should be lag-interface",
 154                                 "lag-interface",
 155                                 query.getResultType());
 156     }
 157
 158         /**
 159          * Naming exceptions.
 160          *
 161          * @throws JAXBException the JAXB exception
 162          * @throws UnsupportedEncodingException the unsupported encoding exception
 163          * @throws AAIException the AAI exception
 164          */
 165         @Ignore
 166         @Test
 167     public void namingExceptions() throws JAXBException, UnsupportedEncodingException, AAIException {
 168                 String content =
 169                                 "{"
 170                                 + "\"related-to\" : \"cvlan-tag\","
 171                                 + "\"relationship-data\" : [{"
 172                                 + "\"relationship-key\" : \"vce.vnf-id\","
 173                                 + "\"relationship-value\" : \"key1\""
 174                                 + "}, {"
 175                                 + "\"relationship-key\" : \"port-group.interface-id\","
 176                                 + "\"relationship-value\" : \"key2\""
 177                                 + "},{"
 178                                 + "\"relationship-key\" : \"cvlan-tag.cvlan-tag\","
 179                                 + "\"relationship-value\" : \"655\""
 180                                 + "}]"
 181                                 + "}";
 182
 183                 Unmarshaller unmarshaller = context.createUnmarshaller();
 184             unmarshaller.setProperty(UnmarshallerProperties.MEDIA_TYPE, "application/json");
 185             unmarshaller.setProperty(UnmarshallerProperties.JSON_INCLUDE_ROOT, false);
 186                 unmarshaller.setProperty(UnmarshallerProperties.JSON_WRAPPER_AS_ARRAY_NAME, true);
 187                 Object obj = context.newDynamicEntity("Relationship");
 188
 189                 DynamicEntity entity = (DynamicEntity)unmarshaller.unmarshal(new StreamSource(new StringReader(content)), obj.getClass()).getValue();
 190
 191                 Introspector wrappedObj = IntrospectorFactory.newInstance(ModelType.MOXY, entity);
 192                 QueryParser query = dbEngine.getQueryBuilder().createQueryFromRelationship(wrappedObj);
 193                 String expected =
 194                                 ".has('vnf-id', 'key1').has('aai-node-type', 'vce').in('org.onap.relationships.inventory.BelongsTo')"
 195                                 + ".has('aai-node-type', 'port-group').has('interface-id', 'key2').in('org.onap.relationships.inventory.BelongsTo').has('aai-node-type', 'cvlan-tag')"
 196                                 + ".has('cvlan-tag', 655)";
 197                 String expectedParent =
 198                                                 ".has('vnf-id', 'key1').has('aai-node-type', 'vce').in('org.onap.relationships.inventory.BelongsTo')"
 199                                                 + ".has('aai-node-type', 'port-group').has('interface-id', 'key2')";
 200
 201                 assertEquals(
 202                                 "gremlin query should be " + expected,
 203                                 expected,
 204                                 query.getQueryBuilder().getQuery());
 205                 assertEquals(
 206                                 "parent gremlin query should be equal the query for port group",
 207                                 expectedParent,
 208                                 query.getQueryBuilder().getParentQuery().getQuery());
 209                 assertEquals(
 210                                 "result type should be cvlan-tag",
 211                                 "cvlan-tag",
 212                                 query.getResultType());
 213
 214     }
 215
 216         /**
 217          * Double key.
 218          *
 219          * @throws JAXBException the JAXB exception
 220          * @throws UnsupportedEncodingException the unsupported encoding exception
 221          * @throws AAIException the AAI exception
 222          */
 223         @Ignore
 224         @Test
 225     public void doubleKey() throws JAXBException, UnsupportedEncodingException, AAIException {
 226                 String content =
 227                                 "{"
 228                                 + "\"related-to\" : \"ctag-pool\","
 229                                 + "\"relationship-data\" : [{"
 230                                 + "\"relationship-key\" : \"complex.physical-location-id\","
 231                                 + "\"relationship-value\" : \"key1\""
 232                                 + " }, { "
 233                                 + "\"relationship-key\" : \"ctag-pool.target-pe\","
 234                                 + " \"relationship-value\" : \"key2\""
 235                                 + " },{"
 236                                 + "\"relationship-key\" : \"ctag-pool.availability-zone-name\","
 237                                 + "\"relationship-value\" : \"key3\""
 238                                 + "}]"
 239                                 + "}";
 240
 241
 242                 Unmarshaller unmarshaller = context.createUnmarshaller();
 243             unmarshaller.setProperty(UnmarshallerProperties.MEDIA_TYPE, "application/json");
 244             unmarshaller.setProperty(UnmarshallerProperties.JSON_INCLUDE_ROOT, false);
 245                 unmarshaller.setProperty(UnmarshallerProperties.JSON_WRAPPER_AS_ARRAY_NAME, true);
 246                 Object obj = context.newDynamicEntity("Relationship");
 247
 248                 DynamicEntity entity = (DynamicEntity)unmarshaller.unmarshal(new StreamSource(new StringReader(content)), obj.getClass()).getValue();
 249
 250                 Introspector wrappedObj = IntrospectorFactory.newInstance(ModelType.MOXY, entity);
 251                 QueryParser query = dbEngine.getQueryBuilder().createQueryFromRelationship(wrappedObj);
 252
 253                 String expected =
 254                                 ".has('physical-location-id', 'key1').has('aai-node-type', 'complex')"
 255                                 + ".in('org.onap.relationships.inventory.BelongsTo').has('aai-node-type', 'ctag-pool')"
 256                                 + ".has('target-pe', 'key2')"
 257                                 + ".has('availability-zone-name', 'key3')";
 258                 String expectedParent =
 259                                 ".has('physical-location-id', 'key1').has('aai-node-type', 'complex')";
 260
 261                 assertEquals(
 262                                 "gremlin query should be " + expected,
 263                                 expected,
 264                                 query.getQueryBuilder().getQuery());
 265                 assertEquals(
 266                                 "parent gremlin query should be equal the query for port group",
 267                                 expectedParent,
 268                                 query.getQueryBuilder().getParentQuery().getQuery());
 269                 assertEquals(
 270                                 "result type should be ctag-pool",
 271                                 "ctag-pool",
 272                                 query.getResultType());
 273
 274     }
 275
 276 }