1 files changed, 12 insertions, 5 deletions

diff --git a/holmes-actions/src/test/java/org/onap/holmes/common/engine/service/impl/EngineEntityServiceImplTest.java b/holmes-actions/src/test/java/org/onap/holmes/common/engine/service/impl/EngineEntityServiceImplTest.java
index 7058943..21a4025 100644
--- a/holmes-actions/src/test/java/org/onap/holmes/common/engine/service/impl/EngineEntityServiceImplTest.java
+++ b/holmes-actions/src/test/java/org/onap/holmes/common/engine/service/impl/EngineEntityServiceImplTest.java
@@ -24,10 +24,7 @@ import org.onap.holmes.common.engine.entity.EngineEntity;
 import org.onap.holmes.common.engine.service.EngineEntityService;
 import org.onap.holmes.common.utils.DbDaoUtil;
 
-import java.util.ArrayList;
-import java.util.HashSet;
-import java.util.List;
-import java.util.Set;
+import java.util.*;
 
 import static com.google.common.base.Predicates.notNull;
 import static org.hamcrest.core.Is.is;
@@ -35,10 +32,15 @@ import static org.hamcrest.core.IsNull.notNullValue;
 import static org.junit.Assert.*;
 
 public class EngineEntityServiceImplTest {
-
     private EngineEntityService service = new EngineEntityServiceImpl(new DbDaoUtilStub());
 
     @Test
+    public void getLegacyEngines() {
+        List<String> legacyEngines = service.getLegacyEngines();
+        assertThat(legacyEngines.size(), is(2));
+    }
+
+    @Test
     public void getEntity() throws Exception {
         EngineEntity entity = service.getEntity("org.onap.holmes_9201");
         assertThat(entity, notNullValue());
@@ -103,6 +105,11 @@ class EngineEntityDaoStub implements EngineEntityDao {
     }
 
     @Override
+    public List<String> getLegacyEngines() {
+        return Arrays.asList("org.onap.holmes", "org.onap.holmes.legacy.1");
+    }
+
+    @Override
     public void insertEntity(EngineEntity entity) {
         entitySet.add(entity);
     }