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 com.thinkaurelius.titan.core.TitanFactory;
  23 import com.thinkaurelius.titan.core.TitanGraph;
  24 import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversalSource;
  25 import org.apache.tinkerpop.gremlin.structure.Graph;
  26 import org.apache.tinkerpop.gremlin.structure.Vertex;
  27 import org.apache.tinkerpop.gremlin.structure.VertexProperty;
  28 import org.junit.*;
  29 import org.junit.rules.ExpectedException;
  30 import org.junit.runner.RunWith;
  31 import org.junit.runners.Parameterized;
  32 import org.mockito.Mock;
  33 import org.mockito.MockitoAnnotations;
  34 import org.onap.aai.AAISetup;
  35 import org.onap.aai.db.props.AAIProperties;
  36 import org.onap.aai.dbmap.DBConnectionType;
  37 import org.onap.aai.exceptions.AAIException;
  38 import org.onap.aai.introspection.*;
  39 import org.onap.aai.parsers.query.QueryParser;
  40 import org.onap.aai.serialization.db.DBSerializer;
  41 import org.onap.aai.serialization.db.EdgeProperty;
  42 import org.onap.aai.serialization.engines.QueryStyle;
  43 import org.onap.aai.serialization.engines.TitanDBEngine;
  44 import org.onap.aai.serialization.engines.TransactionalGraphEngine;
  45
  46 import java.io.UnsupportedEncodingException;
  47 import java.lang.reflect.InvocationTargetException;
  48 import java.net.MalformedURLException;
  49 import java.net.URISyntaxException;
  50 import java.util.Arrays;
  51 import java.util.Collection;
  52 import java.util.List;
  53
  54 import static org.junit.Assert.assertEquals;
  55 import static org.mockito.Mockito.spy;
  56 import static org.mockito.Mockito.when;
  57
  58 @RunWith(value = Parameterized.class)
  59 public class DataLinkTest extends AAISetup {
  60
  61         private static TitanGraph graph;
  62         private final static Version version = Version.getLatest();
  63         private final static ModelType introspectorFactoryType = ModelType.MOXY;
  64         private final static DBConnectionType type = DBConnectionType.REALTIME;
  65         private static Loader loader;
  66         private static TransactionalGraphEngine dbEngine;
  67         @Mock private QueryParser parser;
  68         @Mock private Vertex self;
  69         @Mock private VertexProperty<String> prop;
  70         @Rule
  71         public ExpectedException thrown = ExpectedException.none();
  72
  73         @Parameterized.Parameter(value = 0)
  74         public QueryStyle queryStyle;
  75
  76         @Parameterized.Parameters(name = "QueryStyle.{0}")
  77         public static Collection<Object[]> data() {
  78                 return Arrays.asList(new Object[][]{
  79                                 {QueryStyle.TRAVERSAL}
  80                 });
  81         }
  82
  83         @BeforeClass
  84         public static void setup() throws NoSuchFieldException, SecurityException, Exception {
  85                 graph = TitanFactory.build().set("storage.backend","inmemory").open();
  86                 loader = LoaderFactory.createLoaderForVersion(introspectorFactoryType, version);
  87
  88                 graph.traversal().addV("aai-node-type", "vpn-binding", "vpn-id", "addKey", AAIProperties.AAI_URI, "/network/vpn-bindings/vpn-binding/addKey").as("v1")
  89                 .addV("aai-node-type", "vpn-binding", "vpn-id", "modifyKey", AAIProperties.AAI_URI, "/network/vpn-bindings/vpn-binding/modifyKey").as("v2")
  90                 .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")
  91                                 .addOutE("org.onap.relationships.inventory.BelongsTo", "v2", EdgeProperty.CONTAINS.toString(), true)
  92                 .addV("aai-node-type", "vpn-binding", "vpn-id", "deleteKey",AAIProperties.AAI_URI, "/network/vpn-bindings/vpn-binding/deleteKey").as("v3" )
  93                 .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")
  94                                 .addOutE("org.onap.relationships.inventory.BelongsTo", "v3", EdgeProperty.CONTAINS.toString(), true)
  95                 .addV("aai-node-type", "vpn-binding", "vpn-id", "getKey", AAIProperties.AAI_URI, "/network/vpn-bindings/vpn-binding/getKey").as("v4")
  96                 .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")
  97                                 .addOutE("org.onap.relationships.inventory.BelongsTo", "v4", EdgeProperty.CONTAINS.toString(), true)
  98                 .addV("aai-node-type", "vpn-binding", "vpn-id", "getKeyNoLink", AAIProperties.AAI_URI, "/network/vpn-bindings/vpn-binding/getKeyNoLink").as("v5")
  99                 .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")
 100                                 .addOutE("org.onap.relationships.inventory.BelongsTo", "v5", EdgeProperty.CONTAINS.toString(), true)
 101                 .next();
 102                 graph.tx().commit();
 103
 104                 graph.traversal().V().has("aai-uri","/network/vpn-bindings/vpn-binding/deleteKey").properties().forEachRemaining(p->System.out.println(p.key() +" : " + p.value()));
 105         }
 106
 107         @AfterClass
 108         public static void tearDown() {
 109                 graph.tx().rollback();
 110                 graph.close();
 111         }
 112
 113         @Before
 114         public void initMock() {
 115                 MockitoAnnotations.initMocks(this);
 116                 dbEngine = new TitanDBEngine(
 117                                 queryStyle,
 118                                 type,
 119                                 loader);
 120         }
 121
 122         @Test
 123         public void verifyCreationOfVertex() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
 124
 125                 final Loader loader = LoaderFactory.createLoaderForVersion(ModelType.MOXY, Version.v9);
 126                 final Introspector obj = loader.introspectorFromName("vpn-binding");
 127                 obj.setValue("vpn-id", "addKey");
 128                 obj.setValue("global-route-target", "key1");
 129                 obj.setValue("route-target-role", "key2");
 130                 TransactionalGraphEngine spy = spy(dbEngine);
 131                 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
 132                 Graph g = graph.newTransaction();
 133                 GraphTraversalSource traversal = g.traversal();
 134                 when(spy.asAdmin()).thenReturn(adminSpy);
 135                 when(adminSpy.getTraversalSource()).thenReturn(traversal);
 136                 when(spy.tx()).thenReturn(g);
 137                 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
 138                 when(prop.orElse(null)).thenReturn(obj.getURI());
 139                 DBSerializer serializer = new DBSerializer(version, spy, introspectorFactoryType, "AAI_TEST");
 140                 SideEffectRunner runner = new SideEffectRunner
 141                                 .Builder(spy, serializer).addSideEffect(DataLinkWriter.class).build();
 142
 143                 runner.execute(obj, self);
 144
 145                 assertEquals("route-target vertex found", true, traversal.V()
 146                                 .has(AAIProperties.NODE_TYPE, "route-target").has("global-route-target", "key1").has("route-target-role", "key2").has("linked", true).hasNext());
 147                 g.tx().rollback();
 148
 149         }
 150
 151         @Test
 152         public void verifyModificationOfVertex() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
 153
 154                 final Loader loader = LoaderFactory.createLoaderForVersion(ModelType.MOXY, Version.v9);
 155                 final Introspector obj = loader.introspectorFromName("vpn-binding");
 156                 obj.setValue("vpn-id", "modifyKey");
 157                 obj.setValue("global-route-target", "modifyTargetKey2");
 158                 obj.setValue("route-target-role", "modifyRoleKey2");
 159                 TransactionalGraphEngine spy = spy(dbEngine);
 160                 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
 161                 Graph g = graph.newTransaction();
 162                 GraphTraversalSource traversal = g.traversal();
 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(version, 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", "modifyTargetKey2").has("route-target-role", "modifyRoleKey2").has("linked", true).hasNext());
 176                 assertEquals("previous link removed", true, traversal.V()
 177                                 .has(AAIProperties.NODE_TYPE, "route-target").has("global-route-target", "modifyTargetKey").has("route-target-role", "modifyRoleKey").hasNot("linked").hasNext());
 178                 g.tx().rollback();
 179
 180         }
 181
 182         @Test
 183         public void verifyDeleteOfVertex() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
 184
 185                 final Loader loader = LoaderFactory.createLoaderForVersion(ModelType.MOXY, Version.v9);
 186                 final Introspector obj = loader.introspectorFromName("vpn-binding");
 187                 obj.setValue("vpn-id", "deleteKey");
 188                 TransactionalGraphEngine spy = spy(dbEngine);
 189                 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
 190                 Graph g = graph.newTransaction();
 191                 GraphTraversalSource traversal = g.traversal();
 192                 when(spy.asAdmin()).thenReturn(adminSpy);
 193                 when(adminSpy.getTraversalSource()).thenReturn(traversal);
 194                 when(adminSpy.getReadOnlyTraversalSource()).thenReturn(traversal);
 195                 when(spy.tx()).thenReturn(g);
 196                 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
 197                 when(prop.orElse(null)).thenReturn(obj.getURI());
 198                 DBSerializer serializer = new DBSerializer(version, spy, introspectorFactoryType, "AAI_TEST");
 199                 SideEffectRunner runner = new SideEffectRunner
 200                                 .Builder(spy, serializer).addSideEffect(DataLinkWriter.class).build();
 201
 202                 runner.execute(obj, self);
 203
 204                 assertEquals("route-target vertex not found", false, traversal.V()
 205                                 .has(AAIProperties.NODE_TYPE, "route-target")
 206                                 .has("global-route-target", "deleteTargetKey")
 207                                 .has("route-target-role", "deleteRoleKey")
 208                                 .has("linked", true)
 209                                 .hasNext()
 210                 );
 211
 212                 g.tx().rollback();
 213
 214         }
 215
 216         @Test
 217         public void verifyPropertyPopulation() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
 218
 219                 final Loader loader = LoaderFactory.createLoaderForVersion(ModelType.MOXY, Version.v9);
 220                 final Introspector obj = loader.introspectorFromName("vpn-binding");
 221                 obj.setValue("vpn-id", "getKey");
 222                 TransactionalGraphEngine spy = spy(dbEngine);
 223                 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
 224                 Graph g = graph.newTransaction();
 225                 GraphTraversalSource traversal = g.traversal();
 226                 when(spy.asAdmin()).thenReturn(adminSpy);
 227                 when(adminSpy.getTraversalSource()).thenReturn(traversal);
 228                 when(spy.tx()).thenReturn(g);
 229                 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
 230                 when(prop.orElse(null)).thenReturn(obj.getURI());
 231                 DBSerializer serializer = new DBSerializer(version, spy, introspectorFactoryType, "AAI_TEST");
 232                 SideEffectRunner runner = new SideEffectRunner
 233                                 .Builder(spy, serializer).addSideEffect(DataLinkReader.class).build();
 234
 235                 runner.execute(obj, self);
 236
 237                 assertEquals("both properties have been populated in target object", true, obj.getValue("global-route-target").equals("getTargetKey") && obj.getValue("route-target-role").equals("getRoleKey"));
 238                 g.tx().rollback();
 239
 240         }
 241
 242         @Test
 243         public void verifyPropertyPopulationWithV10OnlyPut() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
 244                 final Introspector obj = loader.introspectorFromName("vpn-binding");
 245                 obj.setValue("vpn-id", "getKeyNoLink");
 246                 final Introspector routeTargets = loader.introspectorFromName("route-targets");
 247                 obj.setValue("route-targets", routeTargets.getUnderlyingObject());
 248                 List<Object> targets = routeTargets.getValue("route-target");
 249                 final Introspector routeTargetOne = loader.introspectorFromName("route-target");
 250                 routeTargetOne.setValue("global-route-target", "getTargetKeyNoLink");
 251                 routeTargetOne.setValue("route-target-role", "getRoleKeyNoLink");
 252                 targets.add(routeTargetOne.getUnderlyingObject());
 253                 final Introspector routeTargetTwo = loader.introspectorFromName("route-target");
 254                 routeTargetTwo.setValue("global-route-target", "getTargetKeyNoLink2");
 255                 routeTargetTwo.setValue("route-target-role", "getRoleKeyNoLink2");
 256                 targets.add(routeTargetTwo.getUnderlyingObject());
 257                 TransactionalGraphEngine spy = spy(dbEngine);
 258                 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
 259                 Graph g = graph.newTransaction();
 260                 GraphTraversalSource traversal = g.traversal();
 261                 when(spy.tx()).thenReturn(g);
 262                 when(spy.asAdmin()).thenReturn(adminSpy);
 263                 when(adminSpy.getTraversalSource()).thenReturn(traversal);
 264                 when(spy.tx()).thenReturn(g);
 265                 when(parser.isDependent()).thenReturn(false);
 266                 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
 267                 when(prop.orElse(null)).thenReturn(obj.getURI());
 268                 DBSerializer serializer = new DBSerializer(version, spy, introspectorFactoryType, "AAI_TEST");
 269                 Vertex v = serializer.createNewVertex(obj);
 270                 serializer.serializeToDb(obj, v, parser, obj.getURI(), "testing");
 271                 Vertex routeTargetOneV = traversal.V().has("global-route-target", "getTargetKeyNoLink").next();
 272                 Vertex routeTargetTwoV = traversal.V().has("global-route-target", "getTargetKeyNoLink2").next();
 273
 274                 assertEquals("first route target put has linked", true, routeTargetOneV.property(AAIProperties.LINKED).orElse(false));
 275                 assertEquals("second route target put does not have linked", false, routeTargetTwoV.property(AAIProperties.LINKED).orElse(false));
 276
 277                 g.tx().rollback();
 278
 279         }
 280 }