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