Fix unit test

Issue-ID:HOLMES-51

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

1 files changed, 6 insertions, 2 deletions

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
--- 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;

         Connection connection = null;

         Session session;

-        Destination destination;

+        Destination destination = null;

         MessageProducer messageProducer;

 

         try {

             connection = connectionFactory.createConnection();

             connection.start();

             session = connection.createSession(true, Session.AUTO_ACKNOWLEDGE);

-            destination = session.createTopic(AlarmConst.MQ_TOPIC_NAME_ALARMS_CORRELATION);

+            if (t instanceof CorrelationResult) {

+                destination = session.createTopic(AlarmConst.MQ_TOPIC_NAME_ALARMS_CORRELATION);

+            } else if (t instanceof Alarm) {

+                destination = session.createTopic(AlarmConst.MQ_TOPIC_NAME_ALARM);

+            }

             messageProducer = session.createProducer(destination);

             ObjectMessage message = session.createObjectMessage(msgEntity);

             messageProducer.send(message);