aai-core/src/test/java/org/openecomp/aai/introspection/sideeffect/DataCopyTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * org.openecomp.aai
   4  * ================================================================================
   5  * Copyright (C) 2017 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
  21 package org.openecomp.aai.introspection.sideeffect;
  22
  23 import static org.junit.Assert.assertEquals;
  24 import static org.mockito.Mockito.spy;
  25 import static org.mockito.Mockito.when;
  26
  27 import java.io.FileInputStream;
  28 import java.io.IOException;
  29 import java.io.UnsupportedEncodingException;
  30 import java.lang.reflect.InvocationTargetException;
  31 import java.net.MalformedURLException;
  32 import java.net.URISyntaxException;
  33
  34 import org.apache.commons.io.IOUtils;
  35 import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversalSource;
  36 import org.apache.tinkerpop.gremlin.structure.Vertex;
  37 import org.apache.tinkerpop.gremlin.structure.VertexProperty;
  38 import org.junit.AfterClass;
  39 import org.junit.Before;
  40 import org.junit.BeforeClass;
  41 import org.junit.Rule;
  42 import org.junit.Test;
  43 import org.junit.rules.ExpectedException;
  44 import org.mockito.Mock;
  45 import org.mockito.MockitoAnnotations;
  46
  47 import org.openecomp.aai.db.props.AAIProperties;
  48 import org.openecomp.aai.dbmap.DBConnectionType;
  49 import org.openecomp.aai.exceptions.AAIException;
  50 import org.openecomp.aai.introspection.Introspector;
  51 import org.openecomp.aai.introspection.Loader;
  52 import org.openecomp.aai.introspection.LoaderFactory;
  53 import org.openecomp.aai.introspection.ModelType;
  54 import org.openecomp.aai.introspection.Version;
  55 import org.openecomp.aai.introspection.sideeffect.exceptions.AAIMissingRequiredPropertyException;
  56 import org.openecomp.aai.parsers.query.QueryParser;
  57 import org.openecomp.aai.serialization.db.DBSerializer;
  58 import org.openecomp.aai.serialization.engines.QueryStyle;
  59 import org.openecomp.aai.serialization.engines.TitanDBEngine;
  60 import org.openecomp.aai.serialization.engines.TransactionalGraphEngine;
  61 import org.openecomp.aai.serialization.queryformats.QueryFormatTestHelper;
  62 import org.openecomp.aai.util.AAIConstants;
  63 import com.thinkaurelius.titan.core.TitanFactory;
  64 import com.thinkaurelius.titan.core.TitanGraph;
  65 import com.thinkaurelius.titan.core.TitanTransaction;
  66
  67 public class DataCopyTest {
  68
  69         private static TitanGraph graph;
  70         private final static Version version = Version.v10;
  71         private final static ModelType introspectorFactoryType = ModelType.MOXY;
  72         private final static QueryStyle queryStyle = QueryStyle.TRAVERSAL;
  73         private final static DBConnectionType type = DBConnectionType.REALTIME;
  74         private static Loader loader;
  75         private static TransactionalGraphEngine dbEngine;
  76         @Mock private Vertex self;
  77         @Mock private VertexProperty<String> prop;
  78         @Mock private QueryParser uriQuery;
  79         @Rule
  80         public ExpectedException thrown = ExpectedException.none();
  81
  82
  83         @BeforeClass
  84         public static void setup() throws NoSuchFieldException, SecurityException, Exception {
  85                 graph = TitanFactory.build().set("storage.backend","inmemory").open();
  86                 System.setProperty("AJSC_HOME", "./src/test/resources/");
  87                 System.setProperty("BUNDLECONFIG_DIR", "bundleconfig-local");
  88                 QueryFormatTestHelper.setFinalStatic(AAIConstants.class.getField("AAI_HOME_ETC_OXM"), "src/test/resources/org/openecomp/aai/introspection/");
  89                 loader = LoaderFactory.createLoaderForVersion(introspectorFactoryType, version);
  90                 dbEngine = new TitanDBEngine(
  91                                 queryStyle,
  92                                 type,
  93                                 loader);
  94
  95                 graph.traversal().addV("aai-node-type", "model", "model-invariant-id", "key1").as("v1")
  96                 .addV("aai-node-type", "model-ver", "model-ver", "myValue", "model-version-id", "key2", "model-version", "testValue").addInE("has", "v1", "isParent", true)
  97                 .addV("aai-node-type", "model", "model-invariant-id", "key3").as("v2")
  98                 .addV("aai-node-type", "model-ver", "model-ver", "myValue", "model-version-id", "key4").addInE("has", "v2", "isParent", true)
  99                 .next();
 100                 graph.tx().commit();
 101         }
 102
 103         @AfterClass
 104         public static void tearDown() {
 105                 graph.tx().rollback();
 106                 graph.close();
 107         }
 108
 109         @Before
 110         public void initMock() {
 111                 MockitoAnnotations.initMocks(this);
 112         }
 113
 114         @Test
 115         public void runPopulatePersonaModelVer() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
 116
 117                 final Loader loader = LoaderFactory.createLoaderForVersion(ModelType.MOXY, Version.v10);
 118                 final Introspector obj = loader.introspectorFromName("test-object");
 119                 obj.setValue("vnf-id", "myId");
 120                 obj.setValue("model-invariant-id", "key1");
 121                 obj.setValue("model-version-id", "key2");
 122                 TransactionalGraphEngine spy = spy(dbEngine);
 123                 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
 124                 TitanTransaction g = graph.newTransaction();
 125                 GraphTraversalSource traversal = g.traversal();
 126                 when(spy.asAdmin()).thenReturn(adminSpy);
 127                 when(adminSpy.getTraversalSource()).thenReturn(traversal);
 128                 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
 129                 when(prop.orElse(null)).thenReturn(obj.getURI());
 130                 DBSerializer serializer = new DBSerializer(version, spy, introspectorFactoryType, "AAI_TEST");
 131                 SideEffectRunner runner = new SideEffectRunner
 132                                 .Builder(spy, serializer).addSideEffect(DataCopy.class).build();
 133
 134                 runner.execute(obj, self);
 135
 136                 assertEquals("value populated", "testValue", obj.getValue("persona-model-ver"));
 137
 138                 g.rollback();
 139
 140
 141         }
 142
 143         @Test
 144         public void runPopulateModelVersionId() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
 145
 146                 final Loader loader = LoaderFactory.createLoaderForVersion(ModelType.MOXY, Version.v9);
 147                 final Introspector obj = loader.introspectorFromName("test-object");
 148                 obj.setValue("vnf-id", "myId");
 149                 obj.setValue("persona-model-id", "key1");
 150                 obj.setValue("persona-model-version", "testValue");
 151                 TransactionalGraphEngine spy = spy(dbEngine);
 152                 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
 153                 TitanTransaction g = graph.newTransaction();
 154                 GraphTraversalSource traversal = g.traversal();
 155                 when(spy.asAdmin()).thenReturn(adminSpy);
 156                 when(adminSpy.getTraversalSource()).thenReturn(traversal);
 157                 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
 158                 when(prop.orElse(null)).thenReturn(obj.getURI());
 159                 DBSerializer serializer = new DBSerializer(version, spy, introspectorFactoryType, "AAI_TEST");
 160                 SideEffectRunner runner = new SideEffectRunner
 161                                 .Builder(spy, serializer).addSideEffect(DataCopy.class).build();
 162
 163                 runner.execute(obj, self);
 164
 165                 assertEquals("value populated", "key2", obj.getValue("model-version-id"));
 166
 167                 g.rollback();
 168         }
 169
 170         @Test
 171         public void verifyNestedSideEffect() throws URISyntaxException, AAIException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, IOException {
 172
 173                 final Loader loader = LoaderFactory.createLoaderForVersion(ModelType.MOXY, Version.v10);
 174                 final Introspector obj = loader.unmarshal("customer", this.getJsonString("nested-case.json"));
 175                 System.out.println(obj.marshal(true));
 176                 TransactionalGraphEngine spy = spy(dbEngine);
 177                 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
 178                 TitanTransaction g = graph.newTransaction();
 179                 GraphTraversalSource traversal = g.traversal();
 180                 when(spy.tx()).thenReturn(g);
 181                 when(spy.asAdmin()).thenReturn(adminSpy);
 182                 when(adminSpy.getTraversalSource()).thenReturn(traversal);
 183                 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
 184                 when(prop.orElse(null)).thenReturn(obj.getURI());
 185                 when(uriQuery.isDependent()).thenReturn(false);
 186                 DBSerializer serializer = new DBSerializer(version, spy, introspectorFactoryType, "AAI_TEST");
 187                 Vertex v= serializer.createNewVertex(obj);
 188                 serializer.serializeToDb(obj, v, uriQuery, obj.getURI(), "test");
 189
 190                 assertEquals("nested value populated", "testValue", g.traversal().V().has("service-instance-id", "nested-instance-key").next().property("persona-model-version").orElse(""));
 191
 192                 g.rollback();
 193
 194         }
 195
 196         @Test
 197         public void expectedMissingPropertyExceptionInURI() throws AAIException, UnsupportedEncodingException {
 198
 199                 final Loader loader = LoaderFactory.createLoaderForVersion(ModelType.MOXY, Version.v10);
 200                 final Introspector obj = loader.introspectorFromName("test-object");
 201                 obj.setValue("vnf-id", "myId");
 202                 obj.setValue("model-invariant-id", "key1");
 203
 204                 TransactionalGraphEngine spy = spy(dbEngine);
 205                 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
 206                 TitanTransaction g = graph.newTransaction();
 207                 GraphTraversalSource traversal = g.traversal();
 208                 when(spy.asAdmin()).thenReturn(adminSpy);
 209                 when(adminSpy.getTraversalSource()).thenReturn(traversal);
 210                 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
 211                 when(prop.orElse(null)).thenReturn(obj.getURI());
 212                 DBSerializer serializer = new DBSerializer(version, spy, introspectorFactoryType, "AAI_TEST");
 213                 SideEffectRunner runner = new SideEffectRunner
 214                                 .Builder(spy, serializer).addSideEffect(DataCopy.class).build();
 215
 216                 thrown.expect(AAIMissingRequiredPropertyException.class);
 217                 runner.execute(obj, self);
 218         }
 219
 220         @Test
 221         public void expectedMissingPropertyExceptionForResultingObject() throws AAIException, UnsupportedEncodingException {
 222                 final Loader loader = LoaderFactory.createLoaderForVersion(ModelType.MOXY, Version.v10);
 223                 final Introspector obj = loader.introspectorFromName("test-object");
 224                 obj.setValue("vnf-id", "myId");
 225                 obj.setValue("model-invariant-id", "key3");
 226                 obj.setValue("model-version-id", "key4");
 227
 228                 TransactionalGraphEngine spy = spy(dbEngine);
 229                 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
 230                 TitanTransaction g = graph.newTransaction();
 231                 GraphTraversalSource traversal = g.traversal();
 232                 when(spy.asAdmin()).thenReturn(adminSpy);
 233                 when(adminSpy.getTraversalSource()).thenReturn(traversal);
 234                 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
 235                 when(prop.orElse(null)).thenReturn(obj.getURI());
 236                 DBSerializer serializer = new DBSerializer(version, spy, introspectorFactoryType, "AAI_TEST");
 237                 SideEffectRunner runner = new SideEffectRunner
 238                                 .Builder(spy, serializer).addSideEffect(DataCopy.class).build();
 239
 240                 thrown.expect(AAIMissingRequiredPropertyException.class);
 241                 runner.execute(obj, self);
 242         }
 243
 244         @Test
 245         public void expectNoProcessingWithNoProperties() throws AAIException, UnsupportedEncodingException {
 246                 final Loader loader = LoaderFactory.createLoaderForVersion(ModelType.MOXY, Version.v10);
 247                 final Introspector obj = loader.introspectorFromName("test-object");
 248                 obj.setValue("vnf-id", "myId");
 249
 250                 TransactionalGraphEngine spy = spy(dbEngine);
 251                 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
 252                 TitanTransaction g = graph.newTransaction();
 253                 GraphTraversalSource traversal = g.traversal();
 254                 when(spy.asAdmin()).thenReturn(adminSpy);
 255                 when(adminSpy.getTraversalSource()).thenReturn(traversal);
 256                 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
 257                 when(prop.orElse(null)).thenReturn(obj.getURI());
 258                 DBSerializer serializer = new DBSerializer(version, spy, introspectorFactoryType, "AAI_TEST");
 259                 SideEffectRunner runner = new SideEffectRunner
 260                                 .Builder(spy, serializer).addSideEffect(DataCopy.class).build();
 261
 262                 runner.execute(obj, self);
 263
 264                 assertEquals("no model-version-id", true, obj.getValue("model-version-id") == null);
 265                 assertEquals("no model-invariant-id", true, obj.getValue("model-invariant-id") == null);
 266
 267         }
 268
 269         private String getJsonString(String filename) throws IOException {
 270
 271
 272                 FileInputStream is = new FileInputStream("src/test/resources/org/openecomp/aai/introspection/sideeffect/" + filename);
 273                 String s =  IOUtils.toString(is, "UTF-8");
 274                 IOUtils.closeQuietly(is);
 275
 276                 return s;
 277         }
 278 }