1 files changed, 53 insertions, 0 deletions
diff --git a/src/kube2msb/msb_work_test.go b/src/kube2msb/msb_work_test.go
index e271d22..9f5a823 100644
--- a/src/kube2msb/msb_work_test.go
+++ b/src/kube2msb/msb_work_test.go
@@ -16,9 +16,62 @@ limitations under the License.
 package main
 
 import (
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"net/http/httptest"
 	"testing"
 )
 
+func TestAddServiceMsb(t *testing.T) {
+	handler := func(res http.ResponseWriter, req *http.Request) {
+		if req.Method != "POST" {
+			t.Errorf("Register() request method should be 'Post' not %s", req.Method)
+		} else if urlPrefix != req.URL.String() {
+			t.Errorf("Register() url should be %s, not %s", urlPrefix, req.URL)
+		} else {
+			body, err := ioutil.ReadAll(req.Body)
+			if err != nil {
+				t.Errorf("Register() fail to read request body")
+			}
+			var su = ServiceUnit{}
+			parseError := json.Unmarshal([]byte(body), &su)
+			if parseError != nil {
+				t.Errorf("Register() request body can not parse to ServiceUnit, %s", body)
+				return
+			} else {
+				res.WriteHeader(200)
+				res.Header().Set("Content-Type", "application/xml")
+				fmt.Fprintln(res, "regist success")
+			}
+		}
+
+	}
+	server := httptest.NewServer(http.HandlerFunc(handler))
+	defer server.Close()
+
+	client := MSBAgentWorker{
+		agent: &MSBAgent{
+			url: server.URL,
+		},
+	}
+
+	serviceInfo := `[{
+		"port":"8080",
+		"serviceName":"resgisterTest",
+		"version":"v1",
+		"url":"/register/test",
+		"protocol":"http",
+		"lb_policy":"random",
+		"visualRange":"1",
+		"path":"rt",
+		"enable_ssl":true
+	}]`
+
+	client.AddService("192.168.1.10", serviceInfo)
+}
+
 func TestMergeIP(t *testing.T) {
 	cases := []struct{ ip, sInfo, want string }{
 		{"127.0.0.1", "{}", "{\"ip\":\"127.0.0.1\",}"},