1 files changed, 145 insertions, 0 deletions
diff --git a/src/test/java/org/onap/aai/migration/v12/DeletePInterfaceTest.java b/src/test/java/org/onap/aai/migration/v12/DeletePInterfaceTest.java
new file mode 100644
index 0000000..665000f
--- /dev/null
+++ b/src/test/java/org/onap/aai/migration/v12/DeletePInterfaceTest.java
@@ -0,0 +1,145 @@
+/**
+ * ============LICENSE_START=======================================================
+ * org.onap.aai
+ * ================================================================================
+ * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+package org.onap.aai.migration.v12;
+
+import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversalSource;
+import org.apache.tinkerpop.gremlin.structure.Vertex;
+import org.janusgraph.core.JanusGraph;
+import org.janusgraph.core.JanusGraphFactory;
+import org.janusgraph.core.JanusGraphTransaction;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.onap.aai.AAISetup;
+import org.onap.aai.dbmap.DBConnectionType;
+import org.onap.aai.introspection.Loader;
+import org.onap.aai.introspection.ModelType;
+import org.onap.aai.migration.Status;
+import org.onap.aai.serialization.engines.JanusGraphDBEngine;
+import org.onap.aai.serialization.engines.QueryStyle;
+import org.onap.aai.serialization.engines.TransactionalGraphEngine;
+import org.onap.aai.setup.SchemaVersions;
+import org.onap.aai.setup.SchemaVersion;
+
+import static org.junit.Assert.assertEquals;
+import static org.mockito.Mockito.spy;
+import static org.mockito.Mockito.when;
+
+public class DeletePInterfaceTest extends AAISetup {
+
+	private final static ModelType introspectorFactoryType = ModelType.MOXY;
+	private final static QueryStyle queryStyle = QueryStyle.TRAVERSAL;
+	private final static DBConnectionType type = DBConnectionType.REALTIME;
+	private Loader loader;
+	private TransactionalGraphEngine dbEngine;
+	private JanusGraph graph;
+	private DeletePInterface migration;
+	private GraphTraversalSource g;
+	private JanusGraphTransaction tx;
+
+	@Before
+	public void setUp() throws Exception {
+		graph = JanusGraphFactory.build().set("storage.backend","inmemory").open();
+		tx = graph.newTransaction();
+		g = tx.traversal();
+		loader = loaderFactory.createLoaderForVersion(introspectorFactoryType, schemaVersions.getDefaultVersion());
+		System.setProperty("AJSC_HOME", ".");
+		System.setProperty("BUNDLECONFIG_DIR", "src/test/resources");
+		dbEngine = new JanusGraphDBEngine(
+				queryStyle,
+				type,
+				loader);
+
+		Vertex pnf1 = g.addV().property("aai-node-type", "pnf")
+			.property("pnf-name", "pnf-name1")
+			.next();
+		Vertex pInterface1 = g.addV().property("aai-node-type", "p-interface")
+				.property("interface-name", "interface-name1")
+				.property("source-of-truth", "AAI-CSVP-INSTARAMS")
+				.next();
+		edgeSerializer.addTreeEdge(g, pnf1, pInterface1);
+		
+		Vertex pnf2 = g.addV().property("aai-node-type", "pnf")
+				.property("pnf-name", "pnf-name2")
+				.next();
+		Vertex pInterface2 = g.addV().property("aai-node-type", "p-interface")
+				.property("interface-name", "interface-name2")
+				.property("source-of-truth", "AAI-CSVP-INSTARAMS")
+				.next();
+		Vertex pLink = g.addV().property("aai-node-type", "physical-link")
+				.property("interface-name", "interface-name1")
+				.next();
+		edgeSerializer.addTreeEdge(g, pnf2, pInterface2);
+		edgeSerializer.addEdge(g, pInterface2, pLink);
+		
+		Vertex pnf3 = g.addV().property("aai-node-type", "pnf")
+				.property("pnf-name", "pnf-name3")
+				.next();
+		Vertex pInterface3 = g.addV().property("aai-node-type", "p-interface")
+				.property("interface-name", "interface-name3")
+				.property("source-of-truth", "AAI-CSVP-INSTARAMS")
+				.next();
+		Vertex lInterface = g.addV().property("aai-node-type", "l-interface")
+				.property("interface-name", "interface-name3")
+				.next();
+		edgeSerializer.addTreeEdge(g, pnf3, pInterface3);
+		edgeSerializer.addTreeEdge(g, pInterface3, lInterface);
+		
+		Vertex pnf4 = g.addV().property("aai-node-type", "pnf")
+				.property("pnf-name", "pnf-name4")
+				.next();
+		Vertex pInterface4 = g.addV().property("aai-node-type", "p-interface")
+				.property("interface-name", "interface-name4")
+				.next();
+		edgeSerializer.addTreeEdge(g, pnf4, pInterface4);
+		
+		TransactionalGraphEngine spy = spy(dbEngine);
+		TransactionalGraphEngine.Admin adminSpy = spy(dbEngine.asAdmin());
+		GraphTraversalSource traversal = g;
+		when(spy.asAdmin()).thenReturn(adminSpy);
+		when(adminSpy.getTraversalSource()).thenReturn(traversal);
+		migration = new DeletePInterface(spy, loaderFactory, edgeIngestor, edgeSerializer, schemaVersions);
+		migration.run();
+	}
+
+	@After
+	public void tearDown() throws Exception {
+		tx.rollback();
+		graph.close();
+	}
+
+	@Test
+	public void test() {
+		assertEquals("pInterface1 deleted", false, g.V().has("aai-node-type", "pnf").has("pnf-name", "pnf-name1")
+				.in("tosca.relationships.network.BindsTo").has("aai-node-type", "p-interface").has("interface-name", "interface-name1").hasNext());
+		
+		assertEquals("pInterface2 skipped", true, g.V().has("aai-node-type", "pnf").has("pnf-name", "pnf-name2")
+				.in("tosca.relationships.network.BindsTo").has("aai-node-type", "p-interface").hasNext());
+		
+		assertEquals("pInterface3 skipped", true, g.V().has("aai-node-type", "pnf").has("pnf-name", "pnf-name3")
+				.in("tosca.relationships.network.BindsTo").has("aai-node-type", "p-interface").hasNext());
+		
+		assertEquals("pInterface4 should not be deleted", true, g.V().has("aai-node-type", "pnf").has("pnf-name", "pnf-name4")
+				.in("tosca.relationships.network.BindsTo").has("aai-node-type", "p-interface").has("interface-name", "interface-name4").hasNext());
+		
+		assertEquals("Status should be success", Status.SUCCESS, migration.getStatus());
+	}
+
+}