jsoniter_map_test.go 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122
  1. package jsoniter
  2. import (
  3. "encoding/json"
  4. "github.com/json-iterator/go/require"
  5. "math/big"
  6. "testing"
  7. )
  8. func Test_read_map(t *testing.T) {
  9. should := require.New(t)
  10. iter := ParseString(ConfigDefault, `{"hello": "world"}`)
  11. m := map[string]string{"1": "2"}
  12. iter.ReadVal(&m)
  13. copy(iter.buf, []byte{0, 0, 0, 0, 0, 0})
  14. should.Equal(map[string]string{"1": "2", "hello": "world"}, m)
  15. }
  16. func Test_read_map_of_interface(t *testing.T) {
  17. should := require.New(t)
  18. iter := ParseString(ConfigDefault, `{"hello": "world"}`)
  19. m := map[string]interface{}{"1": "2"}
  20. iter.ReadVal(&m)
  21. should.Equal(map[string]interface{}{"1": "2", "hello": "world"}, m)
  22. iter = ParseString(ConfigDefault, `{"hello": "world"}`)
  23. should.Equal(map[string]interface{}{"hello": "world"}, iter.Read())
  24. }
  25. func Test_map_wrapper_any_get_all(t *testing.T) {
  26. should := require.New(t)
  27. any := Wrap(map[string][]int{"Field1": {1, 2}})
  28. should.Equal(`{"Field1":1}`, any.Get('*', 0).ToString())
  29. }
  30. func Test_write_val_map(t *testing.T) {
  31. should := require.New(t)
  32. val := map[string]string{"1": "2"}
  33. str, err := MarshalToString(val)
  34. should.Nil(err)
  35. should.Equal(`{"1":"2"}`, str)
  36. }
  37. func Test_slice_of_map(t *testing.T) {
  38. should := require.New(t)
  39. val := []map[string]string{{"1": "2"}}
  40. str, err := MarshalToString(val)
  41. should.Nil(err)
  42. should.Equal(`[{"1":"2"}]`, str)
  43. val = []map[string]string{}
  44. should.Nil(UnmarshalFromString(str, &val))
  45. should.Equal("2", val[0]["1"])
  46. }
  47. func Test_encode_int_key_map(t *testing.T) {
  48. should := require.New(t)
  49. val := map[int]string{1: "2"}
  50. str, err := MarshalToString(val)
  51. should.Nil(err)
  52. should.Equal(`{"1":"2"}`, str)
  53. }
  54. func Test_decode_int_key_map(t *testing.T) {
  55. should := require.New(t)
  56. var val map[int]string
  57. should.Nil(UnmarshalFromString(`{"1":"2"}`, &val))
  58. should.Equal(map[int]string{1: "2"}, val)
  59. }
  60. func Test_encode_TextMarshaler_key_map(t *testing.T) {
  61. should := require.New(t)
  62. f, _, _ := big.ParseFloat("1", 10, 64, big.ToZero)
  63. val := map[*big.Float]string{f: "2"}
  64. str, err := MarshalToString(val)
  65. should.Nil(err)
  66. should.Equal(`{"1":"2"}`, str)
  67. }
  68. func Test_decode_TextMarshaler_key_map(t *testing.T) {
  69. should := require.New(t)
  70. var val map[*big.Float]string
  71. should.Nil(UnmarshalFromString(`{"1":"2"}`, &val))
  72. str, err := MarshalToString(val)
  73. should.Nil(err)
  74. should.Equal(`{"1":"2"}`, str)
  75. }
  76. func Test_map_key_with_escaped_char(t *testing.T) {
  77. type Ttest struct {
  78. Map map[string]string
  79. }
  80. var jsonBytes = []byte(`
  81. {
  82. "Map":{
  83. "k\"ey": "val"
  84. }
  85. }`)
  86. should := require.New(t)
  87. {
  88. var obj Ttest
  89. should.Nil(json.Unmarshal(jsonBytes, &obj))
  90. should.Equal(map[string]string{"k\"ey": "val"}, obj.Map)
  91. }
  92. {
  93. var obj Ttest
  94. should.Nil(Unmarshal(jsonBytes, &obj))
  95. should.Equal(map[string]string{"k\"ey": "val"}, obj.Map)
  96. }
  97. }
  98. func Test_encode_map_with_sorted_keys(t *testing.T) {
  99. should := require.New(t)
  100. m := map[string]interface{}{
  101. "3": 3,
  102. "1": 1,
  103. "2": 2,
  104. }
  105. bytes, err := json.Marshal(m)
  106. should.Nil(err)
  107. output, err := ConfigCompatibleWithStandardLibrary.MarshalToString(m)
  108. should.Nil(err)
  109. should.Equal(string(bytes), output)
  110. }