summaryrefslogtreecommitdiffstats
path: root/engine-d
diff options
context:
space:
mode:
Diffstat (limited to 'engine-d')
-rw-r--r--engine-d/pom.xml16
-rw-r--r--engine-d/src/main/java/org/openo/holmes/engine/manager/DroolsEngine.java24
-rw-r--r--engine-d/src/main/java/org/openo/holmes/engine/wrapper/RuleMgtWrapper.java4
-rw-r--r--engine-d/src/test/java/org/openo/holmes/engine/EngineDActiveAppTest.java27
-rw-r--r--engine-d/src/test/java/org/openo/holmes/engine/EnginedAppConfigTest.java76
-rw-r--r--engine-d/src/test/java/org/openo/holmes/engine/db/CorrelationRuleDaoTest.java50
-rw-r--r--engine-d/src/test/java/org/openo/holmes/engine/db/mapper/CorrelationRuleMapperTest.java59
-rw-r--r--engine-d/src/test/java/org/openo/holmes/engine/manager/DroolsEngineTest.java421
-rw-r--r--engine-d/src/test/java/org/openo/holmes/engine/request/CompileRuleRequestTest.java34
-rw-r--r--engine-d/src/test/java/org/openo/holmes/engine/request/DeployRuleRequestTest.java42
-rw-r--r--engine-d/src/test/java/org/openo/holmes/engine/resources/EngineResourcesTest.java134
-rw-r--r--engine-d/src/test/java/org/openo/holmes/engine/response/CorrelationRuleResponseTest.java35
-rw-r--r--engine-d/src/test/java/org/openo/holmes/engine/utils/AlarmUtilTest.java138
-rw-r--r--engine-d/src/test/java/org/openo/holmes/engine/wrapper/RuleMgtWrapperTest.java76
14 files changed, 1122 insertions, 14 deletions
diff --git a/engine-d/pom.xml b/engine-d/pom.xml
index bb055bc..727f5dc 100644
--- a/engine-d/pom.xml
+++ b/engine-d/pom.xml
@@ -81,6 +81,16 @@
<dependency>
<groupId>io.dropwizard</groupId>
<artifactId>dropwizard-core</artifactId>
+ <exclusions>
+ <exclusion>
+ <groupId>ch.qos.logback</groupId>
+ <artifactId>logback</artifactId>
+ </exclusion>
+ <exclusion>
+ <artifactId>log4j-over-slf4j</artifactId>
+ <groupId>org.slf4j</groupId>
+ </exclusion>
+ </exclusions>
</dependency>
<dependency>
<groupId>org.projectlombok</groupId>
@@ -120,6 +130,12 @@
<scope>provided</scope>
</dependency>
+ <dependency>
+ <groupId>org.powermock</groupId>
+ <artifactId>powermock-api-easymock</artifactId>
+ <version>1.4.10</version>
+ <scope>test</scope>
+ </dependency>
</dependencies>
<build>
diff --git a/engine-d/src/main/java/org/openo/holmes/engine/manager/DroolsEngine.java b/engine-d/src/main/java/org/openo/holmes/engine/manager/DroolsEngine.java
index 473b929..16ae49c 100644
--- a/engine-d/src/main/java/org/openo/holmes/engine/manager/DroolsEngine.java
+++ b/engine-d/src/main/java/org/openo/holmes/engine/manager/DroolsEngine.java
@@ -150,7 +150,9 @@ public class DroolsEngine {
StringReader reader = new StringReader(ruleContent);
Resource res = ResourceFactory.newReaderResource(reader);
- kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
+ if (kbuilder == null) {
+ kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
+ }
kbuilder.add(res, ResourceType.DRL);
@@ -161,8 +163,6 @@ public class DroolsEngine {
throw new EngineException(e);
}
- kbuilder = null;
-
ksession.fireAllRules();
}
@@ -171,7 +171,9 @@ public class DroolsEngine {
StringReader reader = new StringReader(rule.getContent());
Resource res = ResourceFactory.newReaderResource(reader);
- kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
+ if (kbuilder == null) {
+ kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
+ }
kbuilder.add(res, ResourceType.DRL);
@@ -183,9 +185,9 @@ public class DroolsEngine {
throw new CorrelationException(errorMsg);
}
- String packageName = kbuilder.getKnowledgePackages().iterator().next().getName();
+ KnowledgePackage kpackage = kbuilder.getKnowledgePackages().iterator().next();
- if (kbase.getKnowledgePackages().contains(packageName)) {
+ if (kbase.getKnowledgePackages().contains(kpackage)) {
String errorMsg = I18nProxy.getInstance().getValueByArgs(locale,
I18nProxy.ENGINE_CONTENT_ILLEGALITY,new String[]{
@@ -203,13 +205,10 @@ public class DroolsEngine {
throw new CorrelationException(errorMsg, e);
}
- kbuilder = null;
-
ksession.fireAllRules();
- return packageName;
+ return kpackage.getName();
}
-
public synchronized void undeployRule(String packageName, Locale locale)
throws CorrelationException {
@@ -237,7 +236,9 @@ public class DroolsEngine {
StringReader reader = new StringReader(content);
Resource res = ResourceFactory.newReaderResource(reader);
- kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
+ if (kbuilder == null) {
+ kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
+ }
kbuilder.add(res, ResourceType.DRL);
@@ -248,7 +249,6 @@ public class DroolsEngine {
log.error(errorMsg);
throw new CorrelationException(errorMsg);
}
- kbuilder = null;
}
public void putRaisedIntoStream(Alarm raiseAlarm) {
diff --git a/engine-d/src/main/java/org/openo/holmes/engine/wrapper/RuleMgtWrapper.java b/engine-d/src/main/java/org/openo/holmes/engine/wrapper/RuleMgtWrapper.java
index 34dab1f..59110d8 100644
--- a/engine-d/src/main/java/org/openo/holmes/engine/wrapper/RuleMgtWrapper.java
+++ b/engine-d/src/main/java/org/openo/holmes/engine/wrapper/RuleMgtWrapper.java
@@ -39,8 +39,8 @@ public class RuleMgtWrapper {
List<CorrelationRule> ruleTemp = daoUtil.getJdbiDaoByOnDemand(CorrelationRuleDao.class)
.queryRuleByRuleEnable(enable);
- if (ruleTemp != null) {
- throw new DbException(I18nProxy.RULE_MANAGEMENT_REPEAT_RULE_NAME);
+ if (ruleTemp == null) {
+ throw new DbException(I18nProxy.RULE_MANAGEMENT_DB_ERROR);
}
return ruleTemp;
}
diff --git a/engine-d/src/test/java/org/openo/holmes/engine/EngineDActiveAppTest.java b/engine-d/src/test/java/org/openo/holmes/engine/EngineDActiveAppTest.java
new file mode 100644
index 0000000..af79cc6
--- /dev/null
+++ b/engine-d/src/test/java/org/openo/holmes/engine/EngineDActiveAppTest.java
@@ -0,0 +1,27 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.engine;
+
+/**
+ * Created by Administrator on 2017/2/20.
+ */
+
+public class EngineDActiveAppTest {
+ public static void main(String[] args) throws Exception {
+ String filePath = "C:\\Users\\Administrator\\AppData\\Roaming\\IM\\6092002109\\RecvFiles\\correlation-engine.yml";
+ new EngineDActiveApp().run(new String[]{"server", filePath});
+ }
+}
diff --git a/engine-d/src/test/java/org/openo/holmes/engine/EnginedAppConfigTest.java b/engine-d/src/test/java/org/openo/holmes/engine/EnginedAppConfigTest.java
new file mode 100644
index 0000000..c24894c
--- /dev/null
+++ b/engine-d/src/test/java/org/openo/holmes/engine/EnginedAppConfigTest.java
@@ -0,0 +1,76 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.engine;
+
+import io.dropwizard.db.DataSourceFactory;
+import org.hamcrest.core.IsEqual;
+import org.hamcrest.core.IsNull;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.openo.holmes.common.config.MQConfig;
+import org.powermock.api.easymock.PowerMock;
+
+/**
+ * Created by Administrator on 2017/2/20.
+ */
+public class EnginedAppConfigTest {
+ private EngineDAppConfig engineAppConfig;
+
+ @Before
+ public void setUp() {
+ engineAppConfig = new EngineDAppConfig();
+ }
+
+ @Test
+ public void getMqConfig() {
+ MQConfig mqConfig = PowerMock.createMock(MQConfig.class);
+ engineAppConfig.setMqConfig(mqConfig);
+ Assert.assertThat(engineAppConfig.getMqConfig(), IsNull.notNullValue());
+ }
+
+ @Test
+ public void setMqConfig() {
+ MQConfig mqConfig = PowerMock.createMock(MQConfig.class);
+ engineAppConfig.setMqConfig(mqConfig);
+ Assert.assertThat(engineAppConfig.getMqConfig(), IsEqual.equalTo(mqConfig));
+ }
+
+ @Test
+ public void getDataSourceFactory() {
+ Assert.assertThat(engineAppConfig.getDataSourceFactory(), IsNull.<DataSourceFactory>notNullValue());
+ }
+
+ @Test
+ public void setDataSourceFactory() {
+ DataSourceFactory database = new DataSourceFactory();
+ engineAppConfig.setDataSourceFactory(database);
+ Assert.assertThat(engineAppConfig.getDataSourceFactory(), IsEqual.equalTo(database));
+ }
+
+ @Test
+ public void getApidescription() {
+ final String apidescription = "Holmes rule management rest API";
+ Assert.assertThat(engineAppConfig.getApidescription(),IsEqual.equalTo(apidescription));
+ }
+
+ @Test
+ public void setApidescription() {
+ final String apidescription = "set api description";
+ engineAppConfig.setApidescription(apidescription);
+ Assert.assertThat(engineAppConfig.getApidescription(),IsEqual.equalTo(apidescription));
+ }
+}
diff --git a/engine-d/src/test/java/org/openo/holmes/engine/db/CorrelationRuleDaoTest.java b/engine-d/src/test/java/org/openo/holmes/engine/db/CorrelationRuleDaoTest.java
new file mode 100644
index 0000000..3352b08
--- /dev/null
+++ b/engine-d/src/test/java/org/openo/holmes/engine/db/CorrelationRuleDaoTest.java
@@ -0,0 +1,50 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.engine.db;
+
+import org.easymock.EasyMock;
+import org.hamcrest.core.IsNull;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.openo.holmes.common.api.entity.CorrelationRule;
+import org.powermock.api.easymock.PowerMock;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * Created by Administrator on 2017/2/20.
+ */
+public class CorrelationRuleDaoTest {
+ private CorrelationRuleDao correlationRuleDao;
+
+ @Before
+ public void setUp() {
+ correlationRuleDao = PowerMock.createMock(CorrelationRuleDao.class);
+ }
+
+ @Test
+ public void queryRuleByEnable() {
+ int enable = 0;
+ EasyMock.expect(correlationRuleDao.queryRuleByEnable(EasyMock.anyInt())).andReturn(new ArrayList<CorrelationRule>());
+ PowerMock.replayAll();
+ List<CorrelationRule> correlationRules = correlationRuleDao.queryRuleByEnable(enable);
+ PowerMock.verifyAll();
+ Assert.assertThat(correlationRules, IsNull.<List<CorrelationRule>>notNullValue());
+ }
+
+}
diff --git a/engine-d/src/test/java/org/openo/holmes/engine/db/mapper/CorrelationRuleMapperTest.java b/engine-d/src/test/java/org/openo/holmes/engine/db/mapper/CorrelationRuleMapperTest.java
new file mode 100644
index 0000000..ce8f9bf
--- /dev/null
+++ b/engine-d/src/test/java/org/openo/holmes/engine/db/mapper/CorrelationRuleMapperTest.java
@@ -0,0 +1,59 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.engine.db.mapper;
+
+import org.junit.Test;
+import org.powermock.api.easymock.PowerMock;
+
+import java.sql.Date;
+import java.sql.ResultSet;
+import java.util.Properties;
+
+import static org.easymock.EasyMock.expect;
+
+/**
+ * Created by Administrator on 2017/2/20.
+ */
+public class CorrelationRuleMapperTest {
+
+ @Test
+ public void map() throws Exception {
+ CorrelationRuleMapper mapper = new CorrelationRuleMapper();
+ ResultSet resultSet = PowerMock.createMock(ResultSet.class);
+ expect(resultSet.getString("name")).andReturn("");
+ expect(resultSet.getString("rid")).andReturn("");
+ expect(resultSet.getString("description")).andReturn("");
+ expect(resultSet.getInt("enable")).andReturn(0);
+ expect(resultSet.getInt("templateID")).andReturn(0);
+ expect(resultSet.getString("engineID")).andReturn("");
+ expect(resultSet.getString("engineType")).andReturn("");
+ expect(resultSet.getString("creator")).andReturn("");
+ expect(resultSet.getDate("createTime")).andReturn(new Date(System.currentTimeMillis()));
+ expect(resultSet.getString("updator")).andReturn("");
+ expect(resultSet.getDate("updateTime")).andReturn(new Date(System.currentTimeMillis()));
+ Properties propMock = PowerMock.createMock(Properties.class);
+ expect(resultSet.getObject("params")).andReturn(propMock);
+ expect(resultSet.getString("domain")).andReturn("");
+ expect(resultSet.getString("content")).andReturn("");
+ expect(resultSet.getInt("isManual")).andReturn(0);
+ expect(resultSet.getString("vendor")).andReturn("");
+ expect(resultSet.getString("package")).andReturn("");
+
+ PowerMock.replayAll();
+ mapper.map(0, resultSet, null);
+ PowerMock.verify();
+ }
+}
diff --git a/engine-d/src/test/java/org/openo/holmes/engine/manager/DroolsEngineTest.java b/engine-d/src/test/java/org/openo/holmes/engine/manager/DroolsEngineTest.java
new file mode 100644
index 0000000..3877e3f
--- /dev/null
+++ b/engine-d/src/test/java/org/openo/holmes/engine/manager/DroolsEngineTest.java
@@ -0,0 +1,421 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.engine.manager;
+
+import org.drools.KnowledgeBase;
+import org.drools.KnowledgeBaseConfiguration;
+import org.drools.builder.KnowledgeBuilder;
+import org.drools.builder.KnowledgeBuilderErrors;
+import org.drools.builder.ResourceType;
+import org.drools.definition.KnowledgePackage;
+import org.drools.io.Resource;
+import org.drools.runtime.StatefulKnowledgeSession;
+import org.drools.runtime.rule.FactHandle;
+import org.easymock.EasyMock;
+import org.glassfish.hk2.api.IterableProvider;
+import org.hamcrest.core.IsEqual;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.ExpectedException;
+import org.junit.runner.RunWith;
+import org.openo.holmes.common.api.entity.CorrelationRule;
+import org.openo.holmes.common.api.stat.Alarm;
+import org.openo.holmes.common.config.MQConfig;
+import org.openo.holmes.common.exception.CorrelationException;
+import org.openo.holmes.common.exception.EngineException;
+import org.openo.holmes.engine.request.DeployRuleRequest;
+import org.openo.holmes.engine.wrapper.RuleMgtWrapper;
+import org.powermock.api.easymock.PowerMock;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.modules.junit4.PowerMockRunner;
+import org.powermock.reflect.Whitebox;
+
+import javax.jms.*;
+import java.lang.reflect.Method;
+import java.util.*;
+
+import static org.easymock.EasyMock.*;
+
+/**
+ * Created by Administrator on 2017/2/20.
+ */
+@RunWith(PowerMockRunner.class)
+@PrepareForTest(DroolsEngine.class)
+public class DroolsEngineTest {
+ @Rule
+ public ExpectedException thrown = ExpectedException.none();
+
+ private RuleMgtWrapper ruleMgtWrapper;
+
+ private KnowledgeBase kbase;
+
+ private KnowledgeBaseConfiguration kconf;
+
+ private StatefulKnowledgeSession ksession;
+
+ private KnowledgeBuilder kbuilder;
+
+ private IterableProvider<MQConfig> mqConfigProvider;
+
+ private ConnectionFactory connectionFactory;
+
+ private DroolsEngine droolsEngine;
+
+ @Before
+ public void setUp() {
+ droolsEngine = new DroolsEngine();
+
+ ruleMgtWrapper = PowerMock.createMock(RuleMgtWrapper.class);
+ kbase = PowerMock.createMock(KnowledgeBase.class);
+ kconf = PowerMock.createMock(KnowledgeBaseConfiguration.class);
+ ksession = PowerMock.createMock(StatefulKnowledgeSession.class);
+ kbuilder = PowerMock.createMock(KnowledgeBuilder.class);
+ mqConfigProvider = PowerMock.createMock(IterableProvider.class);
+ connectionFactory = PowerMock.createMock(ConnectionFactory.class);
+
+ Whitebox.setInternalState(droolsEngine,"ruleMgtWrapper",ruleMgtWrapper);
+ Whitebox.setInternalState(droolsEngine,"kbase",kbase);
+ Whitebox.setInternalState(droolsEngine,"kconf",kconf);
+ Whitebox.setInternalState(droolsEngine,"ksession",ksession);
+ Whitebox.setInternalState(droolsEngine,"kbuilder",kbuilder);
+ Whitebox.setInternalState(droolsEngine,"mqConfigProvider",mqConfigProvider);
+ Whitebox.setInternalState(droolsEngine,"connectionFactory",connectionFactory);
+
+ PowerMock.resetAll();
+ }
+
+ @Test
+ public void init() throws Exception {
+ MQConfig mqConfig = new MQConfig();
+ mqConfig.brokerIp = "127.0.0.1";
+ mqConfig.brokerPort = 4567;
+ mqConfig.brokerUsername = "admin";
+ mqConfig.brokerPassword = "admin";
+ expect(mqConfigProvider.get()).andReturn(mqConfig).anyTimes();
+ PowerMock.replayAll();
+
+ Method method = DroolsEngine.class.getDeclaredMethod("init");
+ method.setAccessible(true);
+ method.invoke(droolsEngine);
+
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void initDeployRule_exception() throws Exception {
+ thrown.expect(EngineException.class);
+
+ List<CorrelationRule> rules = new ArrayList<CorrelationRule>();
+ CorrelationRule rule = new CorrelationRule();
+ rule.setContent("content");
+ rules.add(rule);
+ expect(ruleMgtWrapper.queryRuleByEnable(anyInt())).andReturn(rules);
+ kbuilder.add(anyObject(Resource.class), anyObject(ResourceType.class));
+ expect(kbuilder.getKnowledgePackages()).andReturn(new ArrayList<KnowledgePackage>());
+ kbase.addKnowledgePackages(anyObject(Collection.class));
+ expectLastCall().andThrow(new RuntimeException(""));
+ PowerMock.replayAll();
+
+ Method method = DroolsEngine.class.getDeclaredMethod("initDeployRule");
+ method.setAccessible(true);
+ method.invoke(droolsEngine);
+
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void initDeployRule_normal() throws Exception {
+ List<CorrelationRule> rules = new ArrayList<CorrelationRule>();
+ CorrelationRule rule = new CorrelationRule();
+ rule.setContent("content");
+ rules.add(rule);
+ expect(ruleMgtWrapper.queryRuleByEnable(anyInt())).andReturn(rules);
+ kbuilder.add(anyObject(Resource.class), anyObject(ResourceType.class));
+ expect(kbuilder.getKnowledgePackages()).andReturn(new ArrayList<KnowledgePackage>());
+ kbase.addKnowledgePackages(anyObject(Collection.class));
+ expect(ksession.fireAllRules()).andReturn(1);
+ PowerMock.replayAll();
+
+ Method method = DroolsEngine.class.getDeclaredMethod("initDeployRule");
+ method.setAccessible(true);
+ method.invoke(droolsEngine);
+
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void deployRule_rull_is_null() throws CorrelationException {
+ Locale locale = PowerMock.createMock(Locale.class);
+
+ thrown.expect(NullPointerException.class);
+
+ droolsEngine.deployRule(null, locale);
+ }
+
+ @Test
+ public void deployRule_kbuilder_has_errors() throws CorrelationException {
+ DeployRuleRequest rule = PowerMock.createMock(DeployRuleRequest.class);
+ Locale locale = PowerMock.createMock(Locale.class);
+
+ thrown.expect(CorrelationException.class);
+
+ KnowledgeBuilderErrors errors = PowerMock.createMock(KnowledgeBuilderErrors.class);
+ expect(rule.getContent()).andReturn("rule");
+ kbuilder.add(anyObject(Resource.class), anyObject(ResourceType.class));
+ expect(kbuilder.hasErrors()).andReturn(true);
+ expect(kbuilder.getErrors()).andReturn(errors);
+ PowerMock.replayAll();
+ droolsEngine.deployRule(rule, locale);
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void deployRule_kbase_knowledgePackages_contains_package() throws CorrelationException {
+ DeployRuleRequest rule = PowerMock.createMock(DeployRuleRequest.class);
+ Locale locale = PowerMock.createMock(Locale.class);
+
+ thrown.expect(CorrelationException.class);
+
+ KnowledgePackage kPackage = PowerMock.createMock(KnowledgePackage.class);
+ Collection<KnowledgePackage> builderColl = PowerMock.createMock(Collection.class);
+ Iterator<KnowledgePackage> iterator = PowerMock.createMock(Iterator.class);
+ Collection<KnowledgePackage> baseColl = new ArrayList<KnowledgePackage>();
+ baseColl.add(kPackage);
+ expect(rule.getContent()).andReturn("rule");
+ expect(kbuilder.hasErrors()).andReturn(false);
+ kbuilder.add(anyObject(Resource.class), anyObject(ResourceType.class));
+ expect(kbuilder.getKnowledgePackages()).andReturn(builderColl);
+ expect(builderColl.iterator()).andReturn(iterator);
+ expect(iterator.next()).andReturn(kPackage);
+ expect(kbase.getKnowledgePackages()).andReturn(baseColl);
+ PowerMock.replayAll();
+ droolsEngine.deployRule(rule, locale);
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void deployRule_add_knowledge_packages_exception() throws CorrelationException {
+ DeployRuleRequest rule = PowerMock.createMock(DeployRuleRequest.class);
+ Locale locale = PowerMock.createMock(Locale.class);
+
+ thrown.expect(CorrelationException.class);
+
+ KnowledgePackage kPackage = PowerMock.createMock(KnowledgePackage.class);
+ Collection<KnowledgePackage> builderColl = PowerMock.createMock(Collection.class);
+ Iterator<KnowledgePackage> iterator = PowerMock.createMock(Iterator.class);
+ Collection<KnowledgePackage> baseColl = new ArrayList<KnowledgePackage>();
+ expect(rule.getContent()).andReturn("rule");
+ expect(kbuilder.hasErrors()).andReturn(false);
+ kbuilder.add(anyObject(Resource.class), anyObject(ResourceType.class));
+ expect(kbuilder.getKnowledgePackages()).andReturn(builderColl).times(2);
+ expect(builderColl.iterator()).andReturn(iterator);
+ expect(iterator.next()).andReturn(kPackage);
+ expect(kbase.getKnowledgePackages()).andReturn(baseColl);
+ kbase.addKnowledgePackages(anyObject(Collection.class));
+ EasyMock.expectLastCall().andThrow(new RuntimeException(""));
+ PowerMock.replayAll();
+ droolsEngine.deployRule(rule, locale);
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void deployRule_normal() throws CorrelationException {
+ DeployRuleRequest rule = PowerMock.createMock(DeployRuleRequest.class);
+ Locale locale = PowerMock.createMock(Locale.class);
+
+ final String pkgName = "pkgName";
+ KnowledgePackage kPackage = PowerMock.createMock(KnowledgePackage.class);
+ Collection<KnowledgePackage> builderColl = PowerMock.createMock(Collection.class);
+ Iterator<KnowledgePackage> iterator = PowerMock.createMock(Iterator.class);
+ Collection<KnowledgePackage> baseColl = new ArrayList<KnowledgePackage>();
+ expect(rule.getContent()).andReturn("rule");
+ expect(kbuilder.hasErrors()).andReturn(false);
+ kbuilder.add(anyObject(Resource.class), anyObject(ResourceType.class));
+ expect(kbuilder.getKnowledgePackages()).andReturn(builderColl).times(2);
+ expect(builderColl.iterator()).andReturn(iterator);
+ expect(iterator.next()).andReturn(kPackage);
+ expect(kbase.getKnowledgePackages()).andReturn(baseColl);
+ kbase.addKnowledgePackages(anyObject(Collection.class));
+ expect(ksession.fireAllRules()).andReturn(1);
+ expect(kPackage.getName()).andReturn(pkgName);
+
+ PowerMock.replayAll();
+ String resultPkgName = droolsEngine.deployRule(rule, locale);
+ PowerMock.verifyAll();
+ Assert.assertThat(resultPkgName, IsEqual.equalTo(pkgName));
+ }
+
+ @Test
+ public void undeployRule_knowledgepackage_is_null() throws CorrelationException {
+ String packageName = "packageName";
+ Locale locale = PowerMock.createMock(Locale.class);
+
+ thrown.expect(CorrelationException.class);
+
+ expect(kbase.getKnowledgePackage(anyObject(String.class))).andReturn(null);
+ PowerMock.replayAll();
+ droolsEngine.undeployRule(packageName,locale);
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void undeployRule_remove_knowledge_package_exception() throws CorrelationException {
+ String packageName = "packageName";
+ Locale locale = PowerMock.createMock(Locale.class);
+
+ thrown.expect(CorrelationException.class);
+
+ KnowledgePackage pkg = PowerMock.createMock(KnowledgePackage.class);
+ expect(kbase.getKnowledgePackage(anyObject(String.class))).andReturn(pkg);
+ expect(pkg.getName()).andReturn("");
+ kbase.removeKnowledgePackage(anyObject(String.class));
+ EasyMock.expectLastCall().andThrow(new RuntimeException(""));
+ PowerMock.replayAll();
+ droolsEngine.undeployRule(packageName,locale);
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void undeployRule_normal() throws CorrelationException {
+ String packageName = "packageName";
+ Locale locale = PowerMock.createMock(Locale.class);
+
+ KnowledgePackage pkg = PowerMock.createMock(KnowledgePackage.class);
+ expect(kbase.getKnowledgePackage(anyObject(String.class))).andReturn(pkg);
+ expect(pkg.getName()).andReturn("");
+ kbase.removeKnowledgePackage(anyObject(String.class));
+ PowerMock.replayAll();
+ droolsEngine.undeployRule(packageName,locale);
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void compileRule_kbuilder_has_errors() throws CorrelationException {
+ String content = "content";
+ Locale locale = PowerMock.createMock(Locale.class);
+
+ thrown.expect(CorrelationException.class);
+
+ KnowledgeBuilderErrors errors = PowerMock.createMock(KnowledgeBuilderErrors.class);
+ kbuilder.add(anyObject(Resource.class), anyObject(ResourceType.class));
+ expect(kbuilder.hasErrors()).andReturn(true);
+ expect(kbuilder.getErrors()).andReturn(errors);
+ PowerMock.replayAll();
+ droolsEngine.compileRule(content,locale);
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void compileRule_normal() throws CorrelationException {
+ String content = "content";
+ Locale locale = PowerMock.createMock(Locale.class);
+
+ kbuilder.add(anyObject(Resource.class), anyObject(ResourceType.class));
+ expect(kbuilder.hasErrors()).andReturn(false);
+ PowerMock.replayAll();
+ droolsEngine.compileRule(content,locale);
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void putRaisedIntoStream_facthandle_is_null() {
+ Alarm raiseAlarm = new Alarm();
+
+ expect(ksession.getFactHandle(anyObject(Alarm.class))).andReturn(null);
+ expect(ksession.insert(anyObject(Alarm.class))).andReturn(null);
+ expect(ksession.fireAllRules()).andReturn(0);
+ PowerMock.replayAll();
+ droolsEngine.putRaisedIntoStream(raiseAlarm);
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void putRaisedIntoStream_factHandle_is_not_null() {
+ Alarm raiseAlarm = new Alarm();
+ FactHandle factHandle = PowerMock.createMock(FactHandle.class);
+ expect(ksession.getFactHandle(anyObject(Alarm.class))).andReturn(factHandle);
+ ksession.retract(anyObject(FactHandle.class));
+ expect(ksession.insert(anyObject(Alarm.class))).andReturn(null);
+ expect(ksession.fireAllRules()).andReturn(0);
+ PowerMock.replayAll();
+ droolsEngine.putRaisedIntoStream(raiseAlarm);
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void listener_run_objmessage_is_null() throws JMSException {
+ DroolsEngine.AlarmMqMessageListener listener = droolsEngine.new AlarmMqMessageListener();
+
+ Connection connection = PowerMock.createMock(Connection.class);
+ Session session = PowerMock.createMock(Session.class);
+ Destination destination = PowerMock.createMock(Topic.class);
+ MessageConsumer messageConsumer = PowerMock.createMock(MessageConsumer.class);
+
+ expect(connectionFactory.createConnection()).andReturn(connection);
+ connection.start();
+ expect(connection.createSession(anyBoolean(), anyInt())).andReturn(session);
+ expect(session.createTopic(anyObject(String.class))).andReturn((Topic) destination);
+ expect(session.createConsumer(anyObject(Destination.class))).andReturn(messageConsumer);
+ expect(messageConsumer.receive(anyLong())).andReturn(null);
+
+ PowerMock.replayAll();
+ listener.run();
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void listener_run_objmessage_is_not_null() throws JMSException {
+ DroolsEngine.AlarmMqMessageListener listener = droolsEngine.new AlarmMqMessageListener();
+
+ Connection connection = PowerMock.createMock(Connection.class);
+ Session session = PowerMock.createMock(Session.class);
+ Destination destination = PowerMock.createMock(Topic.class);
+ MessageConsumer messageConsumer = PowerMock.createMock(MessageConsumer.class);
+ ObjectMessage objMessage = PowerMock.createMock(ObjectMessage.class);
+ Alarm raiseAlarm = new Alarm();
+
+ FactHandle factHandle = PowerMock.createMock(FactHandle.class);
+
+ expect(connectionFactory.createConnection()).andReturn(connection);
+ connection.start();
+ expect(connection.createSession(anyBoolean(), anyInt())).andReturn(session);
+ expect(session.createTopic(anyObject(String.class))).andReturn((Topic) destination);
+ expect(session.createConsumer(anyObject(Destination.class))).andReturn(messageConsumer);
+ expect(messageConsumer.receive(anyLong())).andReturn(objMessage);
+ expect(objMessage.getObject()).andReturn(raiseAlarm);
+
+ expect(ksession.getFactHandle(anyObject(Alarm.class))).andReturn(factHandle);
+ ksession.retract(anyObject(FactHandle.class));
+ expect(ksession.insert(anyObject(Alarm.class))).andReturn(null);
+ expect(ksession.fireAllRules()).andReturn(0);
+
+ expect(messageConsumer.receive(anyLong())).andReturn(null);
+
+ PowerMock.replayAll();
+ listener.run();
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void stop() throws Exception {
+ ksession.dispose();
+ PowerMock.replayAll();
+ droolsEngine.stop();
+ PowerMock.verifyAll();
+ }
+}
diff --git a/engine-d/src/test/java/org/openo/holmes/engine/request/CompileRuleRequestTest.java b/engine-d/src/test/java/org/openo/holmes/engine/request/CompileRuleRequestTest.java
new file mode 100644
index 0000000..dfb49b9
--- /dev/null
+++ b/engine-d/src/test/java/org/openo/holmes/engine/request/CompileRuleRequestTest.java
@@ -0,0 +1,34 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.engine.request;
+
+import org.junit.Test;
+
+import static org.hamcrest.core.IsEqual.equalTo;
+import static org.junit.Assert.assertThat;
+
+/**
+ * Created by Administrator on 2017/2/20.
+ */
+public class CompileRuleRequestTest {
+ @Test
+ public void getterAndSetter4Content(){
+ final String content = "content";
+ CompileRuleRequest request = new CompileRuleRequest();
+ request.setContent(content);
+ assertThat(request.getContent(), equalTo(content));
+ }
+}
diff --git a/engine-d/src/test/java/org/openo/holmes/engine/request/DeployRuleRequestTest.java b/engine-d/src/test/java/org/openo/holmes/engine/request/DeployRuleRequestTest.java
new file mode 100644
index 0000000..131330e
--- /dev/null
+++ b/engine-d/src/test/java/org/openo/holmes/engine/request/DeployRuleRequestTest.java
@@ -0,0 +1,42 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.engine.request;
+
+import org.junit.Test;
+
+import static org.hamcrest.core.IsEqual.equalTo;
+import static org.junit.Assert.assertThat;
+
+/**
+ * Created by Administrator on 2017/2/20.
+ */
+public class DeployRuleRequestTest {
+ @Test
+ public void getterAndSetter4Content(){
+ final String content = "content";
+ DeployRuleRequest request = new DeployRuleRequest();
+ request.setContent(content);
+ assertThat(request.getContent(), equalTo(content));
+ }
+
+ @Test
+ public void getterAndSetter4Engineid(){
+ final String engineid = "engineid";
+ DeployRuleRequest request = new DeployRuleRequest();
+ request.setEngineId(engineid);
+ assertThat(request.getEngineId(), equalTo(engineid));
+ }
+}
diff --git a/engine-d/src/test/java/org/openo/holmes/engine/resources/EngineResourcesTest.java b/engine-d/src/test/java/org/openo/holmes/engine/resources/EngineResourcesTest.java
new file mode 100644
index 0000000..2a40336
--- /dev/null
+++ b/engine-d/src/test/java/org/openo/holmes/engine/resources/EngineResourcesTest.java
@@ -0,0 +1,134 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.engine.resources;
+
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.ExpectedException;
+import org.openo.holmes.common.exception.CorrelationException;
+import org.openo.holmes.engine.manager.DroolsEngine;
+import org.openo.holmes.engine.request.CompileRuleRequest;
+import org.openo.holmes.engine.request.DeployRuleRequest;
+import org.powermock.api.easymock.PowerMock;
+import org.powermock.reflect.Whitebox;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.ws.rs.WebApplicationException;
+import java.util.Locale;
+
+import static org.easymock.EasyMock.*;
+
+/**
+ * Created by Administrator on 2017/2/20.
+ */
+public class EngineResourcesTest {
+ @Rule
+ public ExpectedException thrown = ExpectedException.none();
+ DroolsEngine droolsEngine;
+ private EngineResources engineResources;
+
+ @Before
+ public void setUp() {
+ droolsEngine = PowerMock.createMock(DroolsEngine.class);
+ engineResources = new EngineResources();
+
+ Whitebox.setInternalState(engineResources,"droolsEngine",droolsEngine);
+ PowerMock.resetAll();
+ }
+
+ @Test
+ public void deployRule_exception() throws CorrelationException {
+ DeployRuleRequest deployRuleRequest = new DeployRuleRequest();
+ HttpServletRequest httpRequest = PowerMock.createMock(HttpServletRequest.class);
+
+ thrown.expect(WebApplicationException.class);
+
+ expect(httpRequest.getHeader("language-option")).andReturn("en_US");
+ expect(droolsEngine.deployRule(anyObject(DeployRuleRequest.class), anyObject(Locale.class))).
+ andThrow(new CorrelationException(""));
+ PowerMock.replayAll();
+ engineResources.deployRule(deployRuleRequest, httpRequest);
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void deployRule_normal() throws CorrelationException {
+ DeployRuleRequest deployRuleRequest = new DeployRuleRequest();
+ HttpServletRequest httpRequest = PowerMock.createMock(HttpServletRequest.class);
+
+ expect(httpRequest.getHeader("language-option")).andReturn("en_US");
+ expect(droolsEngine.deployRule(anyObject(DeployRuleRequest.class),
+ anyObject(Locale.class))).andReturn("packageName");
+ PowerMock.replayAll();
+ engineResources.deployRule(deployRuleRequest, httpRequest);
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void undeployRule_exception() throws CorrelationException {
+ String packageName = "packageName";
+ HttpServletRequest httpRequest = PowerMock.createMock(HttpServletRequest.class);
+
+ thrown.expect(WebApplicationException.class);
+
+ expect(httpRequest.getHeader("language-option")).andReturn("en_US");
+ droolsEngine.undeployRule(anyObject(String.class), anyObject(Locale.class));
+ expectLastCall().andThrow(new CorrelationException(""));
+ PowerMock.replayAll();
+ engineResources.undeployRule(packageName, httpRequest);
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void undeployRule_normal() throws CorrelationException {
+ String packageName = "packageName";
+ HttpServletRequest httpRequest = PowerMock.createMock(HttpServletRequest.class);
+
+ expect(httpRequest.getHeader("language-option")).andReturn("en_US");
+ droolsEngine.undeployRule(anyObject(String.class), anyObject(Locale.class));
+ PowerMock.replayAll();
+ engineResources.undeployRule(packageName, httpRequest);
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void compileRule_exception() throws CorrelationException {
+ CompileRuleRequest compileRuleRequest = new CompileRuleRequest();
+ HttpServletRequest httpRequest = PowerMock.createMock(HttpServletRequest.class);
+
+ thrown.expect(WebApplicationException.class);
+
+ expect(httpRequest.getHeader("language-option")).andReturn("en_US");
+ droolsEngine.compileRule(anyObject(String.class),anyObject(Locale.class));
+ expectLastCall().andThrow(new CorrelationException(""));
+ PowerMock.replayAll();
+ engineResources.compileRule(compileRuleRequest, httpRequest);
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void compileRule_normal() throws CorrelationException {
+ CompileRuleRequest compileRuleRequest = new CompileRuleRequest();
+ HttpServletRequest httpRequest = PowerMock.createMock(HttpServletRequest.class);
+
+ expect(httpRequest.getHeader("language-option")).andReturn("en_US");
+ droolsEngine.compileRule(anyObject(String.class),anyObject(Locale.class));
+ PowerMock.replayAll();
+ engineResources.compileRule(compileRuleRequest, httpRequest);
+ PowerMock.verifyAll();
+ }
+}
diff --git a/engine-d/src/test/java/org/openo/holmes/engine/response/CorrelationRuleResponseTest.java b/engine-d/src/test/java/org/openo/holmes/engine/response/CorrelationRuleResponseTest.java
new file mode 100644
index 0000000..45cd19b
--- /dev/null
+++ b/engine-d/src/test/java/org/openo/holmes/engine/response/CorrelationRuleResponseTest.java
@@ -0,0 +1,35 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.engine.response;
+
+import org.junit.Test;
+import org.openo.holmes.engine.request.DeployRuleRequest;
+
+import static org.hamcrest.core.IsEqual.equalTo;
+import static org.junit.Assert.assertThat;
+
+/**
+ * Created by Administrator on 2017/2/20.
+ */
+public class CorrelationRuleResponseTest {
+ @Test
+ public void getterAndSetter4RuleId(){
+ final String packageName = "package";
+ CorrelationRuleResponse request = new CorrelationRuleResponse();
+ request.setPackageName(packageName);
+ assertThat(request.getPackageName(), equalTo(packageName));
+ }
+}
diff --git a/engine-d/src/test/java/org/openo/holmes/engine/utils/AlarmUtilTest.java b/engine-d/src/test/java/org/openo/holmes/engine/utils/AlarmUtilTest.java
new file mode 100644
index 0000000..d5ef771
--- /dev/null
+++ b/engine-d/src/test/java/org/openo/holmes/engine/utils/AlarmUtilTest.java
@@ -0,0 +1,138 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.engine.utils;
+
+import org.hamcrest.core.IsEqual;
+import org.hamcrest.core.IsNull;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.openo.holmes.common.api.stat.Alarm;
+import org.openo.holmes.common.producer.MQProducer;
+import org.powermock.api.easymock.PowerMock;
+import org.powermock.reflect.Whitebox;
+
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * Created by Administrator on 2017/2/20.
+ */
+public class AlarmUtilTest {
+ private AlarmUtil alarmUtil;
+ private final Map<String, Map<String, Integer>> rootPriorityMap = new HashMap<String, Map<String, Integer>>();
+ private final Map<String, String> saveRuleMsg = new HashMap<String, String>();
+
+ @Before
+ public void setUp() {
+ alarmUtil = AlarmUtil.getInstance();
+ Whitebox.setInternalState(alarmUtil,"rootPriorityMap",rootPriorityMap);
+ Whitebox.setInternalState(alarmUtil,"saveRuleMsg",saveRuleMsg);
+ PowerMock.resetAll();
+ }
+
+ @Test
+ public void getInstance() {
+ AlarmUtil instance = AlarmUtil.getInstance();
+ Assert.assertThat(instance,IsNull.<AlarmUtil>notNullValue());
+ }
+
+ @Test
+ public void equipTypeFilter_is_nullstr() {
+ String probableCauseStr = "null";
+ String equipType = "equipType";
+ Alarm alarm = new Alarm();
+ boolean filter = alarmUtil.equipTypeFilter(probableCauseStr, equipType, alarm);
+ Assert.assertThat(filter, IsEqual.equalTo(true));
+ }
+
+ @Test
+ public void equipTypeFilter_equals_alarm() {
+ String probableCauseStr = "11,4567";
+ String equipType = "ee,equipType";
+ Alarm alarm = new Alarm();
+ alarm.setProbableCause(4567);
+ alarm.setEquipType("equipType");
+ boolean filter = alarmUtil.equipTypeFilter(probableCauseStr, equipType, alarm);
+ Assert.assertThat(filter, IsEqual.equalTo(true));
+ }
+
+ @Test
+ public void equipTypeFilter_not_equals_alarm() {
+ String probableCauseStr = "11,45";
+ String equipType = "ee,equipType";
+ Alarm alarm = new Alarm();
+ alarm.setProbableCause(4567);
+ alarm.setEquipType("equipType");
+ boolean filter = alarmUtil.equipTypeFilter(probableCauseStr, equipType, alarm);
+ Assert.assertThat(filter, IsEqual.equalTo(false));
+ }
+
+ @Test
+ public void getPriority_rootprioritymap_containskey_ruleid() {
+ String ruleId = "1";
+ String probableCauseStr = "11,4567";
+ String rootAlarmFeatureStr = "0,1";
+ String equipTypeStr = "ee,equipType";
+ Alarm alarm = new Alarm();
+
+ Map<String, Integer> map = new HashMap<String, Integer>();
+ map.put("11-ee", 0);
+ map.put("4567-equipType", 1);
+ rootPriorityMap.put(ruleId, map);
+
+ saveRuleMsg.put(ruleId, "11ee0");
+
+ Integer priority = alarmUtil.getPriority(ruleId, probableCauseStr, rootAlarmFeatureStr, equipTypeStr, alarm);
+ Assert.assertThat(priority,IsEqual.equalTo(0));
+ }
+
+ @Test
+ public void getPriority_rootprioritymap_not_containskey_ruleid() {
+ String ruleId = "1";
+ String probableCauseStr = "11,4567";
+ String rootAlarmFeatureStr = "0,1";
+ String equipTypeStr = "ee,equipType";
+ Alarm alarm = new Alarm();
+
+ saveRuleMsg.put(ruleId, "11ee0");
+
+ Integer priority = alarmUtil.getPriority(ruleId, probableCauseStr, rootAlarmFeatureStr, equipTypeStr, alarm);
+ Assert.assertThat(priority,IsEqual.equalTo(0));
+ }
+
+ @Test
+ public void getPriority_priority_is_not_null() {
+ String ruleId = "1";
+ String probableCauseStr = "11,4567";
+ String rootAlarmFeatureStr = "1,1";
+ String equipTypeStr = "ee,equipType";
+ Alarm alarm = new Alarm();
+ alarm.setProbableCause(11);
+ alarm.setEquipType("ee");
+
+ saveRuleMsg.put(ruleId, "11ee0");
+
+ Integer priority = alarmUtil.getPriority(ruleId, probableCauseStr, rootAlarmFeatureStr, equipTypeStr, alarm);
+ Assert.assertThat(priority,IsEqual.equalTo(1));
+ }
+
+ @Test
+ public void getMqProducer() {
+ MQProducer mqProducer = alarmUtil.getMqProducer();
+ Assert.assertThat(mqProducer, IsNull.<MQProducer>notNullValue());
+ }
+}
diff --git a/engine-d/src/test/java/org/openo/holmes/engine/wrapper/RuleMgtWrapperTest.java b/engine-d/src/test/java/org/openo/holmes/engine/wrapper/RuleMgtWrapperTest.java
new file mode 100644
index 0000000..e8c2e7e
--- /dev/null
+++ b/engine-d/src/test/java/org/openo/holmes/engine/wrapper/RuleMgtWrapperTest.java
@@ -0,0 +1,76 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.engine.wrapper;
+
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.ExpectedException;
+import org.openo.holmes.common.api.entity.CorrelationRule;
+import org.openo.holmes.common.exception.DbException;
+import org.openo.holmes.common.utils.DbDaoUtil;
+import org.openo.holmes.engine.db.CorrelationRuleDao;
+import org.powermock.api.easymock.PowerMock;
+import org.powermock.reflect.Whitebox;
+
+import java.util.ArrayList;
+
+import static org.easymock.EasyMock.*;
+
+/**
+ * Created by Administrator on 2017/2/20.
+ */
+public class RuleMgtWrapperTest {
+ @Rule
+ public ExpectedException thrown = ExpectedException.none();
+ private DbDaoUtil daoUtil;
+ private RuleMgtWrapper ruleMgtWrapper;
+
+ @Before
+ public void setUp() {
+ daoUtil = PowerMock.createMock(DbDaoUtil.class);
+ ruleMgtWrapper = new RuleMgtWrapper();
+
+ Whitebox.setInternalState(ruleMgtWrapper,"daoUtil",daoUtil);
+ PowerMock.resetAll();
+ }
+
+ @Test
+ public void queryRuleByEnable_ruletemp_is_null() throws DbException {
+ int enable = 3;
+
+ thrown.expect(DbException.class);
+
+ CorrelationRuleDao correlationRuleDao = PowerMock.createMock(CorrelationRuleDao.class);
+ expect(daoUtil.getJdbiDaoByOnDemand(anyObject(Class.class))).andReturn(correlationRuleDao);
+ expect(correlationRuleDao.queryRuleByRuleEnable(anyInt())).andReturn(null);
+ PowerMock.replayAll();
+ ruleMgtWrapper.queryRuleByEnable(enable);
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void queryRuleByEnable_normal() throws DbException {
+ int enable = 3;
+
+ CorrelationRuleDao correlationRuleDao = PowerMock.createMock(CorrelationRuleDao.class);
+ expect(daoUtil.getJdbiDaoByOnDemand(anyObject(Class.class))).andReturn(correlationRuleDao);
+ expect(correlationRuleDao.queryRuleByRuleEnable(anyInt())).andReturn(new ArrayList<CorrelationRule>());
+ PowerMock.replayAll();
+ ruleMgtWrapper.queryRuleByEnable(enable);
+ PowerMock.verifyAll();
+ }
+}