gin/binding/json_test.go

// Copyright 2019 Gin Core Team. All rights reserved.
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.

package binding

import (
	"io"
	"net/http/httptest"
	"testing"
	"time"
	"unsafe"

	"github.com/gin-gonic/gin/codec/api"
	"github.com/gin-gonic/gin/codec/json"
	"github.com/gin-gonic/gin/render"
	jsoniter "github.com/json-iterator/go"
	"github.com/modern-go/reflect2"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestJSONBindingBindBody(t *testing.T) {
	var s struct {
		Foo string `json:"foo"`
	}
	err := jsonBinding{}.BindBody([]byte(`{"foo": "FOO"}`), &s)
	require.NoError(t, err)
	assert.Equal(t, "FOO", s.Foo)
}

func TestJSONBindingBindBodyMap(t *testing.T) {
	s := make(map[string]string)
	err := jsonBinding{}.BindBody([]byte(`{"foo": "FOO","hello":"world"}`), &s)
	require.NoError(t, err)
	assert.Len(t, s, 2)
	assert.Equal(t, "FOO", s["foo"])
	assert.Equal(t, "world", s["hello"])
}

func TestCustomJsonCodec(t *testing.T) {
	//Restore json encoding configuration after testing
	oldMarshal := json.Api
	defer func() {
		json.Api = oldMarshal
	}()
	//Custom json api
	json.Api = customJsonApi{}

	//test decode json
	obj := customReq{}
	err := jsonBinding{}.BindBody([]byte(`{"time_empty":null,"time_struct": "2001-12-05 10:01:02.345","time_nil":null,"time_pointer":"2002-12-05 10:01:02.345"}`), &obj)
	require.NoError(t, err)
	assert.Equal(t, zeroTime, obj.TimeEmpty)
	assert.Equal(t, time.Date(2001, 12, 05, 10, 01, 02, 345000000, time.Local), obj.TimeStruct)
	assert.Nil(t, obj.TimeNil)
	assert.Equal(t, time.Date(2002, 12, 05, 10, 01, 02, 345000000, time.Local), *obj.TimePointer)
	//test encode json
	w := httptest.NewRecorder()
	err2 := (render.PureJSON{Data: obj}).Render(w)
	require.NoError(t, err2)
	assert.Equal(t, "{\"time_empty\":null,\"time_struct\":\"2001-12-05 10:01:02.345\",\"time_nil\":null,\"time_pointer\":\"2002-12-05 10:01:02.345\"}\n", w.Body.String())
	assert.Equal(t, "application/json; charset=utf-8", w.Header().Get("Content-Type"))
}

type customReq struct {
	TimeEmpty   time.Time  `json:"time_empty"`
	TimeStruct  time.Time  `json:"time_struct"`
	TimeNil     *time.Time `json:"time_nil"`
	TimePointer *time.Time `json:"time_pointer"`
}

var customConfig = jsoniter.Config{
	EscapeHTML:             true,
	SortMapKeys:            true,
	ValidateJsonRawMessage: true,
}.Froze()

func init() {
	customConfig.RegisterExtension(&TimeEx{})
	customConfig.RegisterExtension(&TimePointerEx{})
}

type customJsonApi struct {
}

func (j customJsonApi) Marshal(v any) ([]byte, error) {
	return customConfig.Marshal(v)
}

func (j customJsonApi) Unmarshal(data []byte, v any) error {
	return customConfig.Unmarshal(data, v)
}

func (j customJsonApi) MarshalIndent(v any, prefix, indent string) ([]byte, error) {
	return customConfig.MarshalIndent(v, prefix, indent)
}

func (j customJsonApi) NewEncoder(writer io.Writer) api.JsonEncoder {
	return customConfig.NewEncoder(writer)
}

func (j customJsonApi) NewDecoder(reader io.Reader) api.JsonDecoder {
	return customConfig.NewDecoder(reader)
}

//region Time Extension

var (
	zeroTime         = time.Time{}
	timeType         = reflect2.TypeOfPtr((*time.Time)(nil)).Elem()
	defaultTimeCodec = &timeCodec{}
)

type TimeEx struct {
	jsoniter.DummyExtension
}

func (te *TimeEx) CreateDecoder(typ reflect2.Type) jsoniter.ValDecoder {
	if typ == timeType {
		return defaultTimeCodec
	}
	return nil
}

func (te *TimeEx) CreateEncoder(typ reflect2.Type) jsoniter.ValEncoder {
	if typ == timeType {
		return defaultTimeCodec
	}
	return nil
}

type timeCodec struct {
}

func (tc timeCodec) IsEmpty(ptr unsafe.Pointer) bool {
	t := *((*time.Time)(ptr))
	return t == zeroTime
}

func (tc timeCodec) Encode(ptr unsafe.Pointer, stream *jsoniter.Stream) {
	t := *((*time.Time)(ptr))
	if t == zeroTime {
		stream.WriteNil()
		return
	}
	stream.WriteString(t.In(time.Local).Format("2006-01-02 15:04:05.000"))
}

func (tc timeCodec) Decode(ptr unsafe.Pointer, iter *jsoniter.Iterator) {
	ts := iter.ReadString()
	if len(ts) == 0 {
		*((*time.Time)(ptr)) = zeroTime
		return
	}
	t, err := time.ParseInLocation("2006-01-02 15:04:05.000", ts, time.Local)
	if err != nil {
		panic(err)
	}
	*((*time.Time)(ptr)) = t
}

//endregion

//region *Time Extension

var (
	timePointerType         = reflect2.TypeOfPtr((**time.Time)(nil)).Elem()
	defaultTimePointerCodec = &timePointerCodec{}
)

type TimePointerEx struct {
	jsoniter.DummyExtension
}

func (tpe *TimePointerEx) CreateDecoder(typ reflect2.Type) jsoniter.ValDecoder {
	if typ == timePointerType {
		return defaultTimePointerCodec
	}
	return nil
}

func (tpe *TimePointerEx) CreateEncoder(typ reflect2.Type) jsoniter.ValEncoder {
	if typ == timePointerType {
		return defaultTimePointerCodec
	}
	return nil
}

type timePointerCodec struct {
}

func (tpc timePointerCodec) IsEmpty(ptr unsafe.Pointer) bool {
	t := *((**time.Time)(ptr))
	return t == nil || *t == zeroTime
}

func (tpc timePointerCodec) Encode(ptr unsafe.Pointer, stream *jsoniter.Stream) {
	t := *((**time.Time)(ptr))
	if t == nil || *t == zeroTime {
		stream.WriteNil()
		return
	}
	stream.WriteString(t.In(time.Local).Format("2006-01-02 15:04:05.000"))
}

func (tpc timePointerCodec) Decode(ptr unsafe.Pointer, iter *jsoniter.Iterator) {
	ts := iter.ReadString()
	if len(ts) == 0 {
		*((**time.Time)(ptr)) = nil
		return
	}
	t, err := time.ParseInLocation("2006-01-02 15:04:05.000", ts, time.Local)
	if err != nil {
		panic(err)
	}
	*((**time.Time)(ptr)) = &t
}

//endregion