From 708ce4cfd811fc72588d414e6c0720d3ee632a23 Mon Sep 17 00:00:00 2001
From: Lvbo163 <lv.bo163@zte.com.cn>
Date: Wed, 7 Mar 2018 17:28:22 +0800
Subject: add ut for SendPodWork

Issue-ID: MSB-166

Change-Id: I53dce8cf99cbd017672c8778201bb36316c09135
Signed-off-by: Lvbo163 <lv.bo163@zte.com.cn>
---
 src/kube2msb/kube2msb_test.go | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/src/kube2msb/kube2msb_test.go b/src/kube2msb/kube2msb_test.go
index 09f7a80..eea0f92 100644
--- 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)
+		}
+	}
+}
-- 
cgit 1.2.3-korg