add ut for SendPodWork

Issue-ID: MSB-166

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

diff --git a/src/kube2msb/kube2msb_test.go b/src/kube2msb/kube2msb_test.go
index 09f7a80..eea0f92 100644 (file)
--- a/src/kube2msb/kube2msb_test.go
+++ b/src/kube2msb/kube2msb_test.go
@@ -89,3 +89,24 @@ func TestSendServiceWork(t *testing.T) {
                }
        }
 }
+
+func TestSendPodWork(t *testing.T) {
+
+       kubeWorkQueue := make(chan KubeWork, 1)
+       podObj := kapi.Pod{}
+
+       cases := []KubeWorkAction{
+               KubeWorkAddPod,
+               KubeWorkRemovePod,
+               KubeWorkUpdatePod,
+       }
+
+       for _, c := range cases {
+               sendPodWork(c, kubeWorkQueue, &podObj)
+               got := <-kubeWorkQueue
+
+               if got.Action != c {
+                       t.Errorf("sendPodWork(%action, queue, service) got %gotAction", c, got.Action)
+               }
+       }
+}