1 files changed, 6 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 0349985608..46326c350c 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
@@ -20,7 +20,11 @@
 
 package org.openecomp.sdc.be.components.impl;
 
+import static org.mockito.Mockito.when;
+
 import fj.data.Either;
+import java.util.HashMap;
+import java.util.Map;
 import junit.framework.Assert;
 import org.junit.Before;
 import org.junit.Test;
@@ -30,11 +34,6 @@ import org.openecomp.sdc.be.impl.ComponentsUtils;
 import org.openecomp.sdc.be.model.DataTypeDefinition;
 import org.openecomp.sdc.be.model.cache.ApplicationDataTypeCache;
 
-import java.util.HashMap;
-import java.util.Map;
-
-import static org.mockito.Mockito.when;
-
 public class DataTypesServiceTest {
     ApplicationDataTypeCache applicationDataTypeCache = Mockito.mock(ApplicationDataTypeCache.class);
     ComponentsUtils componentsUtils = Mockito.mock(ComponentsUtils.class);
@@ -48,7 +47,7 @@ public class DataTypesServiceTest {
     public void setup() {
         mapreturn.put("Demo",new DataTypeDefinition());
         allDataTypes = Either.left(mapreturn);
-        when(applicationDataTypeCache.getAll()).thenReturn(allDataTypes);
+        when(applicationDataTypeCache.getAll(null)).thenReturn(allDataTypes);
 
     }
 
@@ -60,7 +59,7 @@ public class DataTypesServiceTest {
     @Test
     public void getAllDataTypes_failure() {
         allDataTypes = Either.right(janusGraphOperationStatus);
-        when(applicationDataTypeCache.getAll()).thenReturn(allDataTypes);
+        when(applicationDataTypeCache.getAll(null)).thenReturn(allDataTypes);
         Assert.assertEquals(true,dataTypesService.getAllDataTypes(applicationDataTypeCache).isRight());
     }