func TestPathWillAugmentExisting(t *testing.T) {
js, err := NewGson([]byte(`{"this":{"a":"aa","b":"bb","c":"cc"}}`))
assert.Equal(t, nil, err)
js.SetPath([]string{"this", "d"}, "dd")
cases := []struct {
path []string
outcome string
}{
{
path: []string{"this", "a"},
outcome: "aa",
},
{
path: []string{"this", "b"},
outcome: "bb",
},
{
path: []string{"this", "c"},
outcome: "cc",
},
{
path: []string{"this", "d"},
outcome: "dd",
},
}
for _, tc := range cases {
s, err := js.GetPath(tc.path...).String()
assert.Equal(t, nil, err)
assert.Equal(t, tc.outcome, s)
}
}
func TestSetPath(t *testing.T) {
js, err := NewGson([]byte(`{}`))
assert.Equal(t, nil, err)
js.SetPath([]string{"foo", "bar"}, "baz")
s, err := js.GetPath("foo", "bar").String()
assert.Equal(t, nil, err)
assert.Equal(t, "baz", s)
}
func TestSet(t *testing.T) {
js, err := NewGson([]byte(`{}`))
assert.Equal(t, nil, err)
js.Set("baz", "bing")
s, err := js.GetPath("baz").String()
assert.Equal(t, nil, err)
assert.Equal(t, "bing", s)
}
func TestPathWillOverwriteExisting(t *testing.T) {
// notice how "a" is 0.1 - but then we'll try to set at path a, foo
js, err := NewGson([]byte(`{"this":{"a":0.1,"b":"bb","c":"cc"}}`))
assert.Equal(t, nil, err)
js.SetPath([]string{"this", "a", "foo"}, "bar")
s, err := js.GetPath("this", "a", "foo").String()
assert.Equal(t, nil, err)
assert.Equal(t, "bar", s)
}
func TestReplace(t *testing.T) {
js, err := NewGson([]byte(`{}`))
assert.Equal(t, nil, err)
err = js.UnmarshalJSON([]byte(`{"baz":"bing"}`))
assert.Equal(t, nil, err)
s, err := js.GetPath("baz").String()
assert.Equal(t, nil, err)
assert.Equal(t, "bing", s)
}
func TestSetPathNoPath(t *testing.T) {
js, err := NewGson([]byte(`{"some":"data","some_number":1.0,"some_bool":false}`))
assert.Equal(t, nil, err)
f := js.GetPath("some_number").MustFloat64(99.0)
assert.Equal(t, f, 1.0)
js.SetPath([]string{}, map[string]interface{}{"foo": "bar"})
s, err := js.GetPath("foo").String()
assert.Equal(t, nil, err)
assert.Equal(t, "bar", s)
f = js.GetPath("some_number").MustFloat64(99.0)
assert.Equal(t, f, 99.0)
}
func TestHasMobile(t *testing.T) {
assert.Equal(t, true, HasMobile(mobile, true))
assert.Equal(t, true, HasMobile(mobile, false))
assert.Equal(t, true, HasMobile(mobileEase, false))
assert.Equal(t, false, HasMobile(mobileEase, true))
assert.Equal(t, true, HasMobile(mobileString, true))
assert.Equal(t, true, HasMobile(mobileString, false))
}
func TestNewFromReader(t *testing.T) {
//Use New Constructor
buf := bytes.NewBuffer([]byte(`{
"test": {
"array": [1, "2", 3],
"arraywithsubs": [
{"subkeyone": 1},
{"subkeytwo": 2, "subkeythree": 3}
],
"bignum": 9223372036854775807,
"uint64": 18446744073709551615
}
}`))
js, err := NewFromReader(buf)
//Standard Test Case
assert.NotEqual(t, nil, js)
assert.Equal(t, nil, err)
arr, _ := js.Get("test").Get("array").Array()
assert.NotEqual(t, nil, arr)
for i, v := range arr {
var iv int
switch v.(type) {
case json.Number:
i64, err := v.(json.Number).Int64()
assert.Equal(t, nil, err)
iv = int(i64)
case string:
iv, _ = strconv.Atoi(v.(string))
}
assert.Equal(t, i+1, iv)
}
ma := js.Get("test").Get("array").MustArray()
assert.Equal(t, ma, []interface{}{json.Number("1"), "2", json.Number("3")})
mm := js.Get("test").Get("arraywithsubs").GetIndex(0).MustMap()
assert.Equal(t, mm, map[string]interface{}{"subkeyone": json.Number("1")})
assert.Equal(t, js.Get("test").Get("bignum").MustInt64(), int64(9223372036854775807))
assert.Equal(t, js.Get("test").Get("uint64").MustUint64(), uint64(18446744073709551615))
}
func TestStdlibInterfaces(t *testing.T) {
val := new(struct {
Name string `json:"name"`
Params *Gson `json:"params"`
})
val2 := new(struct {
Name string `json:"name"`
Params *Gson `json:"params"`
})
raw := `{"name":"myobject","params":{"string":"simplejson"}}`
assert.Equal(t, nil, json.Unmarshal([]byte(raw), val))
assert.Equal(t, "myobject", val.Name)
assert.NotEqual(t, nil, val.Params.data)
s, _ := val.Params.Get("string").String()
assert.Equal(t, "simplejson", s)
p, err := json.Marshal(val)
assert.Equal(t, nil, err)
assert.Equal(t, nil, json.Unmarshal(p, val2))
assert.Equal(t, val, val2) // stable
}
func TestGetMobiles(t *testing.T) {
assert.Equal(t, 2, len(GetMobiles(mobileString, true)))
assert.Equal(t, 3, len(GetMobiles(mobileString, false)))
assert.Equal(t, 1, len(GetMobiles(mobile, true)))
assert.Equal(t, 1, len(GetMobiles(mobile, false)))
assert.Equal(t, 0, len(GetMobiles(mobileEase, true)))
assert.Equal(t, 1, len(GetMobiles(mobileEase, false)))
assert.Equal(t, "18307400009", GetMobiles(mobile, true)[0])
assert.Equal(t, "18307400009", GetMobiles(mobile, false)[0])
assert.Equal(t, "18307400009", GetMobiles(mobileString, true)[0])
assert.Equal(t, "18307400009", GetMobiles(mobileString, false)[0])
assert.Equal(t, "13973333488", GetMobiles(mobileString, true)[1])
assert.Equal(t, "13973333488", GetMobiles(mobileString, false)[1])
assert.Equal(t, "183****8188", GetMobiles(mobileString, false)[2])
}
func TestCountMobile(t *testing.T) {
assert.Equal(t, 1, CountMobile(mobile, true))
assert.Equal(t, 1, CountMobile(mobile, false))
assert.Equal(t, 1, CountMobile(mobileEase, false))
assert.Equal(t, 0, CountMobile(mobileEase, true))
}
func TestIsMobile(t *testing.T) {
assert.Equal(t, true, IsMobile(mobile, true))
assert.Equal(t, true, IsMobile(mobile, false))
assert.Equal(t, true, IsMobile(mobileEase, false))
assert.Equal(t, false, IsMobile(mobileEase, true))
}
func TestMobiler(t *testing.T) {
mobiler := Mobiler{}
err := mobiler.LoadData("./data.txt")
if err != nil {
assert.Panic(t, err, func() {})
}
mobile := mobiler.Mobile("18307400009")
assert.Equal(t, "怀化", mobile.City)
assert.Equal(t, "湖南", mobile.Province)
assert.Equal(t, "移动", mobile.ISP)
assert.Equal(t, "0745", mobile.AreaCode)
assert.Equal(t, "418000", mobile.ZipCode)
assert.Equal(t, "GSM/3G", mobile.Type)
assert.Equal(t, "18307400009", mobile.Number)
mobile = mobiler.Mobile("183****0009")
assert.Equal(t, "移动", mobile.ISP)
assert.Equal(t, "GSM/3G", mobile.Type)
assert.Equal(t, "183****0009", mobile.Number)
assert.Equal(t, "", mobile.City)
assert.Equal(t, "", mobile.Province)
assert.Equal(t, "", mobile.AreaCode)
assert.Equal(t, "", mobile.ZipCode)
}
func TestGsonson(t *testing.T) {
var ok bool
var err error
js, err := NewGson([]byte(`{
"test": {
"string_array": ["asdf", "ghjk", "zxcv"],
"string_array_null": ["abc", null, "efg"],
"array": [1, "2", 3],
"arraywithsubs": [{"subkeyone": 1},
{"subkeytwo": 2, "subkeythree": 3}],
"int": 10,
"float": 5.150,
"string": "simplejson",
"bool": true,
"sub_obj": {"a": 1}
}
}`))
assert.NotEqual(t, nil, js)
assert.Equal(t, nil, err)
_, ok = js.CheckGet("test")
assert.Equal(t, true, ok)
_, ok = js.CheckGet("missing_key")
assert.Equal(t, false, ok)
aws := js.Get("test").Get("arraywithsubs")
assert.NotEqual(t, nil, aws)
var awsval int
awsval, _ = aws.GetIndex(0).Get("subkeyone").Int()
assert.Equal(t, 1, awsval)
awsval, _ = aws.GetIndex(1).Get("subkeytwo").Int()
assert.Equal(t, 2, awsval)
awsval, _ = aws.GetIndex(1).Get("subkeythree").Int()
assert.Equal(t, 3, awsval)
i, _ := js.Get("test").Get("int").Int()
assert.Equal(t, 10, i)
f, _ := js.Get("test").Get("float").Float64()
assert.Equal(t, 5.150, f)
s, _ := js.Get("test").Get("string").String()
assert.Equal(t, "simplejson", s)
b, _ := js.Get("test").Get("bool").Bool()
assert.Equal(t, true, b)
mi := js.Get("test").Get("int").MustInt()
assert.Equal(t, 10, mi)
mi2 := js.Get("test").Get("missing_int").MustInt(5150)
assert.Equal(t, 5150, mi2)
ms := js.Get("test").Get("string").MustString()
assert.Equal(t, "simplejson", ms)
ms2 := js.Get("test").Get("missing_string").MustString("fyea")
assert.Equal(t, "fyea", ms2)
ma2 := js.Get("test").Get("missing_array").MustArray([]interface{}{"1", 2, "3"})
assert.Equal(t, ma2, []interface{}{"1", 2, "3"})
msa := js.Get("test").Get("string_array").MustStringArray()
assert.Equal(t, msa[0], "asdf")
assert.Equal(t, msa[1], "ghjk")
assert.Equal(t, msa[2], "zxcv")
msa2 := js.Get("test").Get("string_array").MustStringArray([]string{"1", "2", "3"})
assert.Equal(t, msa2[0], "asdf")
assert.Equal(t, msa2[1], "ghjk")
assert.Equal(t, msa2[2], "zxcv")
msa3 := js.Get("test").Get("missing_array").MustStringArray([]string{"1", "2", "3"})
assert.Equal(t, msa3, []string{"1", "2", "3"})
mm2 := js.Get("test").Get("missing_map").MustMap(map[string]interface{}{"found": false})
assert.Equal(t, mm2, map[string]interface{}{"found": false})
strs, err := js.Get("test").Get("string_array").StringArray()
assert.Equal(t, err, nil)
assert.Equal(t, strs[0], "asdf")
assert.Equal(t, strs[1], "ghjk")
assert.Equal(t, strs[2], "zxcv")
strs2, err := js.Get("test").Get("string_array_null").StringArray()
assert.Equal(t, err, nil)
assert.Equal(t, strs2[0], "abc")
assert.Equal(t, strs2[1], "")
assert.Equal(t, strs2[2], "efg")
gp, _ := js.GetPath("test", "string").String()
assert.Equal(t, "simplejson", gp)
gp2, _ := js.GetPath("test", "int").Int()
assert.Equal(t, 10, gp2)
assert.Equal(t, js.Get("test").Get("bool").MustBool(), true)
js.Set("float2", 300.0)
assert.Equal(t, js.Get("float2").MustFloat64(), 300.0)
js.Set("test2", "setTest")
assert.Equal(t, "setTest", js.Get("test2").MustString())
js.Del("test2")
assert.NotEqual(t, "setTest", js.Get("test2").MustString())
js.Get("test").Get("sub_obj").Set("a", 2)
assert.Equal(t, 2, js.Get("test").Get("sub_obj").Get("a").MustInt())
js.GetPath("test", "sub_obj").Set("a", 3)
assert.Equal(t, 3, js.GetPath("test", "sub_obj", "a").MustInt())
}