jsonmap/lodash_test.go at master · datasweet/jsonmap · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
package jsonmap_test

import (
	"fmt"
	"testing"

	"github.com/datasweet/jsonmap"
	"github.com/stretchr/testify/assert"
)

func TestFilter(t *testing.T) {
	values := jsonmap.Filter(jsonmap.FromString("[1, 2, 3, 4, 5]").Values(), func(v *jsonmap.Json) bool {
		return v.AsInt()%2 == 1
	})
	assert.Equal(t, 3, len(values))
	for i, v := range values {
		assert.Equal(t, int64(2*i+1), v.AsInt())
	}

	values = jsonmap.Filter(jsonmap.FromString(`[{ "value": 1 }, { "value": 2 }, { "value": 3 }, { "value": 4 }, { "value": 5 }]`).Values(), func(v *jsonmap.Json) bool {
		return v.Get("value").AsInt()%2 == 1
	})
	assert.Equal(t, 3, len(values))
	for i, v := range values {
		assert.JSONEq(t, fmt.Sprintf(`{ "value": %d }`, 2*i+1), v.Stringify())
	}
}

func TestMap(t *testing.T) {
	values := jsonmap.Map(jsonmap.FromString("[1, 2, 3, 4, 5]").Values(), func(v *jsonmap.Json) *jsonmap.Json {
		if v.AsInt()%2 == 1 {
			return v
		}
		return jsonmap.Nil()
	})
	assert.Equal(t, 5, len(values))
	for i, v := range values {
		if i%2 == 0 {
			assert.Equal(t, int64(i+1), v.AsInt())
		} else {
			assert.True(t, v.IsNil())
		}
	}
}

func TestAssign(t *testing.T) {
	json1 := jsonmap.FromString(`{
		"string": "hello",
		"bool": true,
		"number": 123
	}`)

	json2 := jsonmap.FromString(`{
		"array": [1,2,3,4,5]
	}`)

	json3 := jsonmap.FromString(`{
		"object": {
			"test": "world",
			"sub": [
				{"a": 4, "1": "a" },
				{"a": 5, "1": "b" }
			]
		}
	}`)

	j := jsonmap.Assign(jsonmap.FromString(`{"hello": "world" }`), json1, json2, json3)

	expected := jsonmap.FromString(`{
		"hello": "world",
		"string": "hello",
		"bool": true,
		"number": 123,
		"array": [1,2,3,4,5],
		"object": {
			"test": "world",
			"sub": [
				{"a": 4, "1": "a" },
				{"a": 5, "1": "b" }
			]
		}
	}`)

	assert.JSONEq(t, expected.Stringify(), j.Stringify())
}