2 * ============LICENSE_START=======================================================
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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
20 package org.onap.aai.introspection.sideeffect;
22 import org.janusgraph.core.JanusGraphFactory;
23 import org.janusgraph.core.JanusGraph;
24 import org.apache.commons.io.IOUtils;
25 import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversalSource;
26 import org.apache.tinkerpop.gremlin.structure.Graph;
27 import org.apache.tinkerpop.gremlin.structure.Vertex;
28 import org.apache.tinkerpop.gremlin.structure.VertexProperty;
30 import org.junit.rules.ExpectedException;
31 import org.junit.runner.RunWith;
32 import org.junit.runners.Parameterized;
33 import org.mockito.Mock;
34 import org.mockito.MockitoAnnotations;
35 import org.onap.aai.AAISetup;
36 import org.onap.aai.db.props.AAIProperties;
37 import org.onap.aai.dbmap.DBConnectionType;
38 import org.onap.aai.exceptions.AAIException;
39 import org.onap.aai.introspection.*;
41 import org.onap.aai.introspection.sideeffect.exceptions.AAIMissingRequiredPropertyException;
42 import org.onap.aai.parsers.query.QueryParser;
43 import org.onap.aai.serialization.db.DBSerializer;
44 import org.onap.aai.edges.enums.EdgeProperty;
45 import org.onap.aai.serialization.engines.QueryStyle;
46 import org.onap.aai.serialization.engines.JanusGraphDBEngine;
47 import org.onap.aai.serialization.engines.TransactionalGraphEngine;
49 import java.io.FileInputStream;
50 import java.io.IOException;
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;
58 import static org.junit.Assert.assertEquals;
59 import static org.mockito.Mockito.spy;
60 import static org.mockito.Mockito.when;
62 @RunWith(value = Parameterized.class)
64 public class DataCopyTest extends AAISetup{
68 private static JanusGraph graph;
69 private final static ModelType introspectorFactoryType = ModelType.MOXY;
70 private final static DBConnectionType type = DBConnectionType.REALTIME;
71 private static Loader loader;
72 private static TransactionalGraphEngine dbEngine;
73 @Mock private Vertex self;
74 @Mock private VertexProperty<String> prop;
75 @Mock private QueryParser uriQuery;
76 @Rule public ExpectedException thrown = ExpectedException.none();
80 @Parameterized.Parameter(value = 0)
81 public QueryStyle queryStyle;
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}
93 public static void setup() throws NoSuchFieldException, SecurityException, Exception {
94 graph = JanusGraphFactory.build().set("storage.backend","inmemory").open();
95 System.setProperty("AJSC_HOME", ".");
96 System.setProperty("BUNDLECONFIG_DIR", "src/test/resources/bundleconfig-local");
98 graph.traversal().addV("aai-node-type", "model", "model-invariant-id", "key1", AAIProperties.AAI_URI, "/service-design-and-creation/models/model/key1").as("v1")
99 .addV("aai-node-type", "model-ver", "model-ver", "myValue", "model-version-id", "key2", "model-version", "testValue", AAIProperties.AAI_URI, "/service-design-and-creation/models/model/key1/model-vers/model-ver/key2")
100 .addOutE("org.onap.relationships.inventory.BelongsTo", "v1", EdgeProperty.CONTAINS.toString(), true)
101 .addV("aai-node-type", "model", "model-invariant-id", "key3", AAIProperties.AAI_URI, "/service-design-and-creation/models/model/key3").as("v2")
102 .addV("aai-node-type", "model-ver", "model-ver", "myValue", "model-version-id", "key4", AAIProperties.AAI_URI, "/service-design-and-creation/models/model/key3/model-vers/model-ver/key4")
103 .addOutE("org.onap.relationships.inventory.BelongsTo", "v2", EdgeProperty.CONTAINS.toString(), true)
109 public static void tearDown() {
110 graph.tx().rollback();
115 public void initMock() {
116 loader = loaderFactory.createLoaderForVersion(introspectorFactoryType, schemaVersions.getDefaultVersion());
117 MockitoAnnotations.initMocks(this);
118 dbEngine = new JanusGraphDBEngine(
125 public void runPopulatePersonaModelVer() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
127 final Loader loader = loaderFactory.createLoaderForVersion(ModelType.MOXY, schemaVersions.getDefaultVersion());
128 final Introspector obj = loader.introspectorFromName("generic-vnf");
129 obj.setValue("vnf-id", "myId");
130 obj.setValue("model-invariant-id", "key1");
131 obj.setValue("model-version-id", "key2");
132 TransactionalGraphEngine spy = spy(dbEngine);
133 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
134 Graph g = graph.newTransaction();
135 GraphTraversalSource traversal = g.traversal();
136 when(spy.asAdmin()).thenReturn(adminSpy);
137 when(adminSpy.getTraversalSource()).thenReturn(traversal);
138 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
139 when(prop.orElse(null)).thenReturn(obj.getURI());
140 DBSerializer serializer = new DBSerializer(schemaVersions.getDefaultVersion(), spy, introspectorFactoryType, "AAI_TEST");
141 SideEffectRunner runner = new SideEffectRunner
142 .Builder(spy, serializer).addSideEffect(DataCopy.class).build();
144 runner.execute(obj, self);
146 assertEquals("value populated", "testValue", obj.getValue("persona-model-version"));
154 public void runPopulateModelVersionId() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
156 final Loader loader = loaderFactory.createLoaderForVersion(ModelType.MOXY, schemaVersions.getDepthVersion());
157 final Introspector obj = loader.introspectorFromName("generic-vnf");
158 obj.setValue("vnf-id", "myId");
159 obj.setValue("persona-model-id", "key1");
160 obj.setValue("persona-model-version", "testValue");
161 TransactionalGraphEngine spy = spy(dbEngine);
162 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
163 Graph g = graph.newTransaction();
164 GraphTraversalSource traversal = g.traversal();
165 when(spy.asAdmin()).thenReturn(adminSpy);
166 when(adminSpy.getTraversalSource()).thenReturn(traversal);
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(DataCopy.class).build();
173 runner.execute(obj, self);
175 assertEquals("value populated", "key2", obj.getValue("model-version-id"));
181 public void verifyNestedSideEffect() throws URISyntaxException, AAIException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, IOException {
183 final Loader loader = loaderFactory.createLoaderForVersion(ModelType.MOXY, schemaVersions.getDefaultVersion());
184 final Introspector obj = loader.unmarshal("customer", this.getJsonString("nested-case.json"));
185 //System.out.println(obj.marshal(true));
186 TransactionalGraphEngine spy = spy(dbEngine);
187 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
188 Graph g = graph.newTransaction();
189 GraphTraversalSource traversal = g.traversal();
190 when(spy.tx()).thenReturn(g);
191 when(spy.asAdmin()).thenReturn(adminSpy);
192 when(adminSpy.getTraversalSource()).thenReturn(traversal);
193 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
194 when(prop.orElse(null)).thenReturn(obj.getURI());
195 when(uriQuery.isDependent()).thenReturn(false);
196 DBSerializer serializer = new DBSerializer(schemaVersions.getDefaultVersion(), spy, introspectorFactoryType, "AAI_TEST");
197 Vertex v= serializer.createNewVertex(obj);
198 serializer.serializeToDb(obj, v, uriQuery, obj.getURI(), "test");
200 assertEquals("nested value populated", "testValue", g.traversal().V().has("service-instance-id", "nested-instance-key").next().property("persona-model-version").orElse(""));
207 public void expectedMissingPropertyExceptionInURI() throws AAIException, UnsupportedEncodingException {
209 final Loader loader = loaderFactory.createLoaderForVersion(ModelType.MOXY, schemaVersions.getDefaultVersion());
210 final Introspector obj = loader.introspectorFromName("generic-vnf");
211 obj.setValue("vnf-id", "myId");
212 obj.setValue("model-invariant-id", "key1");
214 TransactionalGraphEngine spy = spy(dbEngine);
215 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
216 Graph g = graph.newTransaction();
217 GraphTraversalSource traversal = g.traversal();
218 when(spy.asAdmin()).thenReturn(adminSpy);
219 when(adminSpy.getTraversalSource()).thenReturn(traversal);
220 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
221 when(prop.orElse(null)).thenReturn(obj.getURI());
222 DBSerializer serializer = new DBSerializer(schemaVersions.getDefaultVersion(), spy, introspectorFactoryType, "AAI_TEST");
223 SideEffectRunner runner = new SideEffectRunner
224 .Builder(spy, serializer).addSideEffect(DataCopy.class).build();
226 thrown.expect(AAIMissingRequiredPropertyException.class);
227 runner.execute(obj, self);
231 public void expectedMissingPropertyExceptionForResultingObject() throws AAIException, UnsupportedEncodingException {
232 final Loader loader = loaderFactory.createLoaderForVersion(ModelType.MOXY, schemaVersions.getDefaultVersion());
233 final Introspector obj = loader.introspectorFromName("generic-vnf");
234 obj.setValue("vnf-id", "myId");
235 obj.setValue("model-invariant-id", "key3");
236 obj.setValue("model-version-id", "key4");
238 TransactionalGraphEngine spy = spy(dbEngine);
239 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
240 Graph g = graph.newTransaction();
241 GraphTraversalSource traversal = g.traversal();
242 when(spy.asAdmin()).thenReturn(adminSpy);
243 when(adminSpy.getTraversalSource()).thenReturn(traversal);
244 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
245 when(prop.orElse(null)).thenReturn(obj.getURI());
246 DBSerializer serializer = new DBSerializer(schemaVersions.getDefaultVersion(), spy, introspectorFactoryType, "AAI_TEST");
247 SideEffectRunner runner = new SideEffectRunner
248 .Builder(spy, serializer).addSideEffect(DataCopy.class).build();
250 thrown.expect(AAIMissingRequiredPropertyException.class);
251 runner.execute(obj, self);
255 public void expectNoProcessingWithNoProperties() throws AAIException, UnsupportedEncodingException {
256 final Loader loader = loaderFactory.createLoaderForVersion(ModelType.MOXY, schemaVersions.getDefaultVersion());
257 final Introspector obj = loader.introspectorFromName("generic-vnf");
258 obj.setValue("vnf-id", "myId");
260 TransactionalGraphEngine spy = spy(dbEngine);
261 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
262 Graph g = graph.newTransaction();
263 GraphTraversalSource traversal = g.traversal();
264 when(spy.asAdmin()).thenReturn(adminSpy);
265 when(adminSpy.getTraversalSource()).thenReturn(traversal);
266 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
267 when(prop.orElse(null)).thenReturn(obj.getURI());
268 DBSerializer serializer = new DBSerializer(schemaVersions.getDefaultVersion(), spy, introspectorFactoryType, "AAI_TEST");
269 SideEffectRunner runner = new SideEffectRunner
270 .Builder(spy, serializer).addSideEffect(DataCopy.class).build();
272 runner.execute(obj, self);
274 assertEquals("no model-version-id", true, obj.getValue("model-version-id") == null);
275 assertEquals("no model-invariant-id", true, obj.getValue("model-invariant-id") == null);
279 private String getJsonString(String filename) throws IOException {
282 FileInputStream is = new FileInputStream("src/test/resources/oxm/sideeffect/" + filename);
283 String s = IOUtils.toString(is, "UTF-8");
284 IOUtils.closeQuietly(is);