mirror of
https://github.com/gin-gonic/gin.git
synced 2026-06-21 08:08:15 +08:00
93e3206653
Decode the rendered msgpack output and assert values instead of comparing raw bytes (which can vary with map iteration order). Enable MsgpackHandle.RawToString so msgpack strings decode as Go strings.
81 lines
1.9 KiB
Go
81 lines
1.9 KiB
Go
// Copyright 2014 Manu Martinez-Almeida. All rights reserved.
|
|
// Use of this source code is governed by a MIT style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
//go:build !nomsgpack
|
|
|
|
package render
|
|
|
|
import (
|
|
"errors"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/ugorji/go/codec"
|
|
)
|
|
|
|
func TestRenderMsgPack(t *testing.T) {
|
|
w := httptest.NewRecorder()
|
|
data := map[string]any{
|
|
"foo": "bar",
|
|
}
|
|
|
|
(MsgPack{data}).WriteContentType(w)
|
|
assert.Equal(t, "application/msgpack; charset=utf-8", w.Header().Get("Content-Type"))
|
|
|
|
err := (MsgPack{data}).Render(w)
|
|
|
|
require.NoError(t, err)
|
|
|
|
var decoded map[string]any
|
|
var mh codec.MsgpackHandle
|
|
mh.RawToString = true
|
|
err = codec.NewDecoderBytes(w.Body.Bytes(), &mh).Decode(&decoded)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, data, decoded)
|
|
assert.Equal(t, "application/msgpack; charset=utf-8", w.Header().Get("Content-Type"))
|
|
}
|
|
|
|
func TestWriteMsgPack(t *testing.T) {
|
|
w := httptest.NewRecorder()
|
|
data := map[string]any{
|
|
"foo": "bar",
|
|
"num": 42,
|
|
}
|
|
|
|
err := WriteMsgPack(w, data)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, "application/msgpack; charset=utf-8", w.Header().Get("Content-Type"))
|
|
|
|
var decoded map[string]any
|
|
var mh codec.MsgpackHandle
|
|
mh.RawToString = true
|
|
err = codec.NewDecoderBytes(w.Body.Bytes(), &mh).Decode(&decoded)
|
|
require.NoError(t, err)
|
|
assert.Len(t, decoded, 2)
|
|
assert.Equal(t, "bar", decoded["foo"])
|
|
assert.EqualValues(t, 42, decoded["num"])
|
|
}
|
|
|
|
type failWriter struct {
|
|
*httptest.ResponseRecorder
|
|
}
|
|
|
|
func (w *failWriter) Write(data []byte) (int, error) {
|
|
return 0, errors.New("write error")
|
|
}
|
|
|
|
func TestRenderMsgPackError(t *testing.T) {
|
|
w := httptest.NewRecorder()
|
|
data := map[string]any{
|
|
"foo": "bar",
|
|
}
|
|
|
|
err := (MsgPack{data}).Render(&failWriter{w})
|
|
require.Error(t, err)
|
|
assert.Contains(t, err.Error(), "write error")
|
|
}
|