add ut for remove pod of kube

Issue-ID: MSB-161

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

diff --git a/src/kube2msb/kube_work_test.go b/src/kube2msb/kube_work_test.go
index a96f5c9..f67608a 100644 (file)
--- a/src/kube2msb/kube_work_test.go
+++ b/src/kube2msb/kube_work_test.go
@@ -218,3 +218,32 @@ func TestAddPodKube(t *testing.T) {
        }
 
 }
+
+
+func TestRemovePodKube(t *testing.T) {
+       client := newClientBookKeeper()
+       msbWorkQueue := make(chan MSBWork, 10)
+       client.msbQueue = msbWorkQueue
+
+       // exception process
+       // TODO ServiceKey not set , cannot check result for there would be no return
+       podWithoutServiceKey := kapi.Pod{}
+       client.RemovePod(&podWithoutServiceKey)
+
+       // TODO Pod not exist , cannot check result for there would be no return
+       podNotExist := createMockPod("podNotExistTest", "192.168.10.10")
+       client.RemovePod(podNotExist)
+
+       // normal process
+       pod := createMockPod("removePodTest", "192.168.10.10")
+       client.AddPod(pod)
+       if _, ok := client.pods[pod.Name]; !ok {
+               t.Errorf("add pod error, pod not exists in client.pods")
+       }
+       msbWorkPodValidate(t, msbWorkQueue, pod, MSBWorkAddPod)
+       client.RemovePod(pod)
+       msbWorkPodValidate(t, msbWorkQueue, pod, MSBWorkRemovePod)
+       if _, ok := client.pods[pod.Name]; ok {
+               t.Errorf("remove pod error, pod still exists in client.pods")
+       }
+}
\ No newline at end of file