1 files changed, 35 insertions, 2 deletions

diff --git a/engine-d/src/test/java/org/onap/holmes/engine/dmaap/DMaaPAlarmPollingTest.java b/engine-d/src/test/java/org/onap/holmes/engine/dmaap/DMaaPAlarmPollingTest.java
index f4804f3..896a384 100644
--- a/engine-d/src/test/java/org/onap/holmes/engine/dmaap/DMaaPAlarmPollingTest.java
+++ b/engine-d/src/test/java/org/onap/holmes/engine/dmaap/DMaaPAlarmPollingTest.java
@@ -15,6 +15,7 @@
  */
 package org.onap.holmes.engine.dmaap;
 
+import static org.hamcrest.CoreMatchers.is;
 import static org.hamcrest.core.IsEqual.equalTo;
 import static org.junit.Assert.*;
 
@@ -22,25 +23,31 @@ import java.lang.reflect.Field;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
+import org.onap.holmes.common.api.entity.AlarmInfo;
+import org.onap.holmes.common.api.stat.VesAlarm;
 import org.onap.holmes.dsa.dmaappolling.Subscriber;
+import org.onap.holmes.engine.db.AlarmInfoDao;
 import org.onap.holmes.engine.manager.DroolsEngine;
 import org.powermock.api.easymock.PowerMock;
 import org.powermock.core.classloader.annotations.PrepareForTest;
 import org.powermock.modules.junit4.PowerMockRunner;
+import org.powermock.reflect.Whitebox;
 
-@PrepareForTest({Subscriber.class, DroolsEngine.class})
+@PrepareForTest({Subscriber.class, DroolsEngine.class,DMaaPAlarmPolling.class})
 @RunWith(PowerMockRunner.class)
 public class DMaaPAlarmPollingTest {
 
     private DMaaPAlarmPolling dMaaPAlarmPolling;
     private Subscriber subscriber;
     private DroolsEngine droolsEngine;
+    private AlarmInfoDao alarmInfoDao;
 
     @Before
     public void setUp() {
         subscriber = PowerMock.createMock(Subscriber.class);
         droolsEngine = PowerMock.createMock(DroolsEngine.class);
-        dMaaPAlarmPolling = new DMaaPAlarmPolling(subscriber, droolsEngine);
+        alarmInfoDao = PowerMock.createMock(AlarmInfoDao.class);
+        dMaaPAlarmPolling = new DMaaPAlarmPolling(subscriber, droolsEngine,alarmInfoDao);
         PowerMock.replayAll();
     }
 
@@ -52,4 +59,30 @@ public class DMaaPAlarmPollingTest {
         assertThat(field.get(dMaaPAlarmPolling), equalTo(false));
     }
 
+    @Test
+    public void testGetAlarmInfo() throws Exception {
+        VesAlarm vesAlarm = new VesAlarm();
+        vesAlarm.setAlarmIsCleared(1);
+        vesAlarm.setSourceName("sourceName");
+        vesAlarm.setSourceId("sourceId");
+        vesAlarm.setStartEpochMicrosec(1L);
+        vesAlarm.setLastEpochMicrosec(1L);
+        vesAlarm.setEventName("eventName");
+        vesAlarm.setEventId("eventId");
+        vesAlarm.setRootFlag(0);
+
+        PowerMock.replayAll();
+        AlarmInfo alarmInfo = Whitebox.invokeMethod(dMaaPAlarmPolling,"getAlarmInfo",vesAlarm);
+        PowerMock.verifyAll();
+
+        assertThat(alarmInfo.getAlarmIsCleared(), is(1));
+        assertThat(alarmInfo.getSourceName(), equalTo("sourceName"));
+        assertThat(alarmInfo.getSourceId(), equalTo("sourceId"));
+        assertThat(alarmInfo.getStartEpochMicroSec(), is(1L));
+        assertThat(alarmInfo.getLastEpochMicroSec(), is(1L));
+        assertThat(alarmInfo.getEventName(), equalTo("eventName"));
+        assertThat(alarmInfo.getEventId(), equalTo("eventId"));
+        assertThat(alarmInfo.getRootFlag(), is(0));
+    }
+
 }
\ No newline at end of file