Add unit test

Issue-ID: VFC-836
Change-Id: Ie893da7ce8c6c85459644b4e5251460e6439b420
Signed-off-by: yangyan <yangyanyj@chinamobile.com>

diff --git a/ems/boco/data/aaa.txt b/ems/boco/data/aaa.txt
new file mode 100644 (file)
index 0000000..e69de29

diff --git a/ems/boco/data/bbb.txt b/ems/boco/data/bbb.txt
index e69de29..458aab0 100644 (file)
--- a/ems/boco/data/bbb.txt
+++ b/ems/boco/data/bbb.txt
@@ -0,0 +1 @@
+test abc
\ No newline at end of file

diff --git a/ems/boco/src/main/java/org/onap/vfc/nfvo/emsdriver/collector/CollectMsgReceiverThread.java b/ems/boco/src/main/java/org/onap/vfc/nfvo/emsdriver/collector/CollectMsgReceiverThread.java
index d3ced48..89cf34b 100644 (file)
--- a/ems/boco/src/main/java/org/onap/vfc/nfvo/emsdriver/collector/CollectMsgReceiverThread.java
+++ b/ems/boco/src/main/java/org/onap/vfc/nfvo/emsdriver/collector/CollectMsgReceiverThread.java
@@ -35,7 +35,6 @@ public class CollectMsgReceiverThread extends DriverThread {
     @Override
     public void dispose() {
         collectChannel = MessageChannelFactory.getMessageChannel(Constant.COLLECT_CHANNEL_KEY);
-
         taskService = TaskThreadService.getInstance(threadMaxNum);
         taskService.start();
 

diff --git a/ems/boco/src/test/java/org/onap/vfc/nfvo/emsdriver/collector/alarm/MessageUtilTest.java b/ems/boco/src/test/java/org/onap/vfc/nfvo/emsdriver/collector/alarm/MessageUtilTest.java
index 2f3d637..171c041 100644 (file)
--- a/ems/boco/src/test/java/org/onap/vfc/nfvo/emsdriver/collector/alarm/MessageUtilTest.java
+++ b/ems/boco/src/test/java/org/onap/vfc/nfvo/emsdriver/collector/alarm/MessageUtilTest.java
@@ -15,10 +15,17 @@
  */
 package org.onap.vfc.nfvo.emsdriver.collector.alarm;
 
-import org.junit.Test;
-
 import static org.junit.Assert.assertNotNull;
 
+import java.io.BufferedInputStream;
+import java.io.BufferedOutputStream;
+import java.io.FileInputStream;
+import java.io.FileNotFoundException;
+import java.io.FileOutputStream;
+import java.io.IOException;
+
+import org.junit.Test;
+
 public class MessageUtilTest {
 
     @Test
@@ -62,6 +69,52 @@ public class MessageUtilTest {
 
         assertNotNull(msg);
     }
+    
+    @Test
+    public void putloginMsg() {
+       String user = "test";
+       String passwd = "test";
+       Msg msg = MessageUtil.putLoginMsg(user, passwd);
+       assertNotNull(msg);     
+    }
 
-
+    @Test
+    public void putHeartBeatMsg() {
+       Msg msg = MessageUtil.putHeartBeatMsg(1);
+       assertNotNull(msg);     
+    }
+    
+    @Test
+    public void readOneMsg(){
+       try {
+       String fileName = System.getProperty("user.dir") + "/data/" + "bbb.txt";
+       FileInputStream fis = new FileInputStream(fileName);
+        BufferedInputStream bis=new BufferedInputStream(fis);
+       Msg msg = MessageUtil.readOneMsg(bis);
+       assertNotNull(msg);             
+               } catch (FileNotFoundException e) {
+                       // TODO Auto-generated catch block
+                  System.out.println("File is not found");
+       } catch (IOException e){
+          System.out.println("There is IOException");
+       }
+    }
+    
+    
+    
+    @Test
+    public void writeMsg(){
+        try {
+       String msgBody = "Heartbeat";
+       Msg msg = new Msg(msgBody,MsgType.ACK_HEARTBEAT);
+       String fileName = System.getProperty("user.dir") + "/data/" + "aaa.txt";
+       FileOutputStream fos=new FileOutputStream(fileName);
+        BufferedOutputStream bos=new BufferedOutputStream(fos);
+        MessageUtil.writeMsg(msg, bos);
+               } catch (IOException e) {
+                       // TODO Auto-generated catch block
+                       System.out.println("There is IOException");
+               }
+        
+    }
 }