From 117d10f41822d05c9162d771d7ab4d127d11ffad Mon Sep 17 00:00:00 2001 From: GuangrongFu Date: Thu, 16 Feb 2017 11:03:26 +0800 Subject: Add UT for Classes Change-Id: I6a539b74697d324be801777c2b0da262648a426f Issue-ID: HOLMES-14 Signed-off-by: GuangrongFu --- .../openo/holmes/rulemgt/RuleActiveAppTest.java | 25 + .../rulemgt/db/CorrelationRuleQueryDaoTest.java | 147 +++++ .../db/mapper/CorrelationRuleMapperTest.java | 54 ++ .../rulemgt/resources/RuleMgtResourcesTest.java | 318 ++++++++++ .../holmes/rulemgt/wrapper/RuleMgtWrapperTest.java | 702 +++++++++++++++++++++ 5 files changed, 1246 insertions(+) create mode 100644 rulemgt/src/test/java/org/openo/holmes/rulemgt/RuleActiveAppTest.java create mode 100644 rulemgt/src/test/java/org/openo/holmes/rulemgt/db/CorrelationRuleQueryDaoTest.java create mode 100644 rulemgt/src/test/java/org/openo/holmes/rulemgt/db/mapper/CorrelationRuleMapperTest.java create mode 100644 rulemgt/src/test/java/org/openo/holmes/rulemgt/resources/RuleMgtResourcesTest.java create mode 100644 rulemgt/src/test/java/org/openo/holmes/rulemgt/wrapper/RuleMgtWrapperTest.java (limited to 'rulemgt/src/test') diff --git a/rulemgt/src/test/java/org/openo/holmes/rulemgt/RuleActiveAppTest.java b/rulemgt/src/test/java/org/openo/holmes/rulemgt/RuleActiveAppTest.java new file mode 100644 index 0000000..69910c6 --- /dev/null +++ b/rulemgt/src/test/java/org/openo/holmes/rulemgt/RuleActiveAppTest.java @@ -0,0 +1,25 @@ +/** + * 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.rulemgt; + +public class RuleActiveAppTest { + + public static void main(String[] args) throws Exception { + String filePath = "E:\\code\\OES_Analytics_FM_Relation\\correlation-mgt\\rulemgt-standalone\\src\\assembly\\resource\\conf\\correlation-rule.yml"; + new RuleActiveApp().run(new String[]{"server", filePath}); + } +} \ No newline at end of file diff --git a/rulemgt/src/test/java/org/openo/holmes/rulemgt/db/CorrelationRuleQueryDaoTest.java b/rulemgt/src/test/java/org/openo/holmes/rulemgt/db/CorrelationRuleQueryDaoTest.java new file mode 100644 index 0000000..29e7f2b --- /dev/null +++ b/rulemgt/src/test/java/org/openo/holmes/rulemgt/db/CorrelationRuleQueryDaoTest.java @@ -0,0 +1,147 @@ +/** + * 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.rulemgt.db; + +import static org.hamcrest.MatcherAssert.assertThat; +import static org.hamcrest.Matchers.is; + +import java.util.ArrayList; +import java.util.Date; +import java.util.HashMap; +import java.util.List; +import java.util.Map; +import java.util.Properties; +import org.easymock.EasyMock; +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.exception.DbException; +import org.openo.holmes.common.utils.DbDaoUtil; +import org.openo.holmes.common.utils.I18nProxy; +import org.openo.holmes.rulemgt.bean.request.RuleQueryCondition; +import org.powermock.api.easymock.PowerMock; +import org.powermock.core.classloader.annotations.SuppressStaticInitializationFor; +import org.powermock.modules.junit4.PowerMockRunner; +import org.powermock.reflect.Whitebox; +import org.skife.jdbi.v2.Handle; +import org.skife.jdbi.v2.Query; + + +@RunWith(PowerMockRunner.class) +@SuppressStaticInitializationFor("org.openo.holmes.common.utils.DbDaoUtil") +public class CorrelationRuleQueryDaoTest { + + @Rule + public ExpectedException thrown = ExpectedException.none(); + + private DbDaoUtil dbDaoUtil; + + private Handle handle; + + private Query query; + + private CorrelationRuleQueryDao correlationRuleQueryDao; + private RuleQueryCondition ruleQueryCondition; + + @Before + public void setUp() throws Exception { + correlationRuleQueryDao = new CorrelationRuleQueryDao(); + + dbDaoUtil = PowerMock.createMock(DbDaoUtil.class); + handle = PowerMock.createMock(Handle.class); + query = PowerMock.createMock(Query.class); + + Whitebox.setInternalState(correlationRuleQueryDao, "dbDaoUtil", dbDaoUtil); + + ruleQueryCondition = createRuleQueryCondition(); + } + + + + @Test + public void getCorrelationRulesByCondition_db_exception() throws Exception { + + thrown.expect(DbException.class); + thrown.expectMessage(I18nProxy.RULE_MANAGEMENT_QUERY_RULE_FAILED); + + EasyMock.expect(dbDaoUtil.getHandle()).andReturn(handle); + EasyMock.expect(handle.createQuery(EasyMock.anyObject(String.class))).andReturn(query); + EasyMock.expect(query.list()).andThrow(new RuntimeException()).anyTimes(); + dbDaoUtil.close(handle); + EasyMock.expectLastCall(); + + PowerMock.replayAll(); + + correlationRuleQueryDao.getCorrelationRulesByCondition(ruleQueryCondition); + + PowerMock.verifyAll(); + } + + @Test + public void getCorrelationRulesByCondition_normal() throws Exception { + EasyMock.expect(dbDaoUtil.getHandle()).andReturn(handle); + EasyMock.expect(handle.createQuery(EasyMock.anyObject(String.class))).andReturn(query); + EasyMock.expect(query.list()).andReturn(createQueryResult()).anyTimes(); + dbDaoUtil.close(handle); + EasyMock.expectLastCall(); + + PowerMock.replayAll(); + + List result = correlationRuleQueryDao.getCorrelationRulesByCondition(ruleQueryCondition); + assertThat(result.size(), is(1)); + + PowerMock.verifyAll(); + } + + private List> createQueryResult() { + List> list = new ArrayList<>(); + Map value = new HashMap<>(); + value.put("name", "Rule-001"); + value.put("rid", "rule_" + System.currentTimeMillis()); + value.put("description", "desc"); + value.put("enable", 0); + value.put("templateID", 1); + value.put("engineId", "engine-001"); + value.put("engineType", "engineType-001"); + value.put("creator", "admin"); + value.put("createTime", new Date()); + value.put("updator", "admin"); + value.put("updateTime", new Date()); + value.put("params", new Properties()); + value.put("domain", "Domain"); + value.put("isManual", 0); + value.put("vendor", "Vendor"); + value.put("content", "Contents"); + value.put("package", "package"); + list.add(value); + return list; + } + + private RuleQueryCondition createRuleQueryCondition() { + RuleQueryCondition ruleQueryCondition = new RuleQueryCondition(); + ruleQueryCondition.setRid("rule_" + System.currentTimeMillis()); + ruleQueryCondition.setName("Rule-001"); + ruleQueryCondition.setEnabled(0); + ruleQueryCondition.setCreator("admin"); + ruleQueryCondition.setModifier("admin"); + return ruleQueryCondition; + } + +} diff --git a/rulemgt/src/test/java/org/openo/holmes/rulemgt/db/mapper/CorrelationRuleMapperTest.java b/rulemgt/src/test/java/org/openo/holmes/rulemgt/db/mapper/CorrelationRuleMapperTest.java new file mode 100644 index 0000000..b69a1f7 --- /dev/null +++ b/rulemgt/src/test/java/org/openo/holmes/rulemgt/db/mapper/CorrelationRuleMapperTest.java @@ -0,0 +1,54 @@ +/** + * 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.rulemgt.db.mapper; + +import static org.easymock.EasyMock.expect; + +import java.sql.Date; +import java.sql.ResultSet; +import java.util.Properties; +import org.junit.Test; +import org.powermock.api.easymock.PowerMock; + +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(1); + 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())); + expect(resultSet.getObject("params")).andReturn(new Properties()); + 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.replay(resultSet); + mapper.map(0, resultSet, null); + PowerMock.verify(resultSet); + } +} \ No newline at end of file diff --git a/rulemgt/src/test/java/org/openo/holmes/rulemgt/resources/RuleMgtResourcesTest.java b/rulemgt/src/test/java/org/openo/holmes/rulemgt/resources/RuleMgtResourcesTest.java new file mode 100644 index 0000000..1f6f331 --- /dev/null +++ b/rulemgt/src/test/java/org/openo/holmes/rulemgt/resources/RuleMgtResourcesTest.java @@ -0,0 +1,318 @@ +/** + * 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.rulemgt.resources; + +import javax.servlet.http.HttpServletRequest; +import javax.ws.rs.WebApplicationException; +import org.easymock.EasyMock; +import org.junit.Before; +import org.junit.Rule; +import org.junit.Test; +import org.junit.rules.ExpectedException; +import org.openo.holmes.common.exception.CallException; +import org.openo.holmes.common.exception.DataFormatException; +import org.openo.holmes.common.exception.DbException; +import org.openo.holmes.rulemgt.bean.request.RuleCreateRequest; +import org.openo.holmes.rulemgt.bean.request.RuleDeleteRequest; +import org.openo.holmes.rulemgt.bean.request.RuleQueryCondition; +import org.openo.holmes.rulemgt.bean.request.RuleUpdateRequest; +import org.openo.holmes.rulemgt.bean.response.RuleAddAndUpdateResponse; +import org.openo.holmes.rulemgt.bean.response.RuleQueryListResponse; +import org.openo.holmes.rulemgt.wrapper.RuleMgtWrapper; +import org.powermock.api.easymock.PowerMock; +import org.powermock.reflect.Whitebox; + +public class RuleMgtResourcesTest { + + @Rule + public ExpectedException thrown = ExpectedException.none(); + + private HttpServletRequest request = PowerMock.createMock(HttpServletRequest.class); + + private RuleMgtWrapper ruleMgtWrapper = PowerMock.createMock(RuleMgtWrapper.class); + + private RuleMgtResources ruleMgtResources = new RuleMgtResources(); + + @Before + public void setUp() throws Exception { + Whitebox.setInternalState(ruleMgtResources, "ruleMgtWrapper", ruleMgtWrapper); + PowerMock.resetAll(); + } + + @Test + public void addCorrelationRule_call_exception() throws Exception { + thrown.expect(WebApplicationException.class); + + final RuleCreateRequest ruleCreateRequest = new RuleCreateRequest(); + EasyMock.expect(ruleMgtWrapper.addCorrelationRule("admin", ruleCreateRequest)) + .andThrow(new CallException("")); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US"); + EasyMock.expect(request.getHeader("username")).andReturn("admin"); + PowerMock.replayAll(); + ruleMgtResources.addCorrelationRule(request, ruleCreateRequest); + PowerMock.verifyAll(); + } + + @Test + public void addCorrelationRule_db_exception() throws Exception { + thrown.expect(WebApplicationException.class); + + final RuleCreateRequest ruleCreateRequest = new RuleCreateRequest(); + EasyMock.expect(ruleMgtWrapper.addCorrelationRule("admin", ruleCreateRequest)).andThrow(new DbException("")); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US"); + EasyMock.expect(request.getHeader("username")).andReturn("admin"); + PowerMock.replayAll(); + ruleMgtResources.addCorrelationRule(request, ruleCreateRequest); + PowerMock.verifyAll(); + } + + @Test + public void addCorrelationRule_data_format_exception() throws Exception { + thrown.expect(WebApplicationException.class); + + final RuleCreateRequest ruleCreateRequest = new RuleCreateRequest(); + EasyMock.expect(ruleMgtWrapper.addCorrelationRule("admin", ruleCreateRequest)) + .andThrow(new DataFormatException("")); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US"); + EasyMock.expect(request.getHeader("username")).andReturn("admin"); + PowerMock.replayAll(); + ruleMgtResources.addCorrelationRule(request, ruleCreateRequest); + PowerMock.verifyAll(); + } + + @Test + public void addCorrelationRule_exception() throws Exception { + thrown.expect(WebApplicationException.class); + + final RuleCreateRequest ruleCreateRequest = new RuleCreateRequest(); + EasyMock.expect(ruleMgtWrapper.addCorrelationRule("admin", ruleCreateRequest)) + .andThrow(new RuntimeException("")); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US"); + EasyMock.expect(request.getHeader("username")).andReturn("admin"); + PowerMock.replayAll(); + ruleMgtResources.addCorrelationRule(request, ruleCreateRequest); + PowerMock.verifyAll(); + } + + @Test + public void addCorrelationRule_normal() throws Exception { + final RuleCreateRequest ruleCreateRequest = new RuleCreateRequest(); + EasyMock.expect(ruleMgtWrapper.addCorrelationRule("admin", + ruleCreateRequest)).andReturn(new RuleAddAndUpdateResponse()); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US"); + EasyMock.expect(request.getHeader("username")).andReturn("admin"); + PowerMock.replayAll(); + ruleMgtResources.addCorrelationRule(request, ruleCreateRequest); + PowerMock.verifyAll(); + } + + @Test + public void updateCorrelationRule_call_exception() throws Exception { + thrown.expect(WebApplicationException.class); + + final RuleUpdateRequest ruleUpdateRequest = new RuleUpdateRequest(); + EasyMock.expect(ruleMgtWrapper.updateCorrelationRule("admin", ruleUpdateRequest)) + .andThrow(new CallException("")); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US"); + EasyMock.expect(request.getHeader("username")).andReturn("admin"); + PowerMock.replayAll(); + ruleMgtResources.updateCorrelationRule(request, ruleUpdateRequest); + PowerMock.verifyAll(); + } + + @Test + public void updateCorrelationRule_data_format_exception() throws Exception { + thrown.expect(WebApplicationException.class); + + final RuleUpdateRequest ruleUpdateRequest = new RuleUpdateRequest(); + EasyMock.expect(ruleMgtWrapper.updateCorrelationRule("admin", ruleUpdateRequest)) + .andThrow(new DataFormatException("")); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US"); + EasyMock.expect(request.getHeader("username")).andReturn("admin"); + PowerMock.replayAll(); + ruleMgtResources.updateCorrelationRule(request, ruleUpdateRequest); + PowerMock.verifyAll(); + } + + @Test + public void updateCorrelationRule_db_exception() throws Exception { + thrown.expect(WebApplicationException.class); + + final RuleUpdateRequest ruleUpdateRequest = new RuleUpdateRequest(); + EasyMock.expect(ruleMgtWrapper.updateCorrelationRule("admin", ruleUpdateRequest)) + .andThrow(new DbException("")); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US"); + EasyMock.expect(request.getHeader("username")).andReturn("admin"); + PowerMock.replayAll(); + ruleMgtResources.updateCorrelationRule(request, ruleUpdateRequest); + PowerMock.verifyAll(); + } + + @Test + public void updateCorrelationRule_exception() throws Exception { + thrown.expect(WebApplicationException.class); + + final RuleUpdateRequest ruleUpdateRequest = new RuleUpdateRequest(); + EasyMock.expect(ruleMgtWrapper.updateCorrelationRule("admin", ruleUpdateRequest)) + .andThrow(new RuntimeException("")); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US"); + EasyMock.expect(request.getHeader("username")).andReturn("admin"); + PowerMock.replayAll(); + ruleMgtResources.updateCorrelationRule(request, ruleUpdateRequest); + PowerMock.verifyAll(); + } + + @Test + public void updateCorrelationRule_normal() throws Exception { + final RuleUpdateRequest ruleUpdateRequest = new RuleUpdateRequest(); + EasyMock.expect(ruleMgtWrapper.updateCorrelationRule("admin", + ruleUpdateRequest)).andReturn(new RuleAddAndUpdateResponse()); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US"); + EasyMock.expect(request.getHeader("username")).andReturn("admin"); + PowerMock.replayAll(); + ruleMgtResources.updateCorrelationRule(request, ruleUpdateRequest); + PowerMock.verifyAll(); + } + + @Test + public void deleteCorrelationRule_call_exception() throws Exception { + thrown.expect(WebApplicationException.class); + + final RuleDeleteRequest ruleDeleteRequest = new RuleDeleteRequest(); + ruleMgtWrapper.deleteCorrelationRule(ruleDeleteRequest); + EasyMock.expectLastCall().andThrow(new CallException("")); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US"); + PowerMock.replayAll(); + ruleMgtResources.deleteCorrelationRule(request, ruleDeleteRequest); + PowerMock.verifyAll(); + } + + @Test + public void deleteCorrelationRule_data_format_exception() throws Exception { + thrown.expect(WebApplicationException.class); + + final RuleDeleteRequest ruleDeleteRequest = new RuleDeleteRequest(); + ruleMgtWrapper.deleteCorrelationRule(ruleDeleteRequest); + EasyMock.expectLastCall().andThrow(new DataFormatException("")); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US"); + PowerMock.replayAll(); + ruleMgtResources.deleteCorrelationRule(request, ruleDeleteRequest); + PowerMock.verifyAll(); + } + + @Test + public void deleteCorrelationRule_db_exception() throws Exception { + thrown.expect(WebApplicationException.class); + + final RuleDeleteRequest ruleDeleteRequest = new RuleDeleteRequest(); + ruleMgtWrapper.deleteCorrelationRule(ruleDeleteRequest); + EasyMock.expectLastCall().andThrow(new DbException("")); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US"); + PowerMock.replayAll(); + ruleMgtResources.deleteCorrelationRule(request, ruleDeleteRequest); + PowerMock.verifyAll(); + } + + @Test + public void deleteCorrelationRule_exception() throws Exception { + thrown.expect(WebApplicationException.class); + + final RuleDeleteRequest ruleDeleteRequest = new RuleDeleteRequest(); + ruleMgtWrapper.deleteCorrelationRule(ruleDeleteRequest); + EasyMock.expectLastCall().andThrow(new RuntimeException("")); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US"); + PowerMock.replayAll(); + ruleMgtResources.deleteCorrelationRule(request, ruleDeleteRequest); + PowerMock.verifyAll(); + } + + @Test + public void deleteCorrelationRule_normal() throws Exception { + final RuleDeleteRequest ruleDeleteRequest = new RuleDeleteRequest(); + ruleMgtWrapper.deleteCorrelationRule(ruleDeleteRequest); + EasyMock.expectLastCall(); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US"); + PowerMock.replayAll(); + ruleMgtResources.deleteCorrelationRule(request, ruleDeleteRequest); + PowerMock.verifyAll(); + } + + @Test + public void getCorrelationRules_data_format_exception() throws Exception { + thrown.expect(WebApplicationException.class); + + final String requestStr = "{\"ruleid\":\"rule_001\",\"rulename\":\"Rule-001\"," + + "\"enabled\":0,\"creator\":\"admin\"}"; + EasyMock.expect(ruleMgtWrapper.getCorrelationRuleByCondition(EasyMock.anyObject(RuleQueryCondition.class))) + .andThrow(new DataFormatException("")); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US").times(2); + PowerMock.replayAll(); + ruleMgtResources.getCorrelationRules(request, requestStr); + PowerMock.verifyAll(); + } + + @Test + public void getCorrelationRules_db_exception() throws Exception { + thrown.expect(WebApplicationException.class); + + final String requestStr = "{\"ruleid\":\"rule_001\",\"rulename\":\"Rule-001\"," + + "\"enabled\":0,\"creator\":\"admin\"}"; + EasyMock.expect(ruleMgtWrapper.getCorrelationRuleByCondition(EasyMock.anyObject(RuleQueryCondition.class))) + .andThrow(new DbException("")); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US").times(2); + PowerMock.replayAll(); + ruleMgtResources.getCorrelationRules(request, requestStr); + PowerMock.verifyAll(); + } + + @Test + public void getCorrelationRules_exception() throws Exception { + thrown.expect(WebApplicationException.class); + + final String requestStr = "{\"ruleid\":\"rule_001\",\"rulename\":\"Rule-001\"," + + "\"enabled\":0,\"creator\":\"admin\"}"; + EasyMock.expect(ruleMgtWrapper.getCorrelationRuleByCondition(EasyMock.anyObject(RuleQueryCondition.class))) + .andThrow(new RuntimeException("")); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US").times(2); + PowerMock.replayAll(); + ruleMgtResources.getCorrelationRules(request, requestStr); + PowerMock.verifyAll(); + } + + + @Test + public void getCorrelationRules_normal_request_string_null() throws Exception { + EasyMock.expect(ruleMgtWrapper.getCorrelationRuleByCondition(EasyMock.anyObject(RuleQueryCondition.class))) + .andReturn(new RuleQueryListResponse()); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US").times(2); + PowerMock.replayAll(); + ruleMgtResources.getCorrelationRules(request, null); + PowerMock.verifyAll(); + } + + @Test + public void getCorrelationRules_normal_request_string_enabled_missing() throws Exception { + final String requestStr = "{\"ruleid\":\"rule_001\",\"rulename\":\"Rule-001\"," + + "\"creator\":\"admin\"}"; + EasyMock.expect(ruleMgtWrapper.getCorrelationRuleByCondition(EasyMock.anyObject(RuleQueryCondition.class))) + .andReturn(new RuleQueryListResponse()); + EasyMock.expect(request.getHeader("language-option")).andReturn("en_US").times(2); + PowerMock.replayAll(); + ruleMgtResources.getCorrelationRules(request, requestStr); + PowerMock.verifyAll(); + } +} \ No newline at end of file diff --git a/rulemgt/src/test/java/org/openo/holmes/rulemgt/wrapper/RuleMgtWrapperTest.java b/rulemgt/src/test/java/org/openo/holmes/rulemgt/wrapper/RuleMgtWrapperTest.java new file mode 100644 index 0000000..41cb494 --- /dev/null +++ b/rulemgt/src/test/java/org/openo/holmes/rulemgt/wrapper/RuleMgtWrapperTest.java @@ -0,0 +1,702 @@ +/** + * 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.rulemgt.wrapper; + +import static org.hamcrest.MatcherAssert.assertThat; +import static org.hamcrest.Matchers.equalTo; +import static org.hamcrest.Matchers.is; + + +import java.util.ArrayList; +import java.util.Date; +import java.util.List; +import org.easymock.EasyMock; +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.exception.CallException; +import org.openo.holmes.common.exception.DataFormatException; +import org.openo.holmes.common.exception.DbException; +import org.openo.holmes.common.utils.DbDaoUtil; +import org.openo.holmes.common.utils.I18nProxy; +import org.openo.holmes.rulemgt.bean.request.CorrelationCheckRule4Engine; +import org.openo.holmes.rulemgt.bean.request.CorrelationDeployRule4Engine; +import org.openo.holmes.rulemgt.bean.request.RuleCreateRequest; +import org.openo.holmes.rulemgt.bean.request.RuleDeleteRequest; +import org.openo.holmes.rulemgt.bean.request.RuleQueryCondition; +import org.openo.holmes.rulemgt.bean.request.RuleUpdateRequest; +import org.openo.holmes.rulemgt.bean.response.RuleAddAndUpdateResponse; +import org.openo.holmes.rulemgt.bean.response.RuleQueryListResponse; +import org.openo.holmes.rulemgt.bolt.enginebolt.EngineWrapper; +import org.openo.holmes.rulemgt.db.CorrelationRuleDao; +import org.openo.holmes.rulemgt.db.CorrelationRuleQueryDao; +import org.powermock.api.easymock.PowerMock; +import org.powermock.core.classloader.annotations.SuppressStaticInitializationFor; +import org.powermock.modules.junit4.PowerMockRunner; +import org.powermock.reflect.Whitebox; + +@RunWith(PowerMockRunner.class) +@SuppressStaticInitializationFor("org.openo.holmes.common.utils.DbDaoUtil") +public class RuleMgtWrapperTest { + + @Rule + public ExpectedException thrown = ExpectedException.none(); + + private RuleMgtWrapper ruleMgtWrapper; + + private EngineWrapper engineWrapperMock; + + private DbDaoUtil dbDaoUtilMock; + + private CorrelationRuleQueryDao correlationRuleQueryDaoMock; + + private CorrelationRuleDao correlationRuleDaoMock; + + private static final String USER_NAME = "admin"; + + @Before + public void setUp() throws Exception { + + ruleMgtWrapper = new RuleMgtWrapper(); + + engineWrapperMock = PowerMock.createMock(EngineWrapper.class); + correlationRuleQueryDaoMock = PowerMock.createMock(CorrelationRuleQueryDao.class); + dbDaoUtilMock = PowerMock.createMock(DbDaoUtil.class); + correlationRuleDaoMock = PowerMock.createMock(CorrelationRuleDao.class); + + Whitebox.setInternalState(ruleMgtWrapper, "daoUtil", dbDaoUtilMock); + Whitebox.setInternalState(ruleMgtWrapper, "correlationRuleQueryDao", correlationRuleQueryDaoMock); + Whitebox.setInternalState(ruleMgtWrapper, "engineWarpper", engineWrapperMock); + + PowerMock.resetAll(); + } + + @Test + public void addCorrelationRule_name_is_null() throws Exception { + thrown.expect(DbException.class); + thrown.expectMessage(I18nProxy.RULE_MANAGEMENT_RULE_NAME_IS_EMPTY); + + ruleMgtWrapper.addCorrelationRule(USER_NAME, createRuleCreateRequest(null, "This is a rule for testing.", + "Mocked contents.", 0)); + } + + @Test + public void addCorrelationRule_name_is_empty() throws Exception { + thrown.expect(DbException.class); + thrown.expectMessage(I18nProxy.RULE_MANAGEMENT_RULE_NAME_IS_EMPTY); + + ruleMgtWrapper.addCorrelationRule("admin", createRuleCreateRequest("", "This is a rule for testing.", + "Mocked contents.", 0)); + } + + @Test + public void addCorrelationRule_duplicated_rule() throws Exception { + + final String ruleName = "Rule-001"; + + RuleCreateRequest ruleCreateRequest = createRuleCreateRequest(ruleName, "This is a rule for testing.", + "Mocked contents.", 0); + CorrelationRule correlationRule = convertCreateRequest2CorrelationRule(ruleCreateRequest); + + thrown.expect(DbException.class); + thrown.expectMessage(I18nProxy.RULE_MANAGEMENT_REPEAT_RULE_NAME); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock); + EasyMock.expect(correlationRuleDaoMock.queryRuleByRuleName(ruleName)).andReturn(correlationRule); + + PowerMock.replayAll(); + + ruleMgtWrapper.addCorrelationRule("admin", ruleCreateRequest); + + PowerMock.verifyAll(); + } + + @Test + public void addCorrelationRule_rule_deploy_verify_failure() throws Exception { + + final String ruleName = "Rule-001"; + + RuleCreateRequest ruleCreateRequest = createRuleCreateRequest(ruleName, "This is a rule for testing.", + "Mocked contents.", 0); + + thrown.expect(CallException.class); + thrown.expectMessage(I18nProxy.RULE_MANAGEMENT__CALL_CHECK_RULE_REST_FAILED); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock); + EasyMock.expect(correlationRuleDaoMock.queryRuleByRuleName(ruleName)).andReturn(null); + EasyMock.expect(engineWrapperMock.checkRuleFromEngine(EasyMock.anyObject(CorrelationCheckRule4Engine.class))) + .andThrow(new CallException(I18nProxy.RULE_MANAGEMENT__CALL_CHECK_RULE_REST_FAILED)); + + PowerMock.replayAll(); + + ruleMgtWrapper.addCorrelationRule("admin", ruleCreateRequest); + + PowerMock.verifyAll(); + } + + @Test + public void addCorrelationRule_rule_deploy_rule_enabled_failure() throws Exception { + + final String ruleName = "Rule-001"; + + RuleCreateRequest ruleCreateRequest = createRuleCreateRequest(ruleName, "This is a rule for testing.", + "Mocked contents.", 1); + + thrown.expect(CallException.class); + thrown.expectMessage(I18nProxy.RULE_MANAGEMENT_CALL_DEPLOY_RULE_REST_FAILED); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock); + EasyMock.expect(correlationRuleDaoMock.queryRuleByRuleName(ruleName)).andReturn(null); + EasyMock.expect(engineWrapperMock.checkRuleFromEngine(EasyMock.anyObject(CorrelationCheckRule4Engine.class))) + .andReturn(true); + EasyMock.expect(engineWrapperMock.deployEngine(EasyMock.anyObject(CorrelationDeployRule4Engine.class))) + .andThrow(new CallException(I18nProxy.RULE_MANAGEMENT_CALL_DEPLOY_RULE_REST_FAILED)); + + PowerMock.replayAll(); + + ruleMgtWrapper.addCorrelationRule("admin", ruleCreateRequest); + + PowerMock.verifyAll(); + } + + @Test + public void addCorrelationRule_rule_save_failure() throws Exception { + final String ruleName = "Rule-001"; + + RuleCreateRequest ruleCreateRequest = createRuleCreateRequest(ruleName, "This is a rule for testing.", + "Mocked contents.", 1); + + thrown.expect(DbException.class); + thrown.expectMessage(I18nProxy.RULE_MANAGEMENT_CREATE_RULE_FAILED); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock).times(2); + EasyMock.expect(correlationRuleDaoMock.queryRuleByRuleName(ruleName)).andReturn(null); + EasyMock.expect(engineWrapperMock.checkRuleFromEngine(EasyMock.anyObject(CorrelationCheckRule4Engine.class))) + .andReturn(true); + EasyMock.expect(engineWrapperMock.deployEngine(EasyMock.anyObject(CorrelationDeployRule4Engine.class))) + .andReturn("package-001"); + EasyMock.expect(correlationRuleDaoMock.saveRule(EasyMock.anyObject(CorrelationRule.class))) + .andThrow(new RuntimeException("any message")); + + PowerMock.replayAll(); + + ruleMgtWrapper.addCorrelationRule("admin", ruleCreateRequest); + + PowerMock.verifyAll(); + } + + @Test + public void addCorrelationRule_normal() throws Exception { + final String ruleName = "Rule-001"; + + RuleCreateRequest ruleCreateRequest = createRuleCreateRequest(ruleName, "This is a rule for testing.", + "Mocked contents.", 1); + + CorrelationRule correlationRuleRet = new CorrelationRule(); + correlationRuleRet.setRid("rule_" + System.currentTimeMillis()); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock).times(2); + EasyMock.expect(correlationRuleDaoMock.queryRuleByRuleName(ruleName)).andReturn(null); + EasyMock.expect(engineWrapperMock.checkRuleFromEngine(EasyMock.anyObject(CorrelationCheckRule4Engine.class))) + .andReturn(true); + EasyMock.expect(engineWrapperMock.deployEngine(EasyMock.anyObject(CorrelationDeployRule4Engine.class))) + .andReturn("package-001"); + EasyMock.expect(correlationRuleDaoMock.saveRule(EasyMock.anyObject(CorrelationRule.class))) + .andReturn(correlationRuleRet); + + PowerMock.replayAll(); + + RuleAddAndUpdateResponse response = ruleMgtWrapper.addCorrelationRule("admin", ruleCreateRequest); + assertThat(response.getRuleId(), equalTo(correlationRuleRet.getRid())); + + PowerMock.verifyAll(); + } + + @Test + public void updateCorrelationRule_param_null() throws Exception { + thrown.expect(DataFormatException.class); + thrown.expectMessage(I18nProxy.RULE_MANAGEMENT_REQUEST_OBJECT_IS_EMPTY); + + ruleMgtWrapper.updateCorrelationRule(USER_NAME, null); + } + + @Test + public void updateCorrelationRule_rule_not_exist() throws Exception { + thrown.expect(DbException.class); + thrown.expectMessage(I18nProxy.RULE_MANAGEMENT_RULE_NOT_EXIST_DATABASE); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock); + EasyMock.expect(correlationRuleDaoMock.getRuleByRid(EasyMock.anyObject(String.class))).andReturn(null); + + PowerMock.replayAll(); + + ruleMgtWrapper.updateCorrelationRule(USER_NAME, new RuleUpdateRequest()); + + PowerMock.verifyAll(); + } + + @Test + public void updateCorrelationRule_rule_status_illegal_with_contents() throws Exception { + thrown.expect(DataFormatException.class); + thrown.expectMessage(I18nProxy.RULE_MANAGEMENT_PARAMETER_ENABLED_ERROR); + + RuleUpdateRequest ruleUpdateRequest = createRuleUpdateRequest("rule_" + System.currentTimeMillis(), + "desc", "contents", 3); + CorrelationRule correlationRule = convertUpdateRequest2CorrelationRule(ruleUpdateRequest); + correlationRule.setContent("previous contents"); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock); + EasyMock.expect(correlationRuleDaoMock.getRuleByRid(EasyMock.anyObject(String.class))) + .andReturn(correlationRule); + + PowerMock.replayAll(); + + ruleMgtWrapper.updateCorrelationRule(USER_NAME, ruleUpdateRequest); + + PowerMock.verifyAll(); + } + + @Test + public void updateCorrelationRule_rule_status_illegal_with_null_contents() throws Exception { + thrown.expect(DataFormatException.class); + thrown.expectMessage(I18nProxy.RULE_MANAGEMENT_PARAMETER_ENABLED_ERROR); + + RuleUpdateRequest ruleUpdateRequest = createRuleUpdateRequest("rule_" + System.currentTimeMillis(), + "desc", null, 3); + CorrelationRule correlationRule = convertUpdateRequest2CorrelationRule(ruleUpdateRequest); + correlationRule.setContent("previous contents"); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock); + EasyMock.expect(correlationRuleDaoMock.getRuleByRid(EasyMock.anyObject(String.class))) + .andReturn(correlationRule); + + PowerMock.replayAll(); + + ruleMgtWrapper.updateCorrelationRule(USER_NAME, ruleUpdateRequest); + + PowerMock.verifyAll(); + } + + @Test + public void updateCorrelationRule_rule_disabled_update_failure() throws Exception { + thrown.expect(DbException.class); + thrown.expectMessage(I18nProxy.RULE_MANAGEMENT_UPDATE_RULE_FAILED); + + RuleUpdateRequest ruleUpdateRequest = createRuleUpdateRequest("rule_" + System.currentTimeMillis(), + "desc", "contents", 0); + CorrelationRule correlationRuleOld = convertUpdateRequest2CorrelationRule(ruleUpdateRequest); + correlationRuleOld.setContent("previous contents"); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock).anyTimes(); + EasyMock.expect(correlationRuleDaoMock.getRuleByRid(EasyMock.anyObject(String.class))) + .andReturn(correlationRuleOld); + correlationRuleDaoMock.updateRule(EasyMock.anyObject(CorrelationRule.class)); + EasyMock.expectLastCall().andThrow(new RuntimeException("Failed to update the rule.")); + + PowerMock.replayAll(); + + ruleMgtWrapper.updateCorrelationRule(USER_NAME, ruleUpdateRequest); + + PowerMock.verifyAll(); + } + + @Test + public void updateCorrelationRule_rule_disabled_deploy_verify_exception() throws Exception { + thrown.expect(CallException.class); + thrown.expectMessage(I18nProxy.RULE_MANAGEMENT__CALL_CHECK_RULE_REST_FAILED); + + RuleUpdateRequest ruleUpdateRequest = createRuleUpdateRequest("rule_" + System.currentTimeMillis(), + "desc", null, 0); + CorrelationRule correlationRuleOld = convertUpdateRequest2CorrelationRule(ruleUpdateRequest); + correlationRuleOld.setContent("previous contents"); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock).anyTimes(); + EasyMock.expect(correlationRuleDaoMock.getRuleByRid(EasyMock.anyObject(String.class))) + .andReturn(correlationRuleOld); + correlationRuleDaoMock.updateRule(EasyMock.anyObject(CorrelationRule.class)); + EasyMock.expectLastCall(); + EasyMock.expect(engineWrapperMock.checkRuleFromEngine(EasyMock.anyObject(CorrelationCheckRule4Engine.class))) + .andThrow(new CallException(I18nProxy.RULE_MANAGEMENT__CALL_CHECK_RULE_REST_FAILED)); + + PowerMock.replayAll(); + + ruleMgtWrapper.updateCorrelationRule(USER_NAME, ruleUpdateRequest); + + PowerMock.verifyAll(); + } + + @Test + public void updateCorrelationRule_rule_disabled_deploy_verify_false() throws Exception { + + RuleUpdateRequest ruleUpdateRequest = createRuleUpdateRequest("rule_" + System.currentTimeMillis(), + "desc", null, 0); + CorrelationRule correlationRuleOld = convertUpdateRequest2CorrelationRule(ruleUpdateRequest); + correlationRuleOld.setContent("previous contents"); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock).anyTimes(); + EasyMock.expect(correlationRuleDaoMock.getRuleByRid(EasyMock.anyObject(String.class))) + .andReturn(correlationRuleOld); + correlationRuleDaoMock.updateRule(EasyMock.anyObject(CorrelationRule.class)); + EasyMock.expectLastCall(); + EasyMock.expect(engineWrapperMock.checkRuleFromEngine(EasyMock.anyObject(CorrelationCheckRule4Engine.class))) + .andReturn(false); + + PowerMock.replayAll(); + + RuleAddAndUpdateResponse response = ruleMgtWrapper.updateCorrelationRule(USER_NAME, ruleUpdateRequest); + + assertThat(response.getRuleId(), equalTo(ruleUpdateRequest.getRuleId())); + + PowerMock.verifyAll(); + } + + @Test + public void updateCorrelationRule_rule_enabled_engine_delete_failure() throws Exception { + thrown.expect(DbException.class); + thrown.expectMessage(I18nProxy.RULE_MANAGEMENT_UPDATE_RULE_FAILED); + + RuleUpdateRequest ruleUpdateRequest = createRuleUpdateRequest("rule_" + System.currentTimeMillis(), + "desc", null, 1); + CorrelationRule correlationRuleOld = convertUpdateRequest2CorrelationRule(ruleUpdateRequest); + correlationRuleOld.setContent("previous contents"); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock).anyTimes(); + EasyMock.expect(correlationRuleDaoMock.getRuleByRid(EasyMock.anyObject(String.class))) + .andReturn(correlationRuleOld); + engineWrapperMock.deleteRuleFromEngine(correlationRuleOld.getPackageName()); + EasyMock.expectLastCall().andThrow(new RuntimeException("Failed to delete the rule from the engine.")); + + PowerMock.replayAll(); + + ruleMgtWrapper.updateCorrelationRule(USER_NAME, ruleUpdateRequest); + + PowerMock.verifyAll(); + } + + @Test + public void updateCorrelationRule_rule_enabled_engine_deploy_failure() throws Exception { + thrown.expect(CallException.class); + thrown.expectMessage(I18nProxy.RULE_MANAGEMENT_CALL_DEPLOY_RULE_REST_FAILED); + + RuleUpdateRequest ruleUpdateRequest = createRuleUpdateRequest("rule_" + System.currentTimeMillis(), + "desc", null, 1); + CorrelationRule correlationRuleOld = convertUpdateRequest2CorrelationRule(ruleUpdateRequest); + correlationRuleOld.setContent("previous contents"); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock).anyTimes(); + EasyMock.expect(correlationRuleDaoMock.getRuleByRid(EasyMock.anyObject(String.class))) + .andReturn(correlationRuleOld); + EasyMock.expect(engineWrapperMock.deleteRuleFromEngine(correlationRuleOld.getPackageName())).andReturn(true); + correlationRuleDaoMock.updateRule(EasyMock.anyObject(CorrelationRule.class)); + EasyMock.expectLastCall(); + EasyMock.expect(engineWrapperMock.checkRuleFromEngine(EasyMock.anyObject(CorrelationCheckRule4Engine.class))) + .andReturn(true); + EasyMock.expect(engineWrapperMock.deployEngine(EasyMock.anyObject(CorrelationDeployRule4Engine.class))) + .andThrow(new CallException(I18nProxy.RULE_MANAGEMENT_CALL_DEPLOY_RULE_REST_FAILED)); + + PowerMock.replayAll(); + + ruleMgtWrapper.updateCorrelationRule(USER_NAME, ruleUpdateRequest); + + PowerMock.verifyAll(); + } + + @Test + public void updateCorrelationRule_rule_enabled_deploy_rule_enabled() throws Exception { + + RuleUpdateRequest ruleUpdateRequest = createRuleUpdateRequest("rule_" + System.currentTimeMillis(), + "desc", null, 1); + CorrelationRule correlationRuleOld = convertUpdateRequest2CorrelationRule(ruleUpdateRequest); + correlationRuleOld.setContent("previous contents"); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock).anyTimes(); + EasyMock.expect(correlationRuleDaoMock.getRuleByRid(EasyMock.anyObject(String.class))) + .andReturn(correlationRuleOld); + EasyMock.expect(engineWrapperMock.deleteRuleFromEngine(correlationRuleOld.getPackageName())).andReturn(true); + correlationRuleDaoMock.updateRule(EasyMock.anyObject(CorrelationRule.class)); + EasyMock.expectLastCall(); + EasyMock.expect(engineWrapperMock.checkRuleFromEngine(EasyMock.anyObject(CorrelationCheckRule4Engine.class))) + .andReturn(true); + EasyMock.expect(engineWrapperMock.deployEngine(EasyMock.anyObject(CorrelationDeployRule4Engine.class))) + .andReturn("package-name"); + + PowerMock.replayAll(); + + RuleAddAndUpdateResponse response = ruleMgtWrapper.updateCorrelationRule(USER_NAME, ruleUpdateRequest); + + assertThat(response.getRuleId(), equalTo(ruleUpdateRequest.getRuleId())); + + PowerMock.verifyAll(); + } + + @Test + public void checkCorrelation_content_null() throws Exception { + RuleUpdateRequest ruleCreateRequest = createRuleUpdateRequest("rule_" + System.currentTimeMillis(), + "desc", null, 0); + CorrelationRule correlationRuleOld = convertUpdateRequest2CorrelationRule(ruleCreateRequest); + CorrelationRule correlationRuleNew = convertUpdateRequest2CorrelationRule(ruleCreateRequest); + correlationRuleOld.setContent("previous contents"); + + ruleMgtWrapper.checkCorrelation(correlationRuleNew, correlationRuleOld); + + assertThat(correlationRuleNew.getContent(), equalTo(correlationRuleOld.getContent())); + } + + @Test + public void checkCorrelation_illegal_status() throws Exception { + thrown.expect(DataFormatException.class); + thrown.expectMessage(I18nProxy.RULE_MANAGEMENT_PARAMETER_ENABLED_ERROR); + + RuleUpdateRequest ruleCreateRequest = createRuleUpdateRequest("rule_" + System.currentTimeMillis(), + "desc", null, 2); + CorrelationRule correlationRuleOld = convertUpdateRequest2CorrelationRule(ruleCreateRequest); + CorrelationRule correlationRuleNew = convertUpdateRequest2CorrelationRule(ruleCreateRequest); + correlationRuleOld.setContent("previous contents"); + + ruleMgtWrapper.checkCorrelation(correlationRuleNew, correlationRuleOld); + } + + + @Test + public void deleteCorrelationRule_request_null() throws Exception { + thrown.expect(DataFormatException.class); + thrown.expectMessage((I18nProxy.RULE_MANAGEMENT_REQUEST_OBJECT_IS_EMPTY)); + + ruleMgtWrapper.deleteCorrelationRule(null); + } + + @Test + public void deleteCorrelationRule_rule_not_exit() throws Exception { + thrown.expect(DbException.class); + thrown.expectMessage((I18nProxy.RULE_MANAGEMENT_RULE_NOT_EXIST_DATABASE)); + + RuleDeleteRequest ruleDeleteRequest = createRuleDeleteRequest("rule_" + System.currentTimeMillis()); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock).anyTimes(); + EasyMock.expect(correlationRuleDaoMock.getRuleByRid(ruleDeleteRequest.getRuleId())) + .andReturn(null); + + PowerMock.replayAll(); + + ruleMgtWrapper.deleteCorrelationRule(ruleDeleteRequest); + + PowerMock.verifyAll(); + } + + @Test + public void deleteCorrelationRule_rule_enabled_engine_delete_exception() throws Exception { + thrown.expect(DbException.class); + thrown.expectMessage((I18nProxy.RULE_MANAGEMENT_DELETE_RULE_FAILED)); + + RuleDeleteRequest ruleDeleteRequest = createRuleDeleteRequest("rule_" + System.currentTimeMillis()); + RuleUpdateRequest ruleCreateRequest = createRuleUpdateRequest("rule_" + System.currentTimeMillis(), + "desc", "contents", 1); + CorrelationRule correlationRule = convertUpdateRequest2CorrelationRule(ruleCreateRequest); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock).anyTimes(); + EasyMock.expect(correlationRuleDaoMock.getRuleByRid(ruleDeleteRequest.getRuleId())) + .andReturn(correlationRule); + EasyMock.expect(engineWrapperMock.deleteRuleFromEngine(correlationRule.getPackageName())) + .andThrow(new RuntimeException("Failed to delete the rule from the engine")); + + PowerMock.replayAll(); + + ruleMgtWrapper.deleteCorrelationRule(ruleDeleteRequest); + + PowerMock.verifyAll(); + } + + @Test + public void deleteCorrelationRule_rule_disabled_delete_exception() throws Exception { + thrown.expect(DbException.class); + thrown.expectMessage((I18nProxy.RULE_MANAGEMENT_DELETE_RULE_FAILED)); + + RuleDeleteRequest ruleDeleteRequest = createRuleDeleteRequest("rule_" + System.currentTimeMillis()); + RuleUpdateRequest ruleCreateRequest = createRuleUpdateRequest("rule_" + System.currentTimeMillis(), + "desc", "contents", 0); + CorrelationRule correlationRule = convertUpdateRequest2CorrelationRule(ruleCreateRequest); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock).anyTimes(); + EasyMock.expect(correlationRuleDaoMock.getRuleByRid(ruleDeleteRequest.getRuleId())) + .andReturn(correlationRule); + correlationRuleDaoMock.deleteRule(correlationRule); + EasyMock.expectLastCall().andThrow(new RuntimeException("Failed to delete the rule from the DB")); + + PowerMock.replayAll(); + + ruleMgtWrapper.deleteCorrelationRule(ruleDeleteRequest); + + PowerMock.verifyAll(); + } + + @Test + public void deleteCorrelationRule_normal() throws Exception { + RuleDeleteRequest ruleDeleteRequest = createRuleDeleteRequest("rule_" + System.currentTimeMillis()); + RuleUpdateRequest ruleCreateRequest = createRuleUpdateRequest("rule_" + System.currentTimeMillis(), + "desc", "contents", 1); + CorrelationRule correlationRule = convertUpdateRequest2CorrelationRule(ruleCreateRequest); + + EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn( + correlationRuleDaoMock).anyTimes(); + EasyMock.expect(correlationRuleDaoMock.getRuleByRid(ruleDeleteRequest.getRuleId())) + .andReturn(correlationRule); + EasyMock.expect(engineWrapperMock.deleteRuleFromEngine(correlationRule.getPackageName())).andReturn(true); + correlationRuleDaoMock.deleteRule(correlationRule); + EasyMock.expectLastCall(); + + PowerMock.replayAll(); + + ruleMgtWrapper.deleteCorrelationRule(ruleDeleteRequest); + + PowerMock.verifyAll(); + } + + @Test + public void getCorrelationRuleByCondition_data_format_exception() throws Exception { + thrown.expect(DataFormatException.class); + thrown.expectMessage((I18nProxy.RULE_MANAGEMENT_CREATE_QUERY_SQL_FAILED)); + + EasyMock.expect(correlationRuleQueryDaoMock.getCorrelationRulesByCondition(EasyMock.anyObject( + RuleQueryCondition.class))) + .andThrow(new DataFormatException(I18nProxy.RULE_MANAGEMENT_CREATE_QUERY_SQL_FAILED)); + + PowerMock.replay(correlationRuleQueryDaoMock, CorrelationRuleQueryDao.class); + + ruleMgtWrapper.getCorrelationRuleByCondition(new RuleQueryCondition()); + + PowerMock.verify(correlationRuleQueryDaoMock, CorrelationRuleQueryDao.class); + } + + @Test + public void getCorrelationRuleByCondition_db_exception() throws Exception { + thrown.expect(DbException.class); + thrown.expectMessage((I18nProxy.RULE_MANAGEMENT_QUERY_RULE_FAILED)); + + EasyMock.expect(correlationRuleQueryDaoMock.getCorrelationRulesByCondition(EasyMock.anyObject( + RuleQueryCondition.class))) + .andThrow(new DbException(I18nProxy.RULE_MANAGEMENT_QUERY_RULE_FAILED)); + + PowerMock.replay(correlationRuleQueryDaoMock, CorrelationRuleQueryDao.class); + + ruleMgtWrapper.getCorrelationRuleByCondition(new RuleQueryCondition()); + + PowerMock.verify(correlationRuleQueryDaoMock, CorrelationRuleQueryDao.class); + } + + @Test + public void getCorrelationRuleByCondition_normal() throws Exception { + List correlationRuleList = new ArrayList(10); + for (int i = 0; i < 10; ++i) { + CorrelationRule correlationRule = new CorrelationRule(); + correlationRule.setContent("content" + i); + correlationRule.setName("name" + i); + correlationRule.setRid("rule_" + i); + correlationRule.setIsManual(i % 2); + correlationRule.setEngineType("engineType" + (i % 2 + 1)); + correlationRule.setEngineId("engineId" + i); + correlationRule.setCreateTime(new Date()); + correlationRule.setCreator(USER_NAME); + correlationRule.setDescription("description" + i); + correlationRule.setPackageName("package" + i); + correlationRuleList.add(correlationRule); + } + + EasyMock.expect(correlationRuleQueryDaoMock.getCorrelationRulesByCondition(EasyMock.anyObject( + RuleQueryCondition.class))).andReturn(correlationRuleList); + + PowerMock.replay(correlationRuleQueryDaoMock, CorrelationRuleQueryDao.class); + + RuleQueryListResponse response = ruleMgtWrapper.getCorrelationRuleByCondition(new RuleQueryCondition()); + + PowerMock.verify(correlationRuleQueryDaoMock, CorrelationRuleQueryDao.class); + + assertThat(response.getTotalCount(), is(10)); + + for (int i = 0; i < 10; ++i) { + assertThat(response.getCorrelationRules().get(i).getRuleId(), + equalTo(correlationRuleList.get(i).getRid())); + } + } + + private RuleCreateRequest createRuleCreateRequest(String ruleName, String description, String content, + int enabled) { + RuleCreateRequest rcr; + rcr = new RuleCreateRequest(); + rcr.setRuleName(ruleName); + rcr.setDescription(description); + rcr.setContent(content); + rcr.setEnabled(enabled); + return rcr; + } + + private RuleUpdateRequest createRuleUpdateRequest(String ruleId, String description, String content, int enabled) { + RuleUpdateRequest ruleUpdateRequest = new RuleUpdateRequest(); + ruleUpdateRequest.setRuleId(ruleId); + ruleUpdateRequest.setDescription(description); + ruleUpdateRequest.setContent(content); + ruleUpdateRequest.setEnabled(enabled); + return ruleUpdateRequest; + } + + private RuleDeleteRequest createRuleDeleteRequest(String ruleId) { + RuleDeleteRequest ruleDeleteRequest = new RuleDeleteRequest(); + ruleDeleteRequest.setRuleId(ruleId); + return ruleDeleteRequest; + } + + private CorrelationRule convertCreateRequest2CorrelationRule(RuleCreateRequest ruleCreateRequest) { + CorrelationRule correlationRule = new CorrelationRule(); + correlationRule.setContent(ruleCreateRequest.getContent()); + correlationRule.setDescription(ruleCreateRequest.getDescription()); + correlationRule.setName(ruleCreateRequest.getRuleName()); + correlationRule.setCreator(USER_NAME); + correlationRule.setModifier(USER_NAME); + correlationRule.setEnabled(ruleCreateRequest.getEnabled()); + return correlationRule; + } + + private CorrelationRule convertUpdateRequest2CorrelationRule(RuleUpdateRequest ruleUpdateRequest) { + CorrelationRule correlationRule = new CorrelationRule(); + correlationRule.setRid(ruleUpdateRequest.getRuleId()); + correlationRule.setContent(ruleUpdateRequest.getContent()); + correlationRule.setDescription(ruleUpdateRequest.getDescription()); + correlationRule.setEnabled(ruleUpdateRequest.getEnabled()); + correlationRule.setUpdateTime(new Date()); + correlationRule.setModifier(USER_NAME); + return correlationRule; + } +} \ No newline at end of file -- cgit 1.2.3-korg