aai-core/src/test/java/org/openecomp/aai/parsers/query/UniqueRelationshipQueryTest.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 org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversal;
  24 import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.__;
  25 import org.apache.tinkerpop.gremlin.structure.Vertex;
  26 import org.eclipse.persistence.dynamic.DynamicEntity;
  27 import org.eclipse.persistence.jaxb.UnmarshallerProperties;
  28 import org.eclipse.persistence.jaxb.dynamic.DynamicJAXBContext;
  29 import org.junit.Before;
  30 import org.junit.BeforeClass;
  31 import org.junit.Ignore;
  32 import org.junit.Test;
  33 import org.openecomp.aai.exceptions.AAIException;
  34 import org.openecomp.aai.introspection.*;
  35 import org.openecomp.aai.serialization.engines.QueryStyle;
  36 import org.openecomp.aai.serialization.engines.TitanDBEngine;
  37 import org.openecomp.aai.serialization.engines.TransactionalGraphEngine;
  38
  39 import javax.xml.bind.JAXBException;
  40 import javax.xml.bind.Unmarshaller;
  41 import javax.xml.transform.stream.StreamSource;
  42 import java.io.StringReader;
  43 import java.io.UnsupportedEncodingException;
  44
  45 import static org.junit.Assert.assertEquals;
  46
  47 @Ignore
  48 public class UniqueRelationshipQueryTest {
  49         private ModelInjestor injestor = ModelInjestor.getInstance();
  50         private TransactionalGraphEngine dbEngine =
  51                         new TitanDBEngine(QueryStyle.GREMLIN_UNIQUE,
  52                                 LoaderFactory.createLoaderForVersion(ModelType.MOXY, Version.v8),
  53                                 false);
  54         private final Version version = Version.v8;
  55         private DynamicJAXBContext context = injestor.getContextForVersion(version);
  56         private Unmarshaller unmarshaller = null;
  57
  58         /**
  59          * Configure.
  60          */
  61         @BeforeClass
  62         public static void configure() {
  63                 System.setProperty("AJSC_HOME", ".");
  64                 System.setProperty("BUNDLECONFIG_DIR", "bundleconfig-local");
  65         }
  66
  67
  68         /**
  69          * Setup.
  70          *
  71          * @throws JAXBException the JAXB exception
  72          */
  73         @Before
  74         public void setup() throws JAXBException {
  75                 unmarshaller = context.createUnmarshaller();
  76             unmarshaller.setProperty(UnmarshallerProperties.MEDIA_TYPE, "application/json");
  77             unmarshaller.setProperty(UnmarshallerProperties.JSON_INCLUDE_ROOT, false);
  78                 unmarshaller.setProperty(UnmarshallerProperties.JSON_WRAPPER_AS_ARRAY_NAME, true);
  79         }
  80
  81         /**
  82          * Parent query.
  83          *
  84          * @throws JAXBException the JAXB exception
  85          * @throws UnsupportedEncodingException the unsupported encoding exception
  86          * @throws AAIException the AAI exception
  87          */
  88         @Test
  89     public void parentQuery() throws JAXBException, UnsupportedEncodingException, AAIException {
  90
  91                 String content =
  92                                 "{"
  93                                 + "\"related-to\" : \"pserver\","
  94                                 + "\"relationship-data\" : [{"
  95                                 + "\"relationship-key\" : \"pserver.hostname\","
  96                                 + "\"relationship-value\" : \"key1\""
  97                                 + "}]"
  98                                 + "}";
  99
 100                 Object obj = context.newDynamicEntity("Relationship");
 101
 102                 DynamicEntity entity = (DynamicEntity)unmarshaller.unmarshal(new StreamSource(new StringReader(content)), obj.getClass()).getValue();
 103
 104                 Introspector wrappedObj = IntrospectorFactory.newInstance(ModelType.MOXY, entity);
 105                 QueryParser query = dbEngine.getQueryBuilder().createQueryFromRelationship(wrappedObj);
 106                 String key = "pserver/key1";
 107                 GraphTraversal<Vertex, Vertex> expected =
 108                                 __.<Vertex>start().has("aai-unique-key", key);
 109                 String resultType = "pserver";
 110                 String containerType = "";
 111
 112                 testSet(query, expected, expected, resultType, containerType);
 113
 114     }
 115
 116         /**
 117          * Child query.
 118          *
 119          * @throws JAXBException the JAXB exception
 120          * @throws UnsupportedEncodingException the unsupported encoding exception
 121          * @throws AAIException the AAI exception
 122          */
 123         @Test
 124     public void childQuery() throws JAXBException, UnsupportedEncodingException, AAIException {
 125                 String content =
 126                                 "{"
 127                                 + "\"related-to\" : \"lag-interface\","
 128                                 + "\"relationship-data\" : [{"
 129                                 + "\"relationship-key\" : \"pserver.hostname\","
 130                                 + "\"relationship-value\" : \"key1\""
 131                                 + "}, {"
 132                                 + "\"relationship-key\" : \"lag-interface.interface-name\","
 133                                 + "\"relationship-value\" : \"key2\""
 134                                 + "}]"
 135                                 + "}";
 136
 137                 Object obj = context.newDynamicEntity("Relationship");
 138
 139                 DynamicEntity entity = (DynamicEntity)unmarshaller.unmarshal(new StreamSource(new StringReader(content)), obj.getClass()).getValue();
 140
 141                 Introspector wrappedObj = IntrospectorFactory.newInstance(ModelType.MOXY, entity);
 142                 QueryParser query = dbEngine.getQueryBuilder().createQueryFromRelationship(wrappedObj);
 143
 144                 String key = "pserver/key1/lag-interface/key2";
 145                 GraphTraversal<Vertex, Vertex> expected =
 146                                 __.<Vertex>start().has("aai-unique-key", key);
 147                 GraphTraversal<Vertex, Vertex> parentExpected =
 148                                 __.<Vertex>start().has("aai-unique-key", "pserver/key1");
 149                 String resultType = "lag-interface";
 150                 String containerType = "";
 151
 152                 testSet(query, expected, parentExpected, resultType, containerType);
 153     }
 154
 155         /**
 156          * Naming exceptions.
 157          *
 158          * @throws JAXBException the JAXB exception
 159          * @throws UnsupportedEncodingException the unsupported encoding exception
 160          * @throws AAIException the AAI exception
 161          */
 162         @Test
 163     public void namingExceptions() throws JAXBException, UnsupportedEncodingException, AAIException {
 164                 String content =
 165                                 "{"
 166                                 + "\"related-to\" : \"cvlan-tag\","
 167                                 + "\"relationship-data\" : [{"
 168                                 + "\"relationship-key\" : \"vce.vnf-id\","
 169                                 + "\"relationship-value\" : \"key1\""
 170                                 + "}, {"
 171                                 + "\"relationship-key\" : \"port-group.interface-id\","
 172                                 + "\"relationship-value\" : \"key2\""
 173                                 + "},{"
 174                                 + "\"relationship-key\" : \"cvlan-tag.cvlan-tag\","
 175                                 + "\"relationship-value\" : \"655\""
 176                                 + "}]"
 177                                 + "}";
 178
 179                 Object obj = context.newDynamicEntity("Relationship");
 180
 181                 DynamicEntity entity = (DynamicEntity)unmarshaller.unmarshal(new StreamSource(new StringReader(content)), obj.getClass()).getValue();
 182
 183                 Introspector wrappedObj = IntrospectorFactory.newInstance(ModelType.MOXY, entity);
 184                 QueryParser query = dbEngine.getQueryBuilder().createQueryFromRelationship(wrappedObj);
 185                 String key = "vce/key1/port-group/key2/cvlan-tag/655";
 186                 GraphTraversal<Vertex, Vertex> expected =
 187                                 __.<Vertex>start().has("aai-unique-key", key);
 188                 GraphTraversal<Vertex, Vertex> parentExpected =
 189                                 __.<Vertex>start().has("aai-unique-key", "vce/key1/port-group/key2");
 190                 String resultType = "cvlan-tag";
 191                 String containerType = "";
 192
 193                 testSet(query, expected, parentExpected, resultType, containerType);
 194
 195     }
 196
 197         /**
 198          * Double key.
 199          *
 200          * @throws JAXBException the JAXB exception
 201          * @throws UnsupportedEncodingException the unsupported encoding exception
 202          * @throws AAIException the AAI exception
 203          */
 204         @Test
 205     public void doubleKey() throws JAXBException, UnsupportedEncodingException, AAIException {
 206                 String content =
 207                                 "{"
 208                                 + "\"related-to\" : \"service-capability\","
 209                                 + "\"relationship-data\" : [{"
 210                                 + "\"relationship-key\" : \"service-capability.service-type\","
 211                                 + "\"relationship-value\" : \"key1\""
 212                                 + " }, { "
 213                                 + "\"relationship-key\" : \"service-capability.vnf-type\","
 214                                 + " \"relationship-value\" : \"key2\""
 215                                 + " }]"
 216                                 + "}";
 217
 218                 Object obj = context.newDynamicEntity("Relationship");
 219
 220                 DynamicEntity entity = (DynamicEntity)unmarshaller.unmarshal(new StreamSource(new StringReader(content)), obj.getClass()).getValue();
 221
 222                 Introspector wrappedObj = IntrospectorFactory.newInstance(ModelType.MOXY, entity);
 223                 QueryParser query = dbEngine.getQueryBuilder().createQueryFromRelationship(wrappedObj);
 224
 225                 String key = "service-capability/key1/key2";
 226                 GraphTraversal<Vertex, Vertex> expected =
 227                                 __.<Vertex>start().has("aai-unique-key", key);
 228                 GraphTraversal<Vertex, Vertex> parentExpected =
 229                                 __.<Vertex>start().has("aai-unique-key", "service-capability/key1/key2");
 230                 String resultType = "service-capability";
 231                 String containerType = "";
 232
 233                 testSet(query, expected, parentExpected, resultType, containerType);
 234
 235     }
 236
 237         /**
 238          * Short circuit.
 239          *
 240          * @throws JAXBException the JAXB exception
 241          * @throws UnsupportedEncodingException the unsupported encoding exception
 242          * @throws AAIException the AAI exception
 243          */
 244         @Test
 245     public void shortCircuit() throws JAXBException, UnsupportedEncodingException, AAIException {
 246                 String content =
 247                                 "{"
 248                                 + "\"related-to\" : \"cvlan-tag\","
 249                                 + "\"related-link\" : \"http://mock-system-name.com:8443/aai/v6/network/vces/vce/key1/port-groups/port-group/key2/cvlan-tags/cvlan-tag/655\","
 250                                 + "\"relationship-data\" : [{"
 251                                 + "\"relationship-key\" : \"vce.hostname\","
 252                                 + "\"relationship-value\" : \"key1\""
 253                                 + "}, {"
 254                                 + "\"relationship-key\" : \"port-group.interface-name\","
 255                                 + "\"relationship-value\" : \"key2\""
 256                                 + "},{"
 257                                 + "\"relationship-key\" : \"cvlan-tag.-name\","
 258                                 + "\"relationship-value\" : \"655\""
 259                                 + "}]"
 260                                 + "}";
 261
 262                 Object obj = context.newDynamicEntity("Relationship");
 263
 264                 DynamicEntity entity = (DynamicEntity)unmarshaller.unmarshal(new StreamSource(new StringReader(content)), obj.getClass()).getValue();
 265
 266                 Introspector wrappedObj = IntrospectorFactory.newInstance(ModelType.MOXY, entity);
 267                 QueryParser query = dbEngine.getQueryBuilder().createQueryFromRelationship(wrappedObj);
 268                 String key = "vce/key1/port-group/key2/cvlan-tag/655";
 269                 GraphTraversal<Vertex, Vertex> expected = __.<Vertex>start().has("aai-unique-key", key);
 270                 GraphTraversal<Vertex, Vertex> parentExpected = __.<Vertex>start().has("aai-unique-key", "vce/key1/port-group/key2");
 271                 String resultType = "cvlan-tag";
 272                 String containerType = "";
 273
 274                 testSet(query, expected, parentExpected, resultType, containerType);
 275
 276     }
 277
 278         /**
 279          * Test set.
 280          *
 281          * @param query the query
 282          * @param expected the expected
 283          * @param parentExpected the parent expected
 284          * @param resultType the result type
 285          * @param containerType the container type
 286          */
 287         public void testSet(QueryParser query, GraphTraversal<Vertex, Vertex> expected, GraphTraversal<Vertex, Vertex> parentExpected, String resultType, String containerType) {
 288                 assertEquals(
 289                                 "gremlin query should be " + expected,
 290                                 expected,
 291                                 query.getQueryBuilder().getQuery());
 292                 assertEquals(
 293                                 "parent gremlin query should be " + parentExpected,
 294                                 parentExpected,
 295                                 query.getParentQueryBuilder().getParentQuery());
 296                 assertEquals(
 297                                 "result type should be " + resultType,
 298                                 resultType,
 299                                 query.getResultType());
 300                 assertEquals(
 301                                 "container type should be " + containerType,
 302                                 containerType,
 303                                 query.getContainerType());
 304         }
 305 }