remove duplicate code

This commit is contained in:
thinkerou 2017-11-01 09:54:27 +08:00
parent 6653d5d588
commit 58b02903a7

View File

@ -13,6 +13,8 @@ import (
"github.com/stretchr/testify/assert"
"github.com/ugorji/go/codec"
"github.com/gin-gonic/gin"
)
// TODO unit tests
@ -91,35 +93,9 @@ func TestRenderSecureJSON(t *testing.T) {
assert.Equal(t, "application/json; charset=utf-8", w2.Header().Get("Content-Type"))
}
type xmlmap map[string]interface{}
// Allows type H to be used with xml.Marshal
func (h xmlmap) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
start.Name = xml.Name{
Space: "",
Local: "map",
}
if err := e.EncodeToken(start); err != nil {
return err
}
for key, value := range h {
elem := xml.StartElement{
Name: xml.Name{Space: "", Local: key},
Attr: []xml.Attr{},
}
if err := e.EncodeElement(value, elem); err != nil {
return err
}
}
if err := e.EncodeToken(xml.EndElement{Name: start.Name}); err != nil {
return err
}
return nil
}
func TestRenderXML(t *testing.T) {
w := httptest.NewRecorder()
data := xmlmap{
data := gin.H{
"foo": "bar",
}