add ut for MergeIP

Issue-ID: MSB-177

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

1 files changed, 37 insertions, 0 deletions

diff --git a/src/kube2msb/msb_work_test.go b/src/kube2msb/msb_work_test.go
new file mode 100644
index 0000000..e271d22
--- /dev/null
+++ b/src/kube2msb/msb_work_test.go
@@ -0,0 +1,37 @@
+/*
+Copyright 2018 ZTE, Inc. and others.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+package main
+
+import (
+	"testing"
+)
+
+func TestMergeIP(t *testing.T) {
+	cases := []struct{ ip, sInfo, want string }{
+		{"127.0.0.1", "{}", "{\"ip\":\"127.0.0.1\",}"},
+		{"127.0.0.1", "[{}]", "[{\"ip\":\"127.0.0.1\",}]"},
+		{"127.0.0.1", "{\"name\":\"msb\"}", "{\"ip\":\"127.0.0.1\",\"name\":\"msb\"}"},
+		{"127.0.0.1", "{\"name\":\"msb\", \"child\":{\"name\":\"childname\"}}",
+			"{\"ip\":\"127.0.0.1\",\"name\":\"msb\", \"child\":{\"ip\":\"127.0.0.1\",\"name\":\"childname\"}}"},
+	}
+
+	for _, c := range cases {
+		got := mergeIP(c.ip, c.sInfo)
+		if got != c.want {
+			t.Errorf("mergeIP(%ip, %sInfo) == %got, want %want", c.ip, c.sInfo, got, c.want)
+		}
+	}
+}