X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=engine-d%2Fsrc%2Ftest%2Fjava%2Forg%2Fonap%2Fholmes%2Fengine%2Fmanager%2FDroolsEngineTest.java;h=3325efa22919810d74a7c8f51d8097e6a5b317ea;hb=8754c29599ef6e59be3de0536514367f2256142f;hp=885b5e44515b0e7200c6eb446674e941fd3cb295;hpb=3c2a7c4614ed0b925a7ac9d0c2ff4907c89e6b51;p=holmes%2Fengine-management.git

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 885b5e4..3325efa 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,15 +23,18 @@ import java.lang.reflect.Method;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Locale;
-import org.drools.KnowledgeBase;
-import org.drools.KnowledgeBaseConfiguration;
-import org.drools.KnowledgeBaseFactory;
-import org.drools.conf.EventProcessingOption;
-import org.drools.runtime.StatefulKnowledgeSession;
 import org.junit.Before;
 import org.junit.Rule;
 import org.junit.Test;
 import org.junit.rules.ExpectedException;
+import org.kie.api.KieBase;
+import org.kie.api.KieServices;
+import org.kie.api.builder.KieBuilder;
+import org.kie.api.builder.KieFileSystem;
+import org.kie.api.builder.KieRepository;
+import org.kie.api.io.KieResources;
+import org.kie.api.runtime.KieContainer;
+import org.kie.api.runtime.KieSession;
 import org.onap.holmes.common.api.stat.VesAlarm;
 import org.onap.holmes.engine.request.DeployRuleRequest;
 import org.onap.holmes.common.api.entity.CorrelationRule;
@@ -39,7 +42,6 @@ import org.onap.holmes.common.constant.AlarmConst;
 import org.onap.holmes.common.exception.CorrelationException;
 import org.onap.holmes.engine.wrapper.RuleMgtWrapper;
 import org.powermock.api.easymock.PowerMock;
-import org.powermock.modules.junit4.rule.PowerMockRule;
 import org.powermock.reflect.Whitebox;
 
 public class DroolsEngineTest {
@@ -47,37 +49,45 @@ public class DroolsEngineTest {
     @Rule
     public ExpectedException thrown = ExpectedException.none();
 
-    @Rule
-    public PowerMockRule powerMockRule = new PowerMockRule();
-
     private RuleMgtWrapper ruleMgtWrapper;
 
-    private KnowledgeBase kbase;
-
-    private KnowledgeBaseConfiguration kconf;
-
-    private StatefulKnowledgeSession ksession;
+    private KieBase kieBase;
+    private KieSession kieSession;
+    private KieContainer kieContainer;
+    private KieFileSystem kfs;
+    private KieServices ks;
+    private KieBuilder kieBuilder;
+    private KieResources resources;
+    private KieRepository kieRepository;
 
 
     private DroolsEngine droolsEngine;
 
     @Before
-    public void setUp() {
+    public void setUp() throws Exception {
         droolsEngine = new DroolsEngine();
 
-        this.kconf = KnowledgeBaseFactory.newKnowledgeBaseConfiguration();
-        this.kconf.setOption(EventProcessingOption.STREAM);
-        this.kconf.setProperty("drools.assertBehaviour", "equality");
-        this.kbase = KnowledgeBaseFactory.newKnowledgeBase("D-ENGINE", this.kconf);
-        this.ksession = kbase.newStatefulKnowledgeSession();
+        ks = KieServices.Factory.get();
+        resources = ks.getResources();
+        kieRepository = ks.getRepository();
+        kfs = Whitebox.invokeMethod(droolsEngine, "createKieFileSystemWithKProject", ks);
+        kieBuilder = ks.newKieBuilder(kfs).buildAll();
+        kieContainer = ks.newKieContainer(kieRepository.getDefaultReleaseId());
+        kieBase = kieContainer.getKieBase();
+        kieSession = kieContainer.newKieSession();
 
         ruleMgtWrapper = PowerMock.createMock(RuleMgtWrapper.class);
 
         Whitebox.setInternalState(droolsEngine, "ruleMgtWrapper", ruleMgtWrapper);
 
-        Whitebox.setInternalState(droolsEngine, "kconf", kconf);
-        Whitebox.setInternalState(droolsEngine, "kbase", kbase);
-        Whitebox.setInternalState(droolsEngine, "ksession", ksession);
+        Whitebox.setInternalState(droolsEngine, "kieBase", kieBase);
+        Whitebox.setInternalState(droolsEngine, "kieSession", kieSession);
+        Whitebox.setInternalState(droolsEngine, "kieContainer", kieContainer);
+        Whitebox.setInternalState(droolsEngine, "kfs", kfs);
+        Whitebox.setInternalState(droolsEngine, "ks", ks);
+        Whitebox.setInternalState(droolsEngine, "kieBuilder", kieBuilder);
+        Whitebox.setInternalState(droolsEngine, "resources", resources);
+        Whitebox.setInternalState(droolsEngine, "kieRepository", kieRepository);
 
         PowerMock.resetAll();
     }
@@ -87,7 +97,9 @@ public class DroolsEngineTest {
 
         List<CorrelationRule> rules = new ArrayList<CorrelationRule>();
         CorrelationRule rule = new CorrelationRule();
-        rule.setContent("content");
+        rule.setContent("package content");
+        rule.setClosedControlLoopName("test");
+        rule.setPackageName("org.onap.holmes");
         rules.add(rule);
 
         expect(ruleMgtWrapper.queryRuleByEnable(anyInt())).andReturn(rules);
@@ -166,17 +178,19 @@ public class DroolsEngineTest {
     }
 
     @Test
-    public void putRaisedIntoStream_facthandle_is_null() {
+    public void putRaisedIntoStream_facthandle_is_not_null() {
         VesAlarm raiseAlarm = new VesAlarm();
-        raiseAlarm.setVersion((long) 245235);
+        raiseAlarm.setSourceId("11111");
+        raiseAlarm.setEventName("alarm");
         droolsEngine.putRaisedIntoStream(raiseAlarm);
         droolsEngine.putRaisedIntoStream(raiseAlarm);
     }
 
     @Test
-    public void putRaisedIntoStream_factHandle_is_not_null() {
+    public void putRaisedIntoStream_factHandle_is_null() {
         VesAlarm raiseAlarm = new VesAlarm();
-        raiseAlarm.setVersion((long) 245235);
+        raiseAlarm.setSourceId("11111");
+        raiseAlarm.setEventName("alarm");
         droolsEngine.putRaisedIntoStream(raiseAlarm);
     }