jsoniter_test.go

package jsoniterserialize

import (
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"

	"github.com/gin-gonic/gin"
	jsoniter "github.com/json-iterator/go"
	"gopkg.in/go-playground/assert.v1"
)

type Form struct {
	Name string `json:"name"`
}

var jsoniterAPI = jsoniter.Config{}.Froze()

func jsoniterTest(c *gin.Context) {
	var form Form
	if err := c.ShouldBindWith(&form, JsoniterBinding{jsoniterAPI}); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"message": "bad request"})
	}
	c.Render(http.StatusOK, JsoniterRender{jsoniterAPI, map[string]string{"message": "hello " + form.Name}})
}

func TestSerialize(t *testing.T) {
	router := gin.Default()
	router.POST("/test", jsoniterTest)
	req := httptest.NewRequest("POST", "/test", strings.NewReader(`{"name":"world"}`))
	req.Header.Add("Content-Type", "application/json")
	w := httptest.NewRecorder()
	router.ServeHTTP(w, req)
	assert.Equal(t, w.Code, http.StatusOK)
	assert.Equal(t, jsoniterAPI.Get(w.Body.Bytes(), "message").ToString(), "hello world")
}