From: Lvbo163 <lv.bo163@zte.com.cn>
Date: Wed, 7 Mar 2018 09:28:22 +0000 (+0800)
Subject: add ut for SendPodWork
X-Git-Tag: v1.1.0~3
X-Git-Url: https://gerrit.onap.org/r/gitweb?a=commitdiff_plain;h=refs%2Fchanges%2F77%2F34477%2F1;p=oom%2Fregistrator.git

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
--- 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)
+		}
+	}
+}