add ut for SendServiceWork

Issue-ID: MSB-165

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

diff --git a/src/kube2msb/kube2msb_test.go b/src/kube2msb/kube2msb_test.go
index 435fb3f..09f7a80 100644 (file)
--- 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)
+               }
+       }
+}