1 files changed, 8 insertions, 4 deletions

diff --git a/engine-d/src/test/java/org/onap/holmes/engine/manager/DroolsEngineTest.java b/engine-d/src/test/java/org/onap/holmes/engine/manager/DroolsEngineTest.java
index 8da4910..f7b7fbd 100644
--- a/engine-d/src/test/java/org/onap/holmes/engine/manager/DroolsEngineTest.java
+++ b/engine-d/src/test/java/org/onap/holmes/engine/manager/DroolsEngineTest.java
@@ -23,6 +23,7 @@ import org.junit.rules.ExpectedException;
 import org.onap.holmes.common.api.entity.AlarmInfo;

 import org.onap.holmes.common.api.entity.CorrelationRule;

 import org.onap.holmes.common.api.stat.VesAlarm;

+import org.onap.holmes.common.dmaap.store.ClosedLoopControlNameCache;

 import org.onap.holmes.common.exception.CorrelationException;

 import org.onap.holmes.common.utils.DbDaoUtil;

 import org.onap.holmes.engine.db.AlarmInfoDao;

@@ -47,18 +48,21 @@ public class DroolsEngineTest {
 

     private RuleMgtWrapper ruleMgtWrapper;

 

-    private AlarmInfoDao alarmInfoDaoMock;

-

     private DroolsEngine droolsEngine;

 

     private DbDaoUtil dbDaoUtilStub;

 

+    private ClosedLoopControlNameCache closedLoopControlNameCache;

+

     public DroolsEngineTest() throws Exception {

         droolsEngine = new DroolsEngine();

         ruleMgtWrapper = new RuleMgtWrapperStub();

         dbDaoUtilStub = new DbDaoUtilStub();

-        Whitebox.setInternalState(droolsEngine, "daoUtil", dbDaoUtilStub);

-        Whitebox.setInternalState(droolsEngine, "ruleMgtWrapper", ruleMgtWrapper);

+        closedLoopControlNameCache = new ClosedLoopControlNameCache();

+        droolsEngine.setClosedLoopControlNameCache(closedLoopControlNameCache);

+        droolsEngine.setDaoUtil(dbDaoUtilStub);

+        droolsEngine.setRuleMgtWrapper(ruleMgtWrapper);

+

         Whitebox.invokeMethod(droolsEngine, "init");

     }