8cb390f8fe
* Add YAML binding for application/x-yaml. * Add YAML binding methods to Context. * Review fixes. * Revert accidentally removed import.
73 lines
1.5 KiB
Go
73 lines
1.5 KiB
Go
package binding
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin/testdata/protoexample"
|
|
"github.com/golang/protobuf/proto"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/ugorji/go/codec"
|
|
)
|
|
|
|
func TestBindingBody(t *testing.T) {
|
|
for _, tt := range []struct {
|
|
name string
|
|
binding BindingBody
|
|
body string
|
|
want string
|
|
}{
|
|
{
|
|
name: "JSON binding",
|
|
binding: JSON,
|
|
body: `{"foo":"FOO"}`,
|
|
},
|
|
{
|
|
name: "XML binding",
|
|
binding: XML,
|
|
body: `<?xml version="1.0" encoding="UTF-8"?>
|
|
<root>
|
|
<foo>FOO</foo>
|
|
</root>`,
|
|
},
|
|
{
|
|
name: "MsgPack binding",
|
|
binding: MsgPack,
|
|
body: msgPackBody(t),
|
|
},
|
|
{
|
|
name: "YAML binding",
|
|
binding: YAML,
|
|
body: `foo: FOO`,
|
|
},
|
|
} {
|
|
t.Logf("testing: %s", tt.name)
|
|
req := requestWithBody("POST", "/", tt.body)
|
|
form := FooStruct{}
|
|
body, _ := ioutil.ReadAll(req.Body)
|
|
assert.NoError(t, tt.binding.BindBody(body, &form))
|
|
assert.Equal(t, FooStruct{"FOO"}, form)
|
|
}
|
|
}
|
|
|
|
func msgPackBody(t *testing.T) string {
|
|
test := FooStruct{"FOO"}
|
|
h := new(codec.MsgpackHandle)
|
|
buf := bytes.NewBuffer(nil)
|
|
assert.NoError(t, codec.NewEncoder(buf, h).Encode(test))
|
|
return buf.String()
|
|
}
|
|
|
|
func TestBindingBodyProto(t *testing.T) {
|
|
test := protoexample.Test{
|
|
Label: proto.String("FOO"),
|
|
}
|
|
data, _ := proto.Marshal(&test)
|
|
req := requestWithBody("POST", "/", string(data))
|
|
form := protoexample.Test{}
|
|
body, _ := ioutil.ReadAll(req.Body)
|
|
assert.NoError(t, ProtoBuf.BindBody(body, &form))
|
|
assert.Equal(t, test, form)
|
|
}
|