aai-core/src/test/java/org/openecomp/aai/parsers/query/UniqueURIQueryTest.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
  30 import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversal;
  31 import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.__;
  32 import org.apache.tinkerpop.gremlin.structure.Vertex;
  33 import org.eclipse.persistence.jaxb.dynamic.DynamicJAXBContext;
  34 import org.junit.BeforeClass;
  35 import org.junit.Ignore;
  36 import org.junit.Test;
  37
  38 import org.openecomp.aai.exceptions.AAIException;
  39 import org.openecomp.aai.introspection.LoaderFactory;
  40 import org.openecomp.aai.introspection.ModelInjestor;
  41 import org.openecomp.aai.introspection.ModelType;
  42 import org.openecomp.aai.introspection.Version;
  43 import org.openecomp.aai.serialization.engines.QueryStyle;
  44 import org.openecomp.aai.serialization.engines.TitanDBEngine;
  45 import org.openecomp.aai.serialization.engines.TransactionalGraphEngine;
  46
  47 @Ignore
  48 public class UniqueURIQueryTest {
  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
  57         /**
  58          * Configure.
  59          */
  60         @BeforeClass
  61         public static void configure() {
  62                 System.setProperty("AJSC_HOME", ".");
  63                 System.setProperty("BUNDLECONFIG_DIR", "bundleconfig-local");
  64         }
  65
  66
  67         /**
  68          * Parent query.
  69          *
  70          * @throws UnsupportedEncodingException the unsupported encoding exception
  71          * @throws AAIException the AAI exception
  72          */
  73         @Test
  74     public void parentQuery() throws UnsupportedEncodingException, AAIException {
  75                 URI uri = UriBuilder.fromPath("cloud-infrastructure/complexes/complex/key1").build();
  76                 String key = "complex/key1";
  77                 QueryParser query = dbEngine.getQueryBuilder().createQueryFromURI(uri);
  78                 GraphTraversal<Vertex, Vertex> expected = __.<Vertex>start().has("aai-unique-key", key);
  79                 String parentResultType = "";
  80                 String resultType = "complex";
  81                 String containerType = "";
  82
  83                 testSet(query, expected, expected, parentResultType, resultType, containerType);
  84
  85     }
  86
  87         /**
  88          * Parent plural query.
  89          *
  90          * @throws UnsupportedEncodingException the unsupported encoding exception
  91          * @throws AAIException the AAI exception
  92          */
  93         @Test
  94     public void parentPluralQuery() throws UnsupportedEncodingException, AAIException {
  95                 URI uri = UriBuilder.fromPath("cloud-infrastructure/complexes").build();
  96                 QueryParser query = dbEngine.getQueryBuilder().createQueryFromURI(uri);
  97                 GraphTraversal<Vertex, Vertex> expected = __.<Vertex>start().has("aai-node-type", "complex");
  98                 String parentResultType = "";
  99                 String resultType = "complex";
 100                 String containerType = "complexes";
 101
 102                 testSet(query, expected, expected, parentResultType, resultType, containerType);
 103
 104     }
 105
 106         /**
 107          * Child query.
 108          *
 109          * @throws UnsupportedEncodingException the unsupported encoding exception
 110          * @throws AAIException the AAI exception
 111          */
 112         @Test
 113     public void childQuery() throws UnsupportedEncodingException, AAIException {
 114                 URI uri = UriBuilder.fromPath("cloud-infrastructure/complexes/complex/key1/ctag-pools/ctag-pool/key2/key3").build();
 115                 QueryParser query = dbEngine.getQueryBuilder().createQueryFromURI(uri);
 116                 String parentKey = "complex/key1";
 117                 String key = parentKey + "/ctag-pool/key2/key3";
 118                 GraphTraversal<Vertex, Vertex> expected = __.<Vertex>start().has("aai-unique-key", key);
 119                 GraphTraversal<Vertex, Vertex> parentExpected =  __.<Vertex>start().has("aai-unique-key", parentKey);
 120                 String parentResultType = "complex";
 121                 String resultType = "ctag-pool";
 122                 String containerType = "";
 123
 124                 testSet(query, expected, parentExpected, parentResultType, resultType, containerType);
 125
 126     }
 127
 128         /**
 129          * Naming exceptions.
 130          *
 131          * @throws UnsupportedEncodingException the unsupported encoding exception
 132          * @throws AAIException the AAI exception
 133          */
 134         @Test
 135     public void namingExceptions() throws UnsupportedEncodingException, AAIException {
 136                 URI uri = UriBuilder.fromPath("network/vces/vce/key1/port-groups/port-group/key2/cvlan-tags/cvlan-tag/655").build();
 137                 QueryParser query = dbEngine.getQueryBuilder().createQueryFromURI(uri);
 138                 String parentKey = "vce/key1/port-group/key2";
 139                 String key = parentKey + "/cvlan-tag/655";
 140                 GraphTraversal<Vertex, Vertex> expected = __.<Vertex>start().has("aai-unique-key", key);
 141                 GraphTraversal<Vertex, Vertex> parentExpected = __.<Vertex>start().has("aai-unique-key", parentKey);
 142                 String parentResultType = "port-group";
 143                 String resultType = "cvlan-tag";
 144                 String containerType = "";
 145
 146                 testSet(query, expected, parentExpected, parentResultType, resultType, containerType);
 147
 148     }
 149
 150         /**
 151          * Gets the all.
 152          *
 153          * @return the all
 154          * @throws UnsupportedEncodingException the unsupported encoding exception
 155          * @throws AAIException the AAI exception
 156          */
 157         @Test
 158     public void getAll() throws UnsupportedEncodingException, AAIException {
 159                 String parentURI = "network/vces/vce/key1/port-groups/port-group/key2";
 160                 String parentKey = "vce/key1/port-group/key2";
 161                 URI uri = UriBuilder.fromPath(parentURI + "/cvlan-tags").build();
 162                 QueryParser query = dbEngine.getQueryBuilder().createQueryFromURI(uri);
 163                 GraphTraversal<Vertex, Vertex> expected = __.<Vertex>start().has("aai-unique-key", parentKey).out("hasCTag").has("aai-node-type", "cvlan-tag");
 164                 GraphTraversal<Vertex, Vertex> parentExpected = __.<Vertex>start().has("aai-unique-key",parentKey);
 165                 String parentResultType = "port-group";
 166                 String resultType = "cvlan-tag";
 167                 String containerType = "cvlan-tags";
 168
 169                 testSet(query, expected, parentExpected, parentResultType, resultType, containerType);
 170
 171     }
 172
 173         /**
 174          * Test set.
 175          *
 176          * @param query the query
 177          * @param expected the expected
 178          * @param parentExpected the parent expected
 179          * @param parentResultType the parent result type
 180          * @param resultType the result type
 181          * @param containerType the container type
 182          */
 183         public void testSet(QueryParser query, GraphTraversal<Vertex, Vertex> expected, GraphTraversal<Vertex, Vertex> parentExpected, String parentResultType, String resultType, String containerType) {
 184                 assertEquals(
 185                                 "gremlin query should be " + expected,
 186                                 expected,
 187                                 query.getQueryBuilder().getQuery());
 188                 assertEquals(
 189                                 "parent gremlin query should be " + parentExpected,
 190                                 parentExpected,
 191                                 query.getQueryBuilder().getParentQuery().getQuery());
 192                 assertEquals(
 193                                 "parent result type should be " + parentResultType,
 194                                 parentResultType,
 195                                 query.getParentResultType());
 196                 assertEquals(
 197                                 "result type should be " + resultType,
 198                                 resultType,
 199                                 query.getResultType());
 200                 assertEquals(
 201                                 "container type should be " + containerType,
 202                                 containerType,
 203                                 query.getContainerType());
 204         }
 205 }