+\r
+ @Test\r
+ public void testConvertVesAlarm2AlarmInfo() throws Exception {\r
+ VesAlarm vesAlarm = new VesAlarm();\r
+ vesAlarm.setEventId("eventId");\r
+ vesAlarm.setEventName("eventName");\r
+ vesAlarm.setStartEpochMicrosec(1L);\r
+ vesAlarm.setLastEpochMicrosec(1L);\r
+ vesAlarm.setSourceId("sourceId");\r
+ vesAlarm.setSourceName("sourceName");\r
+ vesAlarm.setRootFlag(0);\r
+ vesAlarm.setAlarmIsCleared(1);\r
+\r
+ AlarmInfo alarmInfo = Whitebox.invokeMethod(droolsEngine, "convertVesAlarm2AlarmInfo", vesAlarm);\r
+\r
+ assertThat(alarmInfo.getAlarmIsCleared(), is(1));\r
+ assertThat(alarmInfo.getSourceName(), equalTo("sourceName"));\r
+ assertThat(alarmInfo.getSourceId(), equalTo("sourceId"));\r
+ assertThat(alarmInfo.getStartEpochMicroSec(), is(1L));\r
+ assertThat(alarmInfo.getLastEpochMicroSec(), is(1L));\r
+ assertThat(alarmInfo.getEventName(), equalTo("eventName"));\r
+ assertThat(alarmInfo.getEventId(), equalTo("eventId"));\r
+ assertThat(alarmInfo.getRootFlag(), is(0));\r
+ }\r
+\r
+ @Test\r
+ public void testQueryPackagesFromEngine() throws CorrelationException {\r
+\r
+ DeployRuleRequest rule = new DeployRuleRequest();\r
+ rule.setContent("package packageCheck; rule \"test\" when eval(1==1) then System.out.println(1); end");\r
+ rule.setLoopControlName(UUID.randomUUID().toString());\r
+\r
+ droolsEngine.deployRule(rule);\r
+\r
+ List<String> packages = droolsEngine.queryPackagesFromEngine();\r
+\r
+ assertThat(packages.contains("packageCheck"), is(true));\r
+ }\r
+}\r
+\r
+class RuleMgtWrapperStub extends RuleMgtWrapper {\r
+ private List<CorrelationRule> rules;\r
+\r
+ public RuleMgtWrapperStub() {\r
+ rules = new ArrayList<>();\r
+ CorrelationRule rule = new CorrelationRule();\r
+ rule.setEnabled(1);\r
+ rule.setContent("package org.onap.holmes;");\r
+ rule.setPackageName("UT");\r
+ rule.setClosedControlLoopName(UUID.randomUUID().toString());\r
+ rules.add(rule);\r
+ }\r
+\r
+ public List<CorrelationRule> getRules() {\r
+ return rules;\r
+ }\r
+\r
+ public void setRules(List<CorrelationRule> rules) {\r
+ this.rules = rules;\r
+ }\r
+\r
+ @Override\r
+ public List<CorrelationRule> queryRuleByEnable(int enabled) throws CorrelationException {\r
+ return rules.stream().filter(rule -> rule.getEnabled() == enabled).collect(Collectors.toList());\r
+ }\r
+}\r
+\r
+class AlarmInfoDaoStub extends AlarmInfoDao {\r
+\r
+ private List<AlarmInfo> alarms;\r
+\r
+ public AlarmInfoDaoStub() {\r
+ alarms = new ArrayList<>();\r
+ AlarmInfo info = new AlarmInfo();\r
+ info.setEventId("eventId");\r
+ info.setEventName("eventName");\r
+ info.setStartEpochMicroSec(1L);\r
+ info.setLastEpochMicroSec(1L);\r
+ info.setSourceId("sourceId");\r
+ info.setSourceName("sourceName");\r
+ info.setRootFlag(0);\r
+ info.setAlarmIsCleared(1);\r
+ alarms.add(info);\r
+ }\r
+\r
+ @Override\r
+ protected String addAlarm(AlarmInfo alarmInfo) {\r
+ alarms.add(alarmInfo);\r
+ return null;\r
+ }\r
+\r
+ @Override\r
+ protected List<AlarmInfo> queryAlarm() {\r
+ return alarms;\r
+ }\r
+\r
+ @Override\r
+ protected int deleteAlarmByAlarmIsCleared(int alarmIsCleared) {\r
+ return 1;\r
+ }\r
+}\r
+\r
+class DbDaoUtilStub extends DbDaoUtil {\r
+ private AlarmInfoDao dao = new AlarmInfoDaoStub();\r
+ @Override\r
+ public <T> T getJdbiDaoByOnDemand(Class<T> daoClazz) {\r
+\r
+ return (T)dao;\r
+\r
+ }\r