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 verifyNestedSideEffect() throws URISyntaxException, AAIException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, IOException {
156 final Loader loader = loaderFactory.createLoaderForVersion(ModelType.MOXY, schemaVersions.getDefaultVersion());
157 final Introspector obj = loader.unmarshal("customer", this.getJsonString("nested-case.json"));
158 //System.out.println(obj.marshal(true));
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.tx()).thenReturn(g);
164 when(spy.asAdmin()).thenReturn(adminSpy);
165 when(adminSpy.getTraversalSource()).thenReturn(traversal);
166 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
167 when(prop.orElse(null)).thenReturn(obj.getURI());
168 when(uriQuery.isDependent()).thenReturn(false);
169 DBSerializer serializer = new DBSerializer(schemaVersions.getDefaultVersion(), spy, introspectorFactoryType, "AAI_TEST");
170 Vertex v= serializer.createNewVertex(obj);
171 serializer.serializeToDb(obj, v, uriQuery, obj.getURI(), "test");
173 assertEquals("nested value populated", "testValue", g.traversal().V().has("service-instance-id", "nested-instance-key").next().property("persona-model-version").orElse(""));
180 public void expectedMissingPropertyExceptionInURI() throws AAIException, UnsupportedEncodingException {
182 final Loader loader = loaderFactory.createLoaderForVersion(ModelType.MOXY, schemaVersions.getDefaultVersion());
183 final Introspector obj = loader.introspectorFromName("generic-vnf");
184 obj.setValue("vnf-id", "myId");
185 obj.setValue("model-invariant-id", "key1");
187 TransactionalGraphEngine spy = spy(dbEngine);
188 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
189 Graph g = graph.newTransaction();
190 GraphTraversalSource traversal = g.traversal();
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 DBSerializer serializer = new DBSerializer(schemaVersions.getDefaultVersion(), spy, introspectorFactoryType, "AAI_TEST");
196 SideEffectRunner runner = new SideEffectRunner
197 .Builder(spy, serializer).addSideEffect(DataCopy.class).build();
199 thrown.expect(AAIMissingRequiredPropertyException.class);
200 runner.execute(obj, self);
204 public void expectedMissingPropertyExceptionForResultingObject() throws AAIException, UnsupportedEncodingException {
205 final Loader loader = loaderFactory.createLoaderForVersion(ModelType.MOXY, schemaVersions.getDefaultVersion());
206 final Introspector obj = loader.introspectorFromName("generic-vnf");
207 obj.setValue("vnf-id", "myId");
208 obj.setValue("model-invariant-id", "key3");
209 obj.setValue("model-version-id", "key4");
211 TransactionalGraphEngine spy = spy(dbEngine);
212 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
213 Graph g = graph.newTransaction();
214 GraphTraversalSource traversal = g.traversal();
215 when(spy.asAdmin()).thenReturn(adminSpy);
216 when(adminSpy.getTraversalSource()).thenReturn(traversal);
217 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
218 when(prop.orElse(null)).thenReturn(obj.getURI());
219 DBSerializer serializer = new DBSerializer(schemaVersions.getDefaultVersion(), spy, introspectorFactoryType, "AAI_TEST");
220 SideEffectRunner runner = new SideEffectRunner
221 .Builder(spy, serializer).addSideEffect(DataCopy.class).build();
223 thrown.expect(AAIMissingRequiredPropertyException.class);
224 runner.execute(obj, self);
228 public void expectNoProcessingWithNoProperties() throws AAIException, UnsupportedEncodingException {
229 final Loader loader = loaderFactory.createLoaderForVersion(ModelType.MOXY, schemaVersions.getDefaultVersion());
230 final Introspector obj = loader.introspectorFromName("generic-vnf");
231 obj.setValue("vnf-id", "myId");
233 TransactionalGraphEngine spy = spy(dbEngine);
234 TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
235 Graph g = graph.newTransaction();
236 GraphTraversalSource traversal = g.traversal();
237 when(spy.asAdmin()).thenReturn(adminSpy);
238 when(adminSpy.getTraversalSource()).thenReturn(traversal);
239 when(self.<String>property(AAIProperties.AAI_URI)).thenReturn(prop);
240 when(prop.orElse(null)).thenReturn(obj.getURI());
241 DBSerializer serializer = new DBSerializer(schemaVersions.getDefaultVersion(), spy, introspectorFactoryType, "AAI_TEST");
242 SideEffectRunner runner = new SideEffectRunner
243 .Builder(spy, serializer).addSideEffect(DataCopy.class).build();
245 runner.execute(obj, self);
247 assertEquals("no model-version-id", true, obj.getValue("model-version-id") == null);
248 assertEquals("no model-invariant-id", true, obj.getValue("model-invariant-id") == null);
252 private String getJsonString(String filename) throws IOException {
255 FileInputStream is = new FileInputStream("src/test/resources/oxm/sideeffect/" + filename);
256 String s = IOUtils.toString(is, "UTF-8");
257 IOUtils.closeQuietly(is);