engine-d/src/test/java/org/onap/holmes/engine/utils/AlarmUtilTest.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.engine.utils;\r
  18 \r
  19 import org.hamcrest.core.IsEqual;\r
  20 import org.hamcrest.core.IsNull;\r
  21 import org.junit.Assert;\r
  22 import org.junit.Before;\r
  23 import org.junit.Test;\r
  24 import org.onap.holmes.common.api.stat.Alarm;\r
  25 import org.powermock.api.easymock.PowerMock;\r
  26 import org.powermock.reflect.Whitebox;\r
  27 \r
  28 import java.util.HashMap;\r
  29 import java.util.Map;\r
  30 \r
  31 public class AlarmUtilTest {\r
  32     private AlarmUtil alarmUtil;\r
  33     private final Map<String, Map<String, Integer>> rootPriorityMap = new HashMap<String, Map<String, Integer>>();\r
  34     private final Map<String, String> saveRuleMsg = new HashMap<String, String>();\r
  35 \r
  36     @Before\r
  37     public void setUp() {\r
  38         alarmUtil = AlarmUtil.getInstance();\r
  39         Whitebox.setInternalState(alarmUtil,"rootPriorityMap",rootPriorityMap);\r
  40         Whitebox.setInternalState(alarmUtil,"saveRuleMsg",saveRuleMsg);\r
  41         PowerMock.resetAll();\r
  42     }\r
  43 \r
  44     @Test\r
  45     public void getInstance() {\r
  46         AlarmUtil instance = AlarmUtil.getInstance();\r
  47         Assert.assertThat(instance,IsNull.<AlarmUtil>notNullValue());\r
  48     }\r
  49 \r
  50     @Test\r
  51     public void equipTypeFilter_is_nullstr() {\r
  52         String probableCauseStr = "null";\r
  53         String equipType = "equipType";\r
  54         Alarm alarm = new Alarm();\r
  55         boolean filter = alarmUtil.equipTypeFilter(probableCauseStr, equipType, alarm);\r
  56         Assert.assertThat(filter, IsEqual.equalTo(true));\r
  57     }\r
  58 \r
  59     @Test\r
  60     public void equipTypeFilter_equals_alarm() {\r
  61         String probableCauseStr = "11,4567";\r
  62         String equipType = "ee,equipType";\r
  63         Alarm alarm = new Alarm();\r
  64         alarm.setProbableCause(4567);\r
  65         alarm.setEquipType("equipType");\r
  66         boolean filter = alarmUtil.equipTypeFilter(probableCauseStr, equipType, alarm);\r
  67         Assert.assertThat(filter, IsEqual.equalTo(true));\r
  68     }\r
  69 \r
  70     @Test\r
  71     public void equipTypeFilter_not_equals_alarm() {\r
  72         String probableCauseStr = "11,45";\r
  73         String equipType = "ee,equipType";\r
  74         Alarm alarm = new Alarm();\r
  75         alarm.setProbableCause(4567);\r
  76         alarm.setEquipType("equipType");\r
  77         boolean filter = alarmUtil.equipTypeFilter(probableCauseStr, equipType, alarm);\r
  78         Assert.assertThat(filter, IsEqual.equalTo(false));\r
  79     }\r
  80 \r
  81     @Test\r
  82     public void getPriority_rootprioritymap_containskey_ruleid() {\r
  83         String ruleId = "1";\r
  84         String probableCauseStr = "11,4567";\r
  85         String rootAlarmFeatureStr = "0,1";\r
  86         String equipTypeStr = "ee,equipType";\r
  87         Alarm alarm = new Alarm();\r
  88 \r
  89         Map<String, Integer> map = new HashMap<String, Integer>();\r
  90         map.put("11-ee", 0);\r
  91         map.put("4567-equipType", 1);\r
  92         rootPriorityMap.put(ruleId, map);\r
  93 \r
  94         saveRuleMsg.put(ruleId, "11ee0");\r
  95 \r
  96         Integer priority = alarmUtil.getPriority(ruleId, probableCauseStr, rootAlarmFeatureStr, equipTypeStr, alarm);\r
  97         Assert.assertThat(priority,IsEqual.equalTo(0));\r
  98     }\r
  99 \r
 100     @Test\r
 101     public void getPriority_rootprioritymap_not_containskey_ruleid() {\r
 102         String ruleId = "1";\r
 103         String probableCauseStr = "11,4567";\r
 104         String rootAlarmFeatureStr = "0,1";\r
 105         String equipTypeStr = "ee,equipType";\r
 106         Alarm alarm = new Alarm();\r
 107 \r
 108         saveRuleMsg.put(ruleId, "11ee0");\r
 109 \r
 110         Integer priority = alarmUtil.getPriority(ruleId, probableCauseStr, rootAlarmFeatureStr, equipTypeStr, alarm);\r
 111         Assert.assertThat(priority,IsEqual.equalTo(0));\r
 112     }\r
 113 \r
 114     @Test\r
 115     public void getPriority_priority_is_not_null() {\r
 116         String ruleId = "1";\r
 117         String probableCauseStr = "11,4567";\r
 118         String rootAlarmFeatureStr = "1,1";\r
 119         String equipTypeStr = "ee,equipType";\r
 120         Alarm alarm = new Alarm();\r
 121         alarm.setProbableCause(11);\r
 122         alarm.setEquipType("ee");\r
 123 \r
 124         saveRuleMsg.put(ruleId, "11ee0");\r
 125 \r
 126         Integer priority = alarmUtil.getPriority(ruleId, probableCauseStr, rootAlarmFeatureStr, equipTypeStr, alarm);\r
 127         Assert.assertThat(priority,IsEqual.equalTo(1));\r
 128     }\r
 129 }\r