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 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;
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;
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;
54 import static org.junit.Assert.assertEquals;
55 import static org.mockito.Mockito.spy;
56 import static org.mockito.Mockito.when;
58 @RunWith(value = Parameterized.class)
59 public class DataLinkTest extends AAISetup {
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;
71 public ExpectedException thrown = ExpectedException.none();
73 @Parameterized.Parameter(value = 0)
74 public QueryStyle queryStyle;
76 @Parameterized.Parameters(name = "QueryStyle.{0}")
77 public static Collection<Object[]> data() {
78 return Arrays.asList(new Object[][]{
79 {QueryStyle.TRAVERSAL}
84 public static void setup() throws NoSuchFieldException, SecurityException, Exception {
85 graph = TitanFactory.build().set("storage.backend","inmemory").open();
86 loader = LoaderFactory.createLoaderForVersion(introspectorFactoryType, version);
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)
104 graph.traversal().V().has("aai-uri","/network/vpn-bindings/vpn-binding/deleteKey").properties().forEachRemaining(p->System.out.println(p.key() +" : " + p.value()));
108 public static void tearDown() {
109 graph.tx().rollback();
114 public void initMock() {
115 MockitoAnnotations.initMocks(this);
116 dbEngine = new TitanDBEngine(
123 public void verifyCreationOfVertex() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
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();
143 runner.execute(obj, self);
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());
152 public void verifyModificationOfVertex() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
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();
172 runner.execute(obj, self);
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());
183 public void verifyDeleteOfVertex() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
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();
202 runner.execute(obj, self);
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")
217 public void verifyPropertyPopulation() throws URISyntaxException, AAIException, UnsupportedEncodingException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, SecurityException, InstantiationException, NoSuchMethodException, MalformedURLException {
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();
235 runner.execute(obj, self);
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"));
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();
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));