go support AssertAsMap

aliyun · Feb 11, 2020 · 40c994f · 40c994f
1 parent f367afc
commit 40c994f
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/golang/service/service.go b/golang/service/service.go
@@ -107,6 +107,16 @@ func ToBytes(a string) []byte {
 	return []byte(a)
 }
 
+func AssertAsMap(a interface{}) map[string]interface{} {
+	byt, _ := json.Marshal(a)
+	res := make(map[string]interface{})
+	err := json.Unmarshal(byt, &res)
+	if err == nil {
+		return res
+	}
+	return res
+}
+
 func ParseJSON(a string) interface{} {
 	tmp := make(map[string]interface{})
 	err := json.Unmarshal([]byte(a), &tmp)

diff --git a/golang/service/service_test.go b/golang/service/service_test.go
@@ -138,3 +138,11 @@ func Test_StringifyMapValue(t *testing.T) {
 	utils.AssertEqual(t, "10", out["num"])
 	utils.AssertEqual(t, `{"test":"ok"}`, out["json"])
 }
+
+func Test_AssertAsMap(t *testing.T) {
+	in := map[string]string{
+		"num": "10",
+	}
+	out := AssertAsMap(in)
+	utils.AssertEqual(t, "10", out["num"].(string))
+}