aai-core/src/test/java/org/onap/aai/introspection/sideeffect/DataLinkTest.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.introspection.sideeffect;
  21
  22 import org.janusgraph.core.JanusGraphFactory;
  23 import org.janusgraph.core.JanusGraph;
  24 import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversalSource;
  25 import org.apache.tinkerpop.gremlin.structure.Edge;
  26 import org.apache.tinkerpop.gremlin.structure.Graph;
  27 import org.apache.tinkerpop.gremlin.structure.Vertex;
  28 import org.apache.tinkerpop.gremlin.structure.Direction;
  29 import org.apache.tinkerpop.gremlin.structure.VertexProperty;
  30 import static org.hamcrest.MatcherAssert.assertThat;
  31 import static org.hamcrest.Matchers.*;
  32 import org.junit.*;
  33 import org.junit.rules.ExpectedException;
  34 import org.junit.runner.RunWith;
  35 import org.junit.runners.Parameterized;
  36 import org.mockito.Mock;
  37 import org.mockito.MockitoAnnotations;
  38 import org.onap.aai.AAISetup;
  39 import org.onap.aai.db.props.AAIProperties;
  40 import org.onap.aai.dbmap.DBConnectionType;
  41 import org.onap.aai.exceptions.AAIException;
  42 import org.onap.aai.introspection.*;
  43 import org.onap.aai.parsers.query.QueryParser;
  44 import org.onap.aai.serialization.db.DBSerializer;
  45 import org.onap.aai.edges.enums.EdgeProperty;
  46 import org.onap.aai.serialization.engines.QueryStyle;
  47 import org.onap.aai.serialization.engines.JanusGraphDBEngine;
  48 import org.onap.aai.serialization.engines.TransactionalGraphEngine;
  49
  50 import java.io.UnsupportedEncodingException;
  51 import java.lang.reflect.InvocationTargetException;
  52 import java.net.MalformedURLException;
  53 import java.net.URISyntaxException;
  54 import java.util.Arrays;
  55 import java.util.Collection;
  56 import java.util.Iterator;
  57 import java.util.List;
  58
  59 import static org.junit.Assert.*;
  60 import static org.mockito.Mockito.spy;
  61 import static org.mockito.Mockito.when;
  62
  63 @RunWith(value = Parameterized.class)
  64 public class DataLinkTest extends AAISetup {
  65
  66         private static JanusGraph graph;
  67         private final static ModelType introspectorFactoryType = ModelType.MOXY;
  68         private final static DBConnectionType type = DBConnectionType.REALTIME;
  69         private static Loader loader;
  70         private static TransactionalGraphEngine dbEngine;
  71         @Mock private QueryParser parser;
  72         @Mock private Vertex self;
  73         @Mock private VertexProperty<String> prop;
  74         @Rule
  75         public ExpectedException thrown = ExpectedException.none();
  76
  77
  78
  79         @Parameterized.Parameter(value = 0)
  80         public QueryStyle queryStyle;
  81
  82         @Parameterized.Parameters(name = "QueryStyle.{0}")
  83         public static Collection<Object[]> data() {
  84                 return Arrays.asList(new Object[][]{
  85                                 {QueryStyle.TRAVERSAL},
  86                                 {QueryStyle.TRAVERSAL_URI}
  87                 });
  88         }
  89
  90         @BeforeClass
  91         public static void setup() throws NoSuchFieldException, SecurityException, Exception {
  92                 graph = JanusGraphFactory.build().set("storage.backend","inmemory").open();
  93
  94
  95                 graph.traversal().addV("aai-node-type", "vpn-binding", "vpn-id", "addKey", AAIProperties.AAI_URI, "/network/vpn-bindings/vpn-binding/addKey").as("v1")
  96                 .addV("aai-node-type", "vpn-binding", "vpn-id", "modifyKey", AAIProperties.AAI_URI, "/network/vpn-bindings/vpn-binding/modifyKey").as("v2")
  97                 .addV("aai-node-type", "route-target", "global-route-target", "modifyTargetKey", "route-target-role", "modifyRoleKey", "linked", true, AAIProperties.AAI_URI, "/network/vpn-bindings/vpn-binding/modifyKey/route-targets/route-target/modifyTargetKey/modifyRoleKey")
  98                                 .addOutE("org.onap.relationships.inventory.BelongsTo", "v2", EdgeProperty.CONTAINS.toString(), true)
  99                 .addV("aai-node-type", "vpn-binding", "vpn-id", "deleteKey",AAIProperties.AAI_URI, "/network/vpn-bindings/vpn-binding/deleteKey").as("v3" )
 100                 .addV("aai-node-type", "route-target", "global-route-target", "deleteTargetKey", "route-target-role", "deleteRoleKey", "linked", true, AAIProperties.AAI_URI, "/network/vpn-bindings/vpn-binding/deleteKey/route-targets/route-target/deleteTargetKey/deleteRoleKey")
 101                                 .addOutE("org.onap.relationships.inventory.BelongsTo", "v3", EdgeProperty.CONTAINS.toString(), true)
 102                 .addV("aai-node-type", "vpn-binding", "vpn-id", "getKey", AAIProperties.AAI_URI, "/network/vpn-bindings/vpn-binding/getKey").as("v4")
 103                 .addV("aai-node-type", "route-target", "global-route-target", "getTargetKey", "route-target-role", "getRoleKey", "linked", true, AAIProperties.AAI_URI, "/network/vpn-bindings/vpn-binding/getKey/route-targets/route-target/getTargetKeyNoLink/getRoleKeyNoLink")
 104                                 .addOutE("org.onap.relationships.inventory.BelongsTo", "v4", EdgeProperty.CONTAINS.toString(), true)
 105                 .addV("aai-node-type", "vpn-binding", "vpn-id", "getKeyNoLink", AAIProperties.AAI_URI, "/network/vpn-bindings/vpn-binding/getKeyNoLink").as("v5")
 106                 .addV("aai-node-type", "route-target", "global-route-target", "getTargetKeyNoLink", "route-target-role", "getRoleKeyNoLink", AAIProperties.AAI_URI, "/network/vpn-bindings/vpn-binding/getKeyNoLink/route-targets/route-target/getTargetKeyNoLink/getRoleKeyNoLink")
 107                                 .addOutE("org.onap.relationships.inventory.BelongsTo", "v5", EdgeProperty.CONTAINS.toString(), true)
 108                 .next();
 109                 graph.tx().commit();
 110
 111                 /*Commented for SysOut issues
 112                  */
 113                 //graph.traversal().V().has("aai-uri","/network/vpn-bindings/vpn-binding/deleteKey").properties().forEachRemaining(p->System.out.println(p.key() +" : " + p.value()));
 114
 115         }
 116
 117         @AfterClass
 118         public static void tearDown() {
 119                 graph.tx().rollback();
 120                 graph.close();
 121         }
 122
 123         @Before
 124         public void initMock() {
 125                 loader = loaderFactory.createLoaderForVersion(introspectorFactoryType, schemaVersions.getDefaultVersion());
 126                 MockitoAnnotations.initMocks(this);
 127                 dbEngine = new JanusGraphDBEngine(
 128                                 queryStyle,
 129                                 type,
 130                                 loader);
 131         }
 132
 133         @Test
 134         public void verifyCreationOfVertex() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
 135
 136                 final Loader loader = loaderFactory.createLoaderForVersion(ModelType.MOXY, schemaVersions.getDepthVersion());
 137                 final Introspector obj = loader.introspectorFromName("vpn-binding");
 138                 obj.setValue("vpn-id", "addKey");
 139                 obj.setValue("global-route-target", "key1");
 140                 obj.setValue("route-target-role", "key2");
 141                 TransactionalGraphEngine spy = spy(dbEngine);
 142                 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
 143                 Graph g = graph.newTransaction();
 144                 GraphTraversalSource traversal = g.traversal();
 145 //              Graph g = graph.newTransaction();
 146 //              GraphTraversalSource traversal = g;
 147         //      System.out.println("Begin method inventory:");
 148                 Iterator<Vertex> vertexItr = traversal.V();
 149                 while( vertexItr != null && vertexItr.hasNext() ){
 150                         Vertex v = vertexItr.next();
 151                 //      System.out.println("\nnodeType="+v.<String>property("aai-node-type"));
 152                         for(String key: v.keys()) {
 153                         //      System.out.println("label="+v.label()+";key= "+key+";value= "+v.value(key)+";id= "+v.id());
 154                         }
 155                         Direction d = null;
 156                         Iterator<Edge> edgeItr = v.edges(Direction.BOTH);
 157                         while( edgeItr != null && edgeItr.hasNext() ){
 158                                 Edge e = edgeItr.next();
 159                                 //System.out.println("outV="+e.outVertex().property(AAIProperties.NODE_TYPE)+"/"+e.outVertex().id()+";inV= "+e.inVertex().property(AAIProperties.NODE_TYPE)+"/"+e.inVertex().id());
 160                         }
 161                 }
 162                 //System.out.println("End method inventory:");
 163                 when(spy.asAdmin()).thenReturn(adminSpy);
 164                 when(adminSpy.getTraversalSource()).thenReturn(traversal);
 165                 when(spy.tx()).thenReturn(g);
 166                 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
 167                 when(prop.orElse(null)).thenReturn(obj.getURI());
 168                 DBSerializer serializer = new DBSerializer(schemaVersions.getDefaultVersion(), spy, introspectorFactoryType, "AAI_TEST");
 169                 SideEffectRunner runner = new SideEffectRunner
 170                                 .Builder(spy, serializer).addSideEffect(DataLinkWriter.class).build();
 171
 172                 runner.execute(obj, self);
 173
 174                 assertEquals("route-target vertex found", true, traversal.V()
 175                                 .has(AAIProperties.NODE_TYPE, "route-target").has("global-route-target", "key1").has("route-target-role", "key2").has("linked", true).hasNext());
 176                 g.tx().rollback();
 177
 178         }
 179
 180         @Test
 181         public void verifyModificationOfVertex() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
 182
 183                 final Loader loader = loaderFactory.createLoaderForVersion(ModelType.MOXY, schemaVersions.getDepthVersion());
 184                 final Introspector obj = loader.introspectorFromName("vpn-binding");
 185                 obj.setValue("vpn-id", "modifyKey");
 186                 obj.setValue("global-route-target", "modifyTargetKey2");
 187                 obj.setValue("route-target-role", "modifyRoleKey2");
 188                 TransactionalGraphEngine spy = spy(dbEngine);
 189                 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
 190 //              Graph g = graph.newTransaction();
 191 //              GraphTraversalSource traversal = g;
 192                 Graph g = graph.newTransaction();
 193                 GraphTraversalSource traversal = g.traversal();
 194                 Iterator<Vertex> vertexItr = traversal.V();
 195                 while( vertexItr != null && vertexItr.hasNext() ){
 196                         Vertex v = vertexItr.next();
 197                         //System.out.println("\nnodeType="+v.<String>property("aai-node-type"));
 198                         for(String key: v.keys()) {
 199                                 //System.out.println("label="+v.label()+";key= "+key+";value= "+v.value(key)+"/"+v.id());
 200                         }
 201                         Direction d = null;
 202                         Iterator<Edge> edgeItr = v.edges(Direction.BOTH);
 203                         while( edgeItr != null && edgeItr.hasNext() ){
 204                                 Edge e = edgeItr.next();
 205                         //      System.out.println("outV="+e.outVertex().property(AAIProperties.NODE_TYPE)+"/"+e.outVertex().id()+";inV= "+e.inVertex().property(AAIProperties.NODE_TYPE)+"/"+e.inVertex().id());
 206                         }
 207                 }
 208         //      System.out.println("End method inventory:");
 209
 210                 when(spy.asAdmin()).thenReturn(adminSpy);
 211                 when(adminSpy.getTraversalSource()).thenReturn(traversal);
 212 //              when(spy.tx()).thenReturn(graph);
 213                 when(spy.tx()).thenReturn(g);
 214                 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
 215                 when(prop.orElse(null)).thenReturn(obj.getURI());
 216                 DBSerializer serializer = new DBSerializer(schemaVersions.getDefaultVersion(), spy, introspectorFactoryType, "AAI_TEST");
 217                 SideEffectRunner runner = new SideEffectRunner
 218                                 .Builder(spy, serializer).addSideEffect(DataLinkWriter.class).build();
 219                 //System.out.println("Traversal Source: "+traversal.toString());
 220                 vertexItr = traversal.V();
 221         //      System.out.println("Begin method inventory:");
 222                 while( vertexItr != null && vertexItr.hasNext() ){
 223                         Vertex v = vertexItr.next();
 224                         //System.out.println("\nnodeType="+v.<String>property("aai-node-type"));
 225                         for(String key: v.keys()) {
 226                         //      System.out.println("label="+v.label()+";key= "+key+";value= "+v.value(key)+"/"+v.id());
 227                         }
 228                         Iterator<Edge> edgeItr = v.edges(Direction.BOTH);
 229                         while( edgeItr != null && edgeItr.hasNext() ){
 230                                 Edge e = edgeItr.next();
 231                                 //System.out.println("outV="+e.outVertex().property(AAIProperties.NODE_TYPE)+"/"+e.outVertex().id()+";inV= "+e.inVertex().property(AAIProperties.NODE_TYPE)+"/"+e.inVertex().id());
 232                         }
 233                 }
 234                 //System.out.println("End method inventory:");
 235                 try {
 236                 runner.execute(obj, self);
 237                 } catch(Exception e) {
 238
 239                 }
 240 //              runner.execute(obj, self);
 241                 //System.out.println("=================\n");
 242                 vertexItr = traversal.V();
 243                 while( vertexItr != null && vertexItr.hasNext() ){
 244                         Vertex v = vertexItr.next();
 245                         //System.out.println("\nnodeType="+v.<String>property("aai-node-type"));
 246                         for(String key: v.keys()) {
 247                         //      System.out.println("label="+v.label()+";key= "+key+";value= "+v.value(key)+"/"+v.id());
 248                         }
 249                         Iterator<Edge> edgeItr = v.edges(Direction.BOTH);
 250                         while( edgeItr != null && edgeItr.hasNext() ){
 251                                 Edge e = edgeItr.next();
 252                         //      System.out.println("outV="+e.outVertex().property(AAIProperties.NODE_TYPE)+"/"+e.outVertex().id()+";inV= "+e.inVertex().property(AAIProperties.NODE_TYPE)+"/"+e.inVertex().id());
 253                         }
 254                 }
 255                 assertThat("new route-target vertex found with/or without link", traversal.V()
 256                                 .has(AAIProperties.NODE_TYPE, "route-target").has("global-route-target", "modifyTargetKey2").has("route-target-role", "modifyRoleKey2").hasNext(),is(true));
 257                 assertThat("new route-target vertex found", traversal.V()
 258                                 .has(AAIProperties.NODE_TYPE, "route-target").has("global-route-target", "modifyTargetKey2").has("route-target-role", "modifyRoleKey2").has("linked", true).hasNext(),is(true));
 259                 assertThat("previous link removed", traversal.V()
 260                                 .has(AAIProperties.NODE_TYPE, "route-target").has("global-route-target", "modifyTargetKey").has("route-target-role", "modifyRoleKey").has("linked").hasNext(),is(not(true)));
 261                 assertThat("previous vertex still exists", traversal.V()
 262                                 .has(AAIProperties.NODE_TYPE, "route-target").has("global-route-target", "modifyTargetKey").has("route-target-role", "modifyRoleKey").hasNext(),is(true));
 263                 g.tx().rollback();
 264
 265         }
 266
 267         @Test
 268         public void verifyDeleteOfVertex() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
 269
 270                 final Loader loader = loaderFactory.createLoaderForVersion(ModelType.MOXY, schemaVersions.getDepthVersion());
 271                 final Introspector obj = loader.introspectorFromName("vpn-binding");
 272                 obj.setValue("vpn-id", "deleteKey");
 273                 TransactionalGraphEngine spy = spy(dbEngine);
 274                 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
 275                 Graph g = graph.newTransaction();
 276                 GraphTraversalSource traversal = g.traversal();
 277                 when(spy.asAdmin()).thenReturn(adminSpy);
 278                 when(adminSpy.getTraversalSource()).thenReturn(traversal);
 279                 when(adminSpy.getReadOnlyTraversalSource()).thenReturn(traversal);
 280                 when(spy.tx()).thenReturn(g);
 281                 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
 282                 when(prop.orElse(null)).thenReturn(obj.getURI());
 283                 DBSerializer serializer = new DBSerializer(schemaVersions.getDefaultVersion(), spy, introspectorFactoryType, "AAI_TEST");
 284                 SideEffectRunner runner = new SideEffectRunner
 285                                 .Builder(spy, serializer).addSideEffect(DataLinkWriter.class).build();
 286
 287                 runner.execute(obj, self);
 288
 289                 assertEquals("route-target vertex not found", false, traversal.V()
 290                                 .has(AAIProperties.NODE_TYPE, "route-target")
 291                                 .has("global-route-target", "deleteTargetKey")
 292                                 .has("route-target-role", "deleteRoleKey")
 293                                 .has("linked", true)
 294                                 .hasNext()
 295                 );
 296
 297                 g.tx().rollback();
 298
 299         }
 300
 301         @Test
 302         public void verifyPropertyPopulation() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
 303
 304                 final Loader loader = loaderFactory.createLoaderForVersion(ModelType.MOXY, schemaVersions.getDepthVersion());
 305                 final Introspector obj = loader.introspectorFromName("vpn-binding");
 306                 obj.setValue("vpn-id", "getKey");
 307                 TransactionalGraphEngine spy = spy(dbEngine);
 308                 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
 309                 Graph g = graph.newTransaction();
 310                 GraphTraversalSource traversal = g.traversal();
 311                 when(spy.asAdmin()).thenReturn(adminSpy);
 312                 when(adminSpy.getTraversalSource()).thenReturn(traversal);
 313                 when(spy.tx()).thenReturn(g);
 314                 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
 315                 when(prop.orElse(null)).thenReturn(obj.getURI());
 316                 DBSerializer serializer = new DBSerializer(schemaVersions.getDefaultVersion(), spy, introspectorFactoryType, "AAI_TEST");
 317                 SideEffectRunner runner = new SideEffectRunner
 318                                 .Builder(spy, serializer).addSideEffect(DataLinkReader.class).build();
 319
 320                 runner.execute(obj, self);
 321
 322                 assertEquals("both properties have been populated in target object", true, obj.getValue("global-route-target").equals("getTargetKey") && obj.getValue("route-target-role").equals("getRoleKey"));
 323                 g.tx().rollback();
 324
 325         }
 326
 327         @Test
 328         public void verifyPropertyPopulationWithV10OnlyPut() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
 329                 final Introspector obj = loader.introspectorFromName("vpn-binding");
 330                 obj.setValue("vpn-id", "getKeyNoLink");
 331                 final Introspector routeTargets = loader.introspectorFromName("route-targets");
 332                 obj.setValue("route-targets", routeTargets.getUnderlyingObject());
 333                 List<Object> targets = routeTargets.getValue("route-target");
 334                 final Introspector routeTargetOne = loader.introspectorFromName("route-target");
 335                 routeTargetOne.setValue("global-route-target", "getTargetKeyNoLink");
 336                 routeTargetOne.setValue("route-target-role", "getRoleKeyNoLink");
 337                 targets.add(routeTargetOne.getUnderlyingObject());
 338                 final Introspector routeTargetTwo = loader.introspectorFromName("route-target");
 339                 routeTargetTwo.setValue("global-route-target", "getTargetKeyNoLink2");
 340                 routeTargetTwo.setValue("route-target-role", "getRoleKeyNoLink2");
 341                 targets.add(routeTargetTwo.getUnderlyingObject());
 342                 TransactionalGraphEngine spy = spy(dbEngine);
 343                 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
 344                 Graph g = graph.newTransaction();
 345                 GraphTraversalSource traversal = g.traversal();
 346                 when(spy.tx()).thenReturn(g);
 347                 when(spy.asAdmin()).thenReturn(adminSpy);
 348                 when(adminSpy.getTraversalSource()).thenReturn(traversal);
 349                 when(spy.tx()).thenReturn(g);
 350                 when(parser.isDependent()).thenReturn(false);
 351                 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
 352                 when(prop.orElse(null)).thenReturn(obj.getURI());
 353                 DBSerializer serializer = new DBSerializer(schemaVersions.getDefaultVersion(), spy, introspectorFactoryType, "AAI_TEST");
 354                 Vertex v = serializer.createNewVertex(obj);
 355                 serializer.serializeToDb(obj, v, parser, obj.getURI(), "testing");
 356                 Vertex routeTargetOneV = traversal.V().has("global-route-target", "getTargetKeyNoLink").next();
 357                 Vertex routeTargetTwoV = traversal.V().has("global-route-target", "getTargetKeyNoLink2").next();
 358
 359                 assertEquals("first route target put has linked", true, routeTargetOneV.property(AAIProperties.LINKED).orElse(false));
 360                 assertEquals("second route target put does not have linked", false, routeTargetTwoV.property(AAIProperties.LINKED).orElse(false));
 361
 362                 g.tx().rollback();
 363
 364         }
 365 }