1 files changed, 5 insertions, 7 deletions

diff --git a/catalog-be/src/test/java/org/openecomp/sdc/be/components/impl/DataTypesServiceTest.java b/catalog-be/src/test/java/org/openecomp/sdc/be/components/impl/DataTypesServiceTest.java
index 5cf3ef3a95..f53d2f9094 100644
--- a/catalog-be/src/test/java/org/openecomp/sdc/be/components/impl/DataTypesServiceTest.java
+++ b/catalog-be/src/test/java/org/openecomp/sdc/be/components/impl/DataTypesServiceTest.java
@@ -4,14 +4,12 @@ import fj.data.Either;
 import junit.framework.Assert;
 import org.junit.Before;
 import org.junit.Test;
-import static org.junit.Assert.*;
+
 import static org.mockito.ArgumentMatchers.any;
-import static org.mockito.ArgumentMatchers.anyObject;
-import static org.mockito.ArgumentMatchers.anyString;
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.when;
 import org.mockito.Mockito;
-import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
+import org.openecomp.sdc.be.dao.janusgraph.JanusGraphOperationStatus;
 import org.openecomp.sdc.be.impl.ComponentsUtils;
 import org.openecomp.sdc.be.model.DataTypeDefinition;
 import org.openecomp.sdc.be.model.cache.ApplicationDataTypeCache;
@@ -26,8 +24,8 @@ public class DataTypesServiceTest {
 
     DataTypesService dataTypesService = new DataTypesService(componentsUtils);
     Map<String, DataTypeDefinition> mapreturn = new HashMap<>();
-    TitanOperationStatus titanOperationStatus = TitanOperationStatus.NOT_FOUND;
-    Either<Map<String, DataTypeDefinition>, TitanOperationStatus> allDataTypes;
+    JanusGraphOperationStatus janusGraphOperationStatus = JanusGraphOperationStatus.NOT_FOUND;
+    Either<Map<String, DataTypeDefinition>, JanusGraphOperationStatus> allDataTypes;
 
     @Before
     public void setup() {
@@ -44,7 +42,7 @@ public class DataTypesServiceTest {
 
     @Test
     public void getAllDataTypes_failure() {
-        allDataTypes = Either.right(titanOperationStatus);
+        allDataTypes = Either.right(janusGraphOperationStatus);
         when(applicationDataTypeCache.getAll()).thenReturn(allDataTypes);
         Assert.assertEquals(true,dataTypesService.getAllDataTypes(applicationDataTypeCache).isRight());
     }