rulemgt/src/test/java/org/onap/holmes/rulemgt/wrapper/RuleMgtWrapperTest.java

   1 /**\r
   2  * Copyright 2017 ZTE Corporation.\r
   3  *\r
   4  * Licensed under the Apache License, Version 2.0 (the "License");\r
   5  * you may not use this file except in compliance with the License.\r
   6  * You may obtain a copy of the License at\r
   7  *\r
   8  *      http://www.apache.org/licenses/LICENSE-2.0\r
   9  *\r
  10  * Unless required by applicable law or agreed to in writing, software\r
  11  * distributed under the License is distributed on an "AS IS" BASIS,\r
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  13  * See the License for the specific language governing permissions and\r
  14  * limitations under the License.\r
  15  */\r
  16 \r
  17 package org.onap.holmes.rulemgt.wrapper;\r
  18 \r
  19 import static org.hamcrest.MatcherAssert.assertThat;\r
  20 import static org.hamcrest.Matchers.equalTo;\r
  21 import static org.hamcrest.Matchers.is;\r
  22 \r
  23 import java.util.ArrayList;\r
  24 import java.util.Date;\r
  25 import java.util.List;\r
  26 import org.easymock.EasyMock;\r
  27 import org.junit.Before;\r
  28 import org.junit.Rule;\r
  29 import org.junit.Test;\r
  30 import org.junit.rules.ExpectedException;\r
  31 import org.junit.runner.RunWith;\r
  32 import org.onap.holmes.rulemgt.bean.request.CorrelationCheckRule4Engine;\r
  33 import org.onap.holmes.rulemgt.db.CorrelationRuleDao;\r
  34 import org.onap.holmes.common.api.entity.CorrelationRule;\r
  35 import org.onap.holmes.common.exception.CorrelationException;\r
  36 import org.onap.holmes.common.utils.DbDaoUtil;\r
  37 import org.onap.holmes.rulemgt.bean.request.CorrelationDeployRule4Engine;\r
  38 import org.onap.holmes.rulemgt.bean.request.RuleCreateRequest;\r
  39 import org.onap.holmes.rulemgt.bean.request.RuleDeleteRequest;\r
  40 import org.onap.holmes.rulemgt.bean.request.RuleQueryCondition;\r
  41 import org.onap.holmes.rulemgt.bean.request.RuleUpdateRequest;\r
  42 import org.onap.holmes.rulemgt.bean.response.RuleAddAndUpdateResponse;\r
  43 import org.onap.holmes.rulemgt.bean.response.RuleQueryListResponse;\r
  44 import org.onap.holmes.rulemgt.bolt.enginebolt.EngineWrapper;\r
  45 import org.onap.holmes.rulemgt.db.CorrelationRuleQueryDao;\r
  46 import org.powermock.api.easymock.PowerMock;\r
  47 import org.powermock.modules.junit4.rule.PowerMockRule;\r
  48 import org.powermock.reflect.Whitebox;\r
  49 \r
  50 public class RuleMgtWrapperTest {\r
  51 \r
  52     @Rule\r
  53     public ExpectedException thrown = ExpectedException.none();\r
  54     @Rule\r
  55     public PowerMockRule powerMockRule = new PowerMockRule();\r
  56 \r
  57     private RuleMgtWrapper ruleMgtWrapper;\r
  58 \r
  59     private EngineWrapper engineWrapperMock;\r
  60 \r
  61     private DbDaoUtil dbDaoUtilMock;\r
  62 \r
  63     private CorrelationRuleQueryDao correlationRuleQueryDaoMock;\r
  64 \r
  65     private CorrelationRuleDao correlationRuleDaoMock;\r
  66 \r
  67     private static final String USER_NAME = "admin";\r
  68 \r
  69     @Before\r
  70     public void setUp() throws Exception {\r
  71 \r
  72         ruleMgtWrapper = new RuleMgtWrapper();\r
  73 \r
  74         engineWrapperMock = PowerMock.createMock(EngineWrapper.class);\r
  75         correlationRuleQueryDaoMock = PowerMock.createMock(CorrelationRuleQueryDao.class);\r
  76         dbDaoUtilMock = PowerMock.createMock(DbDaoUtil.class);\r
  77         correlationRuleDaoMock = PowerMock.createMock(CorrelationRuleDao.class);\r
  78 \r
  79         Whitebox.setInternalState(ruleMgtWrapper, "daoUtil", dbDaoUtilMock);\r
  80         Whitebox.setInternalState(ruleMgtWrapper, "correlationRuleQueryDao", correlationRuleQueryDaoMock);\r
  81         Whitebox.setInternalState(ruleMgtWrapper, "engineWarpper", engineWrapperMock);\r
  82         Whitebox.setInternalState(ruleMgtWrapper, "correlationRuleDao", correlationRuleDaoMock);\r
  83 \r
  84         PowerMock.resetAll();\r
  85     }\r
  86 \r
  87     @Test\r
  88     public void initDaoUtil_normal() {\r
  89         ruleMgtWrapper.initDaoUtil();\r
  90     }\r
  91 \r
  92     @Test\r
  93     public void addCorrelationRule_name_is_null() throws Exception {\r
  94         thrown.expect(CorrelationException.class);\r
  95         thrown.expectMessage("The name of the rule can not be empty.");\r
  96 \r
  97         ruleMgtWrapper.addCorrelationRule(USER_NAME, createRuleCreateRequest(null, "This is a rule for testing.",\r
  98                 "Mocked contents.", 0));\r
  99     }\r
 100 \r
 101     @Test\r
 102     public void addCorrelationRule_request_null() throws Exception {\r
 103         thrown.expect(CorrelationException.class);\r
 104         thrown.expectMessage("The request object can not be empty!");\r
 105 \r
 106         ruleMgtWrapper.addCorrelationRule(USER_NAME, null);\r
 107     }\r
 108 \r
 109     @Test\r
 110     public void addCorrelationRule_name_is_empty() throws Exception {\r
 111         thrown.expect(CorrelationException.class);\r
 112         thrown.expectMessage("The name of the rule can not be empty.");\r
 113 \r
 114         ruleMgtWrapper.addCorrelationRule("admin", createRuleCreateRequest("", "This is a rule for testing.",\r
 115                 "Mocked contents.", 0));\r
 116     }\r
 117 \r
 118     @Test\r
 119     public void addCorrelationRule_content_is_empty() throws Exception {\r
 120         thrown.expect(CorrelationException.class);\r
 121         thrown.expectMessage("The contents of the rule can not be empty!");\r
 122 \r
 123         ruleMgtWrapper.addCorrelationRule("admin", createRuleCreateRequest("test", "This is a rule for testing.",\r
 124                 "", 0));\r
 125     }\r
 126 \r
 127     @Test\r
 128     public void addCorrelationRule_enabled_is_off_limit() throws Exception {\r
 129         thrown.expect(CorrelationException.class);\r
 130         thrown.expectMessage("Invalid rule status. Only 0 (disabled) and 1 (enabled) are allowed.");\r
 131 \r
 132         ruleMgtWrapper.addCorrelationRule("admin", createRuleCreateRequest("test", "This is a rule for testing.",\r
 133                 "Mocked contents.", 3));\r
 134     }\r
 135 \r
 136     @Test\r
 137     public void addCorrelationRule_duplicated_rule() throws Exception {\r
 138 \r
 139         final String ruleName = "Rule-001";\r
 140 \r
 141         RuleCreateRequest ruleCreateRequest = createRuleCreateRequest(ruleName, "This is a rule for testing.",\r
 142                 "Mocked contents.", 0);\r
 143         CorrelationRule correlationRule = convertCreateRequest2CorrelationRule(ruleCreateRequest);\r
 144 \r
 145         thrown.expect(CorrelationException.class);\r
 146         thrown.expectMessage("A rule with the same name already exists.");\r
 147 \r
 148         EasyMock.expect(correlationRuleDaoMock.queryRuleByRuleName(ruleName)).andReturn(correlationRule);\r
 149         PowerMock.replayAll();\r
 150 \r
 151         ruleMgtWrapper.addCorrelationRule("admin", ruleCreateRequest);\r
 152 \r
 153         PowerMock.verifyAll();\r
 154     }\r
 155 \r
 156     @Test\r
 157     public void addCorrelationRule_normal() throws Exception {\r
 158         final String ruleName = "Rule-001";\r
 159 \r
 160         RuleCreateRequest ruleCreateRequest = createRuleCreateRequest(ruleName, "This is a rule for testing.",\r
 161                 "Mocked contents.", 1);\r
 162 \r
 163         CorrelationRule correlationRuleRet = new CorrelationRule();\r
 164         correlationRuleRet.setRid("rule_" + System.currentTimeMillis());\r
 165 \r
 166         EasyMock.expect(correlationRuleDaoMock.queryRuleByRuleName(ruleName)).andReturn(null);\r
 167         EasyMock.expect(engineWrapperMock.checkRuleFromEngine(EasyMock.anyObject(CorrelationCheckRule4Engine.class)))\r
 168                 .andReturn(true);\r
 169         EasyMock.expect(engineWrapperMock.deployEngine(EasyMock.anyObject(CorrelationDeployRule4Engine.class)))\r
 170                 .andReturn("package-001");\r
 171         EasyMock.expect(correlationRuleDaoMock.saveRule(EasyMock.anyObject(CorrelationRule.class)))\r
 172                 .andReturn(correlationRuleRet);\r
 173 \r
 174         PowerMock.replayAll();\r
 175 \r
 176         RuleAddAndUpdateResponse response = ruleMgtWrapper.addCorrelationRule("admin", ruleCreateRequest);\r
 177         PowerMock.verifyAll();\r
 178 \r
 179         assertThat(response.getRuleId(), equalTo(correlationRuleRet.getRid()));\r
 180     }\r
 181 \r
 182     @Test\r
 183     public void updateCorrelationRule_param_null() throws Exception {\r
 184         thrown.expect(CorrelationException.class);\r
 185         thrown.expectMessage("The request object can not be empty!");\r
 186 \r
 187         ruleMgtWrapper.updateCorrelationRule(USER_NAME, null);\r
 188     }\r
 189 \r
 190     @Test\r
 191     public void updateCorrelationRule_normal() throws Exception {\r
 192         CorrelationRule oldCorrelationRule = new CorrelationRule();\r
 193         oldCorrelationRule.setRid("rule_1");\r
 194         oldCorrelationRule.setName("name");\r
 195         oldCorrelationRule.setDescription("des1");\r
 196         oldCorrelationRule.setContent("content");\r
 197         oldCorrelationRule.setPackageName("testName");\r
 198         oldCorrelationRule.setEnabled(1);\r
 199         RuleUpdateRequest ruleUpdateRequest = createRuleUpdateRequest("rule_1", "des2", "contetnt2", 1);\r
 200 \r
 201         EasyMock.expect(correlationRuleDaoMock.queryRuleByRid("rule_1")).andReturn(oldCorrelationRule);\r
 202         EasyMock.expect(engineWrapperMock.deleteRuleFromEngine("testName")).andReturn(true);\r
 203         correlationRuleDaoMock.updateRule(EasyMock.anyObject(CorrelationRule.class));\r
 204         EasyMock.expectLastCall();\r
 205         EasyMock.expect(engineWrapperMock.checkRuleFromEngine(EasyMock.anyObject(CorrelationCheckRule4Engine.class)))\r
 206                 .andReturn(true);\r
 207         EasyMock.expect(engineWrapperMock.deployEngine(EasyMock.anyObject(CorrelationDeployRule4Engine.class)))\r
 208                 .andReturn("packageName1");\r
 209         PowerMock.replayAll();\r
 210 \r
 211         ruleMgtWrapper.updateCorrelationRule(USER_NAME, ruleUpdateRequest);\r
 212 \r
 213         PowerMock.verifyAll();\r
 214 \r
 215         assertThat(oldCorrelationRule.getRid(), equalTo(ruleUpdateRequest.getRuleId()));\r
 216     }\r
 217 \r
 218     @Test\r
 219     public void updateCorrelationRule_param_no_change() throws Exception {\r
 220         CorrelationRule oldCorrelationRule = new CorrelationRule();\r
 221         oldCorrelationRule.setRid("rule_1");\r
 222         oldCorrelationRule.setName("name");\r
 223         oldCorrelationRule.setDescription("des1");\r
 224         oldCorrelationRule.setContent("content");\r
 225         oldCorrelationRule.setPackageName("testName");\r
 226         oldCorrelationRule.setEnabled(1);\r
 227         RuleUpdateRequest ruleUpdateRequest = createRuleUpdateRequest("rule_1", "des1", "content", 1);\r
 228 \r
 229         EasyMock.expect(correlationRuleDaoMock.queryRuleByRid("rule_1")).andReturn(oldCorrelationRule);\r
 230 \r
 231         PowerMock.replayAll();\r
 232 \r
 233         ruleMgtWrapper.updateCorrelationRule(USER_NAME, ruleUpdateRequest);\r
 234 \r
 235         PowerMock.verifyAll();\r
 236 \r
 237         assertThat(oldCorrelationRule.getRid(), equalTo(ruleUpdateRequest.getRuleId()));\r
 238     }\r
 239 \r
 240     @Test\r
 241     public void updateCorrelationRule_rule_not_exist() throws Exception {\r
 242         thrown.expect(CorrelationException.class);\r
 243         thrown.expectMessage("You're trying to update a rule which does not exist in the system.");\r
 244 \r
 245         EasyMock.expect(correlationRuleDaoMock.queryRuleByRid(EasyMock.anyObject(String.class))).andReturn(null);\r
 246 \r
 247         PowerMock.replayAll();\r
 248 \r
 249         ruleMgtWrapper.updateCorrelationRule(USER_NAME, new RuleUpdateRequest());\r
 250 \r
 251         PowerMock.verifyAll();\r
 252     }\r
 253 \r
 254     @Test\r
 255     public void deleteCorrelationRule_request_null() throws Exception {\r
 256         thrown.expect(CorrelationException.class);\r
 257         thrown.expectMessage("The request object can not be empty!");\r
 258 \r
 259         ruleMgtWrapper.deleteCorrelationRule(null);\r
 260     }\r
 261 \r
 262     @Test\r
 263     public void deleteCorrelationRule_rule_not_exit() throws Exception {\r
 264         thrown.expect(CorrelationException.class);\r
 265         thrown.expectMessage("You're trying to delete a rule which does not exist in the system.");\r
 266 \r
 267         RuleDeleteRequest ruleDeleteRequest = createRuleDeleteRequest("rule_" + System.currentTimeMillis());\r
 268 \r
 269         EasyMock.expect(dbDaoUtilMock.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn(\r
 270                 correlationRuleDaoMock).anyTimes();\r
 271         EasyMock.expect(correlationRuleDaoMock.queryRuleByRid(ruleDeleteRequest.getRuleId()))\r
 272                 .andReturn(null);\r
 273 \r
 274         PowerMock.replayAll();\r
 275 \r
 276         ruleMgtWrapper.deleteCorrelationRule(ruleDeleteRequest);\r
 277 \r
 278         PowerMock.verifyAll();\r
 279     }\r
 280 \r
 281     @Test\r
 282     public void deleteCorrelationRule_normal() throws Exception {\r
 283         RuleDeleteRequest ruleDeleteRequest = createRuleDeleteRequest("rule_" + System.currentTimeMillis());\r
 284         CorrelationRule correlationRule = new CorrelationRule();\r
 285         correlationRule.setEnabled(1);\r
 286         EasyMock.expect(correlationRuleDaoMock.queryRuleByRid(ruleDeleteRequest.getRuleId()))\r
 287                 .andReturn(correlationRule);\r
 288         EasyMock.expect(engineWrapperMock.deleteRuleFromEngine(EasyMock.anyObject(String.class))).andReturn(true);\r
 289         correlationRuleDaoMock.deleteRule(EasyMock.anyObject(CorrelationRule.class));\r
 290         EasyMock.expectLastCall();\r
 291         PowerMock.replayAll();\r
 292 \r
 293         ruleMgtWrapper.deleteCorrelationRule(ruleDeleteRequest);\r
 294 \r
 295         PowerMock.verifyAll();\r
 296     }\r
 297 \r
 298     @Test\r
 299     public void getCorrelationRuleByCondition_data_format_exception() throws Exception {\r
 300         thrown.expect(CorrelationException.class);\r
 301         thrown.expectMessage("An error occurred while building the query SQL.");\r
 302 \r
 303         EasyMock.expect(correlationRuleQueryDaoMock.getCorrelationRulesByCondition(EasyMock.anyObject(\r
 304                 RuleQueryCondition.class)))\r
 305                 .andThrow(new CorrelationException("An error occurred while building the query SQL."));\r
 306 \r
 307         PowerMock.replay(correlationRuleQueryDaoMock, CorrelationRuleQueryDao.class);\r
 308 \r
 309         ruleMgtWrapper.getCorrelationRuleByCondition(new RuleQueryCondition());\r
 310 \r
 311         PowerMock.verify(correlationRuleQueryDaoMock, CorrelationRuleQueryDao.class);\r
 312     }\r
 313 \r
 314     @Test\r
 315     public void getCorrelationRuleByCondition_db_exception() throws Exception {\r
 316         thrown.expect(CorrelationException.class);\r
 317         thrown.expectMessage("Failed to query the rule.");\r
 318 \r
 319         EasyMock.expect(correlationRuleQueryDaoMock.getCorrelationRulesByCondition(EasyMock.anyObject(\r
 320                 RuleQueryCondition.class)))\r
 321                 .andThrow(new CorrelationException("Failed to query the rule."));\r
 322 \r
 323         PowerMock.replay(correlationRuleQueryDaoMock, CorrelationRuleQueryDao.class);\r
 324 \r
 325         ruleMgtWrapper.getCorrelationRuleByCondition(new RuleQueryCondition());\r
 326 \r
 327         PowerMock.verify(correlationRuleQueryDaoMock, CorrelationRuleQueryDao.class);\r
 328     }\r
 329 \r
 330     @Test\r
 331     public void getCorrelationRuleByCondition_normal() throws Exception {\r
 332         List<CorrelationRule> correlationRuleList = new ArrayList<CorrelationRule>(10);\r
 333         for (int i = 0; i < 10; ++i) {\r
 334             CorrelationRule correlationRule = new CorrelationRule();\r
 335             correlationRule.setContent("content" + i);\r
 336             correlationRule.setName("name" + i);\r
 337             correlationRule.setRid("rule_" + i);\r
 338             correlationRule.setEngineType("engineType" + (i % 2 + 1));\r
 339             correlationRule.setEngineID("engineId" + i);\r
 340             correlationRule.setCreateTime(new Date());\r
 341             correlationRule.setCreator(USER_NAME);\r
 342             correlationRule.setDescription("description" + i);\r
 343             correlationRule.setPackageName("package" + i);\r
 344             correlationRuleList.add(correlationRule);\r
 345         }\r
 346 \r
 347         EasyMock.expect(correlationRuleQueryDaoMock.getCorrelationRulesByCondition(EasyMock.anyObject(\r
 348                 RuleQueryCondition.class))).andReturn(correlationRuleList);\r
 349 \r
 350         PowerMock.replay(correlationRuleQueryDaoMock, CorrelationRuleQueryDao.class);\r
 351 \r
 352         RuleQueryListResponse response = ruleMgtWrapper.getCorrelationRuleByCondition(new RuleQueryCondition());\r
 353 \r
 354         PowerMock.verify(correlationRuleQueryDaoMock, CorrelationRuleQueryDao.class);\r
 355 \r
 356         assertThat(response.getTotalCount(), is(10));\r
 357 \r
 358         for (int i = 0; i < 10; ++i) {\r
 359             assertThat(response.getCorrelationRules().get(i).getRuleId(),\r
 360                     equalTo(correlationRuleList.get(i).getRid()));\r
 361         }\r
 362     }\r
 363 \r
 364     private RuleCreateRequest createRuleCreateRequest(String ruleName, String description, String content,\r
 365             int enabled) {\r
 366         RuleCreateRequest rcr;\r
 367         rcr = new RuleCreateRequest();\r
 368         rcr.setRuleName(ruleName);\r
 369         rcr.setDescription(description);\r
 370         rcr.setContent(content);\r
 371         rcr.setEnabled(enabled);\r
 372         return rcr;\r
 373     }\r
 374 \r
 375     private RuleUpdateRequest createRuleUpdateRequest(String ruleId, String description, String content, int enabled) {\r
 376         RuleUpdateRequest ruleUpdateRequest = new RuleUpdateRequest();\r
 377         ruleUpdateRequest.setRuleId(ruleId);\r
 378         ruleUpdateRequest.setDescription(description);\r
 379         ruleUpdateRequest.setContent(content);\r
 380         ruleUpdateRequest.setEnabled(enabled);\r
 381         return ruleUpdateRequest;\r
 382     }\r
 383 \r
 384     private RuleDeleteRequest createRuleDeleteRequest(String ruleId) {\r
 385         RuleDeleteRequest ruleDeleteRequest = new RuleDeleteRequest();\r
 386         ruleDeleteRequest.setRuleId(ruleId);\r
 387         return ruleDeleteRequest;\r
 388     }\r
 389 \r
 390     private CorrelationRule convertCreateRequest2CorrelationRule(RuleCreateRequest ruleCreateRequest) {\r
 391         CorrelationRule correlationRule = new CorrelationRule();\r
 392         correlationRule.setContent(ruleCreateRequest.getContent());\r
 393         correlationRule.setDescription(ruleCreateRequest.getDescription());\r
 394         correlationRule.setName(ruleCreateRequest.getRuleName());\r
 395         correlationRule.setCreator(USER_NAME);\r
 396         correlationRule.setModifier(USER_NAME);\r
 397         correlationRule.setEnabled(ruleCreateRequest.getEnabled());\r
 398         return correlationRule;\r
 399     }\r
 400 \r
 401     private CorrelationRule convertUpdateRequest2CorrelationRule(RuleUpdateRequest ruleUpdateRequest) {\r
 402         CorrelationRule correlationRule = new CorrelationRule();\r
 403         correlationRule.setRid(ruleUpdateRequest.getRuleId());\r
 404         correlationRule.setContent(ruleUpdateRequest.getContent());\r
 405         correlationRule.setDescription(ruleUpdateRequest.getDescription());\r
 406         correlationRule.setEnabled(ruleUpdateRequest.getEnabled());\r
 407         correlationRule.setUpdateTime(new Date());\r
 408         correlationRule.setModifier(USER_NAME);\r
 409         return correlationRule;\r
 410     }\r
 411 }