aboutsummaryrefslogtreecommitdiffstats
path: root/src/kube2msb/kube_work_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'src/kube2msb/kube_work_test.go')
-rw-r--r--src/kube2msb/kube_work_test.go29
1 files changed, 29 insertions, 0 deletions
diff --git a/src/kube2msb/kube_work_test.go b/src/kube2msb/kube_work_test.go
index a96f5c9..f67608a 100644
--- 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