X-Git-Url: https://gerrit.onap.org/r/gitweb?p=holmes%2Fengine-management.git;a=blobdiff_plain;f=engine-d%2Fsrc%2Ftest%2Fjava%2Forg%2Fonap%2Fholmes%2Fengine%2Fmanager%2FDroolsEngineTest.java;h=f7b7fbd5d6bf0484c56734680408dcd77e8f9851;hp=8da4910d332219b64eccc21248691a8960cf2612;hb=c4dc95bc729afd91c00cab49d0a69bceb102eed9;hpb=eadc4de80148f3457e75b06270370a33dc4751a4

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");
     }