Fix import issue on unit test

[aai/gizmo.git] / src / test / java / org / openecomp / crud / dao / champ / ChampDaoTest.java

diff --git a/src/test/java/org/openecomp/crud/dao/champ/ChampDaoTest.java b/src/test/java/org/openecomp/crud/dao/champ/ChampDaoTest.java
index b11e274..847e67b 100644 (file)
--- a/src/test/java/org/openecomp/crud/dao/champ/ChampDaoTest.java
+++ b/src/test/java/org/openecomp/crud/dao/champ/ChampDaoTest.java
@@ -4,6 +4,7 @@ import org.junit.After;
 import org.junit.Before;
 import org.junit.Ignore;
 import org.junit.Test;
+import org.openecomp.aai.champcore.graph.impl.InMemoryChampGraphImpl;
 import org.openecomp.crud.dao.GraphDao;
 import org.openecomp.crud.entity.Edge;
 import org.openecomp.crud.entity.Vertex;
@@ -12,7 +13,6 @@ import org.openecomp.crud.exception.CrudException;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
-import java.util.Properties;
 
 import static org.junit.Assert.*;
 
@@ -35,10 +35,10 @@ public class ChampDaoTest {
 
     // Create an instance of the Champ DAO, backed by the Champ library's in-memory back end
     // for testing purposes.
-    Properties champDaoProperties = new Properties();
+         Map<String, Object> champDaoProperties = new HashMap<String, Object>();
     champDaoProperties.put(ChampDao.CONFIG_STORAGE_BACKEND, "in-memory");
     champDaoProperties.put(ChampDao.CONFIG_GRAPH_NAME, GRAPH_NAME);
-    champDao = new ChampDao(champDaoProperties);
+    champDao = new ChampDao(new InMemoryChampGraphImpl.Builder().properties(champDaoProperties).build());
   }
 
 
@@ -538,8 +538,8 @@ public class ChampDaoTest {
     // updated edge.
     // *** We will leave this validation commented out for now, as the Champ library actually
     //     merges update properties instead of replacing them...
-//             assertFalse("Property 'key2' should no longer be associated with updated edge",
-//                                 updatedEdge.getProperties().containsKey("key2"));
+               assertFalse("Property 'key2' should no longer be associated with updated edge",
+                                   updatedEdge.getProperties().containsKey("key2"));
   }
 
   @Test