add ut for addPod of msb

Issue-ID: MSB-174

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

1 files changed, 14 insertions, 2 deletions

diff --git a/src/kube2msb/msb_work_test.go b/src/kube2msb/msb_work_test.go
index 9f5a823..b5ba681 100644
--- a/src/kube2msb/msb_work_test.go
+++ b/src/kube2msb/msb_work_test.go
@@ -24,7 +24,7 @@ import (
 	"testing"
 )
 
-func TestAddServiceMsb(t *testing.T) {
+func addServiceOrPodTest(t *testing.T, addType string) {
 	handler := func(res http.ResponseWriter, req *http.Request) {
 		if req.Method != "POST" {
 			t.Errorf("Register() request method should be 'Post' not %s", req.Method)
@@ -69,7 +69,19 @@ func TestAddServiceMsb(t *testing.T) {
 		"enable_ssl":true
 	}]`
 
-	client.AddService("192.168.1.10", serviceInfo)
+	if addType == "Service" {
+		client.AddService("192.168.1.10", serviceInfo)
+	} else {
+		client.AddPod("192.168.1.10", serviceInfo)
+	}
+}
+
+func TestAddServiceMsb(t *testing.T) {
+	addServiceOrPodTest(t, "Service")
+}
+
+func TestAddPodMsb(t *testing.T) {
+	addServiceOrPodTest(t, "Pod")
 }
 
 func TestMergeIP(t *testing.T) {