aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorLvbo163 <lv.bo163@zte.com.cn>2018-03-05 17:15:32 +0800
committerLvbo163 <lv.bo163@zte.com.cn>2018-03-05 17:15:32 +0800
commit79488acab14b9ab402bcb0079989a76164f813f6 (patch)
tree035f03938d9ccac0832b0ce35a7894eb2955edff
parentae6d8b70401361d05f62757349a226931e614fba (diff)
add ut for addPod of kube
Issue-ID: MSB-160 Change-Id: Iac5e9310933cbcff9edc8f829840c6219e5658b1 Signed-off-by: Lvbo163 <lv.bo163@zte.com.cn>
-rw-r--r--src/kube2msb/kube_work_test.go58
1 files changed, 58 insertions, 0 deletions
diff --git a/src/kube2msb/kube_work_test.go b/src/kube2msb/kube_work_test.go
index cc79439..a96f5c9 100644
--- a/src/kube2msb/kube_work_test.go
+++ b/src/kube2msb/kube_work_test.go
@@ -160,3 +160,61 @@ func TestUpdateServiceKube(t *testing.T) {
client.UpdateService(notExistService)
msbWorkValidate(t, msbWorkQueue, notExistService, MSBWorkAddService, "192.168.10.12")
}
+
+func createMockPod(name string, ip string) *kapi.Pod {
+ pod := kapi.Pod{
+ Status: kapi.PodStatus{
+ PodIP: ip,
+ },
+ }
+
+ pod.Name = name
+ pod.Annotations = map[string]string{serviceKey: name}
+
+ return &pod
+}
+
+func msbWorkPodValidate(t *testing.T, queue <-chan MSBWork, pod *kapi.Pod, action MSBWorkAction) {
+ work := <-queue
+
+ if work.Action != action || work.IPAddress != pod.Status.PodIP || work.ServiceInfo != pod.Name {
+ t.Errorf("expect %s,%s,%s to be %s %s,%s",
+ work.Action, work.IPAddress, work.ServiceInfo, action, pod.Status.PodIP, pod.Name)
+ }
+}
+
+func TestAddPodKube(t *testing.T) {
+ client := newClientBookKeeper()
+ msbWorkQueue := make(chan MSBWork, 10)
+ client.msbQueue = msbWorkQueue
+
+ // add ServiceTypeClusterIP
+ pod := createMockPod("addPodTest", "192.168.10.10")
+ client.AddPod(pod)
+ msbWorkPodValidate(t, msbWorkQueue, pod, MSBWorkAddPod)
+ if _, ok := client.pods[pod.Name]; !ok {
+ t.Errorf("add pod error, pod not exists in client.pods")
+ }
+
+ // exception process
+ // TODO servicekey is not set , cannot check result for there would be no return
+ podWithoutServiceKey := &kapi.Pod{}
+ client.AddPod(podWithoutServiceKey)
+
+ // TODO pod already exist , cannot check result for there would be no return
+ client.AddPod(pod)
+
+ // pod.Name == "" || pod.Status.PodIP == ""
+ podWithoutName := createMockPod("", "192.168.10.10")
+ client.AddPod(podWithoutName)
+ if addedPod, ok := addMap[podWithoutName.Name]; !ok || addedPod.Status.PodIP != podWithoutName.Status.PodIP {
+ t.Errorf("podWithoutName didnot add to addMap")
+ }
+
+ podWithoutIp := createMockPod("podWithoutIp", "")
+ client.AddPod(podWithoutIp)
+ if addedPod, ok := addMap[podWithoutIp.Name]; !ok || addedPod.Status.PodIP != podWithoutIp.Status.PodIP {
+ t.Errorf("podWithoutIp didnot add to addMap")
+ }
+
+}