1 files changed, 13 insertions, 11 deletions

diff --git a/catalog-dao/src/test/java/org/openecomp/sdc/be/dao/jsongraph/TitanDaoMockTest.java b/catalog-dao/src/test/java/org/openecomp/sdc/be/dao/jsongraph/TitanDaoMockTest.java
index f31e08e2fb..5703e53c9c 100644
--- a/catalog-dao/src/test/java/org/openecomp/sdc/be/dao/jsongraph/TitanDaoMockTest.java
+++ b/catalog-dao/src/test/java/org/openecomp/sdc/be/dao/jsongraph/TitanDaoMockTest.java
@@ -1,10 +1,9 @@
 package org.openecomp.sdc.be.dao.jsongraph;
 
-import java.util.HashMap;
-import java.util.Iterator;
-import java.util.List;
-import java.util.Map;
-
+import com.thinkaurelius.titan.core.TitanGraph;
+import com.thinkaurelius.titan.core.TitanVertex;
+import fj.data.Either;
+import mockit.Deencapsulation;
 import org.apache.commons.lang3.tuple.ImmutablePair;
 import org.apache.tinkerpop.gremlin.structure.Direction;
 import org.apache.tinkerpop.gremlin.structure.Edge;
@@ -24,11 +23,10 @@ import org.openecomp.sdc.be.dao.titan.TitanGraphClient;
 import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
 import org.openecomp.sdc.be.datatypes.enums.GraphPropertyEnum;
 
-import com.thinkaurelius.titan.core.TitanGraph;
-import com.thinkaurelius.titan.core.TitanVertex;
-
-import fj.data.Either;
-import mockit.Deencapsulation;
+import java.util.HashMap;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
 
 public class TitanDaoMockTest {
 
@@ -223,6 +221,7 @@ public class TitanDaoMockTest {
 		testSubject.parseVertexProperties(graphVertex, JsonParseFlagEnum.NoParse);
 	}
 
+
 	@Test
 	public void testCreateEdge() throws Exception {
 
@@ -294,7 +293,7 @@ public class TitanDaoMockTest {
 		Mockito.when(titanClient.getGraph()).thenReturn(value);
 		
 		// default test
-		result = testSubject.getCatalogVerticies();
+		result = testSubject.getCatalogOrArchiveVerticies(true);
 	}
 
 	@Test
@@ -329,6 +328,7 @@ public class TitanDaoMockTest {
 		result = testSubject.getChildVertex(parentVertex, edgeLabel, parseFlag);
 	}
 
+
 	@Test
 	public void testGetParentVertex_1() throws Exception {
 
@@ -385,6 +385,7 @@ public class TitanDaoMockTest {
 		result = testSubject.getChildrenVertecies(parentVertex, edgeLabel, parseFlag);
 	}
 
+
 	@Test
 	public void testDeleteBelongingEdgeByCriteria() throws Exception {
 
@@ -481,6 +482,7 @@ public class TitanDaoMockTest {
 		result = testSubject.createEdge(from, to, label, edgeToCopy);
 	}
 
+
 	@Test
 	public void testReplaceEdgeLabel() throws Exception {