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 }