Fix unit test

Issue-ID:HOLMES-51

Change-Id: I1fc4a6da12f085cbb8ad2a36ef630823e5707ae3
Signed-off-by:youbowu <wu.youbo@zte.com.cn>

diff --git a/holmes-actions/src/main/java/org/openo/holmes/common/producer/MQProducer.java b/holmes-actions/src/main/java/org/openo/holmes/common/producer/MQProducer.java
index 8cb3d2b..21b62f3 100644 (file)
--- a/holmes-actions/src/main/java/org/openo/holmes/common/producer/MQProducer.java
+++ b/holmes-actions/src/main/java/org/openo/holmes/common/producer/MQProducer.java
@@ -55,14 +55,18 @@ public class MQProducer {
         Serializable msgEntity = (Serializable) t;\r
         Connection connection = null;\r
         Session session;\r
-        Destination destination;\r
+        Destination destination = null;\r
         MessageProducer messageProducer;\r
 \r
         try {\r
             connection = connectionFactory.createConnection();\r
             connection.start();\r
             session = connection.createSession(true, Session.AUTO_ACKNOWLEDGE);\r
-            destination = session.createTopic(AlarmConst.MQ_TOPIC_NAME_ALARMS_CORRELATION);\r
+            if (t instanceof CorrelationResult) {\r
+                destination = session.createTopic(AlarmConst.MQ_TOPIC_NAME_ALARMS_CORRELATION);\r
+            } else if (t instanceof Alarm) {\r
+                destination = session.createTopic(AlarmConst.MQ_TOPIC_NAME_ALARM);\r
+            }\r
             messageProducer = session.createProducer(destination);\r
             ObjectMessage message = session.createObjectMessage(msgEntity);\r
             messageProducer.send(message);\r

diff --git a/holmes-actions/src/test/java/org/openo/holmes/common/api/stat/AlarmTest.java b/holmes-actions/src/test/java/org/openo/holmes/common/api/stat/AlarmTest.java
index 49858ab..c42ee60 100644 (file)
--- a/holmes-actions/src/test/java/org/openo/holmes/common/api/stat/AlarmTest.java
+++ b/holmes-actions/src/test/java/org/openo/holmes/common/api/stat/AlarmTest.java
@@ -16,6 +16,7 @@
 
 package org.openo.holmes.common.api.stat;
 
+import java.util.Date;
 import static org.hamcrest.core.IsEqual.equalTo;
 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertThat;
@@ -84,8 +85,16 @@ public class AlarmTest {
 
     @Test
     public void testToString() throws Exception {
-        final Alarm alarmTemp = new Alarm();
-        assertThat(alarmTemp.toString(), equalTo(alarm.toString()));
+        Alarm alarmTempA = new Alarm();
+        Alarm alarmTempB = new Alarm();
+        Date date = new Date();
+        alarmTempA.setClearedTime(date);
+        alarmTempA.setRaisedTime(date);
+        alarmTempA.setRaisedServerTime(date);
+        alarmTempB.setClearedTime(date);
+        alarmTempB.setRaisedTime(date);
+        alarmTempB.setRaisedServerTime(date);
+        assertThat(alarmTempA.toString(),equalTo(alarmTempB.toString()));
     }
 
     @Test