add ut for SendServiceWork

Issue-ID: MSB-165

Change-Id: I61d11fdf1b4cc9da023b4a0aea23ae098231e289
Signed-off-by: Lvbo163 <lv.bo163@zte.com.cn>

1 files changed, 23 insertions, 0 deletions

diff --git a/src/kube2msb/kube2msb_test.go b/src/kube2msb/kube2msb_test.go
index 435fb3f..09f7a80 100644
--- a/src/kube2msb/kube2msb_test.go
+++ b/src/kube2msb/kube2msb_test.go
@@ -18,6 +18,8 @@ package main
 import (
 	"os"
 	"testing"
+
+	kapi "k8s.io/kubernetes/pkg/api"
 )
 
 func urlFormateValidate(t *testing.T, method string) {
@@ -66,3 +68,24 @@ func TestGetMSBUrl(t *testing.T) {
 func TestGetKubeMasterUrl(t *testing.T) {
 	urlFormateValidate(t, "getKubeMasterUrl")
 }
+
+func TestSendServiceWork(t *testing.T) {
+
+	kubeWorkQueue := make(chan KubeWork, 1)
+	serviceObj := kapi.Service{}
+
+	cases := []KubeWorkAction{
+		KubeWorkAddService,
+		KubeWorkRemoveService,
+		KubeWorkUpdateService,
+	}
+
+	for _, c := range cases {
+		sendServiceWork(c, kubeWorkQueue, &serviceObj)
+		got := <-kubeWorkQueue
+
+		if got.Action != c {
+			t.Errorf("sendServiceWork(%action, queue, service) got %gotAction", c, got.Action)
+		}
+	}
+}