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