Merge branch 'multipart-form-data-fix' of https://github.com/konjoot/gin

- the merge was manually modified before committing.
This commit is contained in:
Manu Mtz-Almeida 2015-05-26 16:47:10 +02:00
commit ec1ce34d32
3 changed files with 71 additions and 3 deletions

View File

@ -42,7 +42,7 @@ func Default(method, contentType string) Binding {
return JSON return JSON
case MIMEXML, MIMEXML2: case MIMEXML, MIMEXML2:
return XML return XML
default: default: //case MIMEPOSTForm, MIMEMultipartPOSTForm:
return Form return Form
} }
} }

View File

@ -32,7 +32,10 @@ func TestBindingDefault(t *testing.T) {
assert.Equal(t, Default("PUT", MIMEXML2), XML) assert.Equal(t, Default("PUT", MIMEXML2), XML)
assert.Equal(t, Default("POST", MIMEPOSTForm), Form) assert.Equal(t, Default("POST", MIMEPOSTForm), Form)
assert.Equal(t, Default("DELETE", MIMEPOSTForm), Form) assert.Equal(t, Default("PUT", MIMEPOSTForm), Form)
assert.Equal(t, Default("POST", MIMEMultipartPOSTForm), Form)
assert.Equal(t, Default("PUT", MIMEMultipartPOSTForm), Form)
} }
func TestBindingJSON(t *testing.T) { func TestBindingJSON(t *testing.T) {
@ -63,7 +66,7 @@ func TestBindingXML(t *testing.T) {
func testFormBinding(t *testing.T, method, path, badPath, body, badBody string) { func testFormBinding(t *testing.T, method, path, badPath, body, badBody string) {
b := Form b := Form
assert.Equal(t, b.Name(), "query") assert.Equal(t, b.Name(), "form")
obj := FooBarStruct{} obj := FooBarStruct{}
req := requestWithBody(method, path, body) req := requestWithBody(method, path, body)

View File

@ -8,6 +8,7 @@ import (
"bytes" "bytes"
"errors" "errors"
"html/template" "html/template"
"mime/multipart"
"net/http" "net/http"
"net/http/httptest" "net/http/httptest"
"testing" "testing"
@ -33,6 +34,26 @@ func createTestContext() (c *Context, w *httptest.ResponseRecorder, r *Engine) {
return return
} }
func createMultipartForm() (body *bytes.Buffer, header string, err error) {
boundary := "--testboundary"
header = MIMEMultipartPOSTForm + "; boundary=" + boundary
body = &bytes.Buffer{}
mw := multipart.NewWriter(body)
defer mw.Close()
if err = mw.SetBoundary(boundary); err != nil {
return
}
if err = mw.WriteField("foo", "bar"); err != nil {
return
}
if err = mw.WriteField("bar", "foo"); err != nil {
return
}
return
}
func TestContextReset(t *testing.T) { func TestContextReset(t *testing.T) {
router := New() router := New()
c := router.allocateContext() c := router.allocateContext()
@ -444,6 +465,28 @@ func TestContextAutoBind(t *testing.T) {
assert.Equal(t, w.Body.Len(), 0) assert.Equal(t, w.Body.Len(), 0)
} }
func TestContextMultipartPostFormAutoBind(t *testing.T) {
c, w, _ := createTestContext()
var obj struct {
Foo string `form:"foo"`
Bar string `form:"bar"`
}
body, header, err := createMultipartForm()
if err != nil {
t.Error(err)
}
c.Request, _ = http.NewRequest("POST", "/", body)
c.Request.Header.Add("Content-Type", header)
assert.NoError(t, c.Bind(&obj))
assert.Equal(t, obj.Bar, "foo")
assert.Equal(t, obj.Foo, "bar")
assert.Equal(t, w.Body.Len(), 0)
}
func TestContextBadAutoBind(t *testing.T) { func TestContextBadAutoBind(t *testing.T) {
c, w, _ := createTestContext() c, w, _ := createTestContext()
c.Request, _ = http.NewRequest("POST", "http://example.com", bytes.NewBufferString("\"foo\":\"bar\", \"bar\":\"foo\"}")) c.Request, _ = http.NewRequest("POST", "http://example.com", bytes.NewBufferString("\"foo\":\"bar\", \"bar\":\"foo\"}"))
@ -477,6 +520,28 @@ func TestContextBindWith(t *testing.T) {
assert.Equal(t, w.Body.Len(), 0) assert.Equal(t, w.Body.Len(), 0)
} }
func TestContextMultipartBindWith(t *testing.T) {
c, w, _ := createTestContext()
var obj struct {
Foo string `form:"foo"`
Bar string `form:"bar"`
}
body, header, err := createMultipartForm()
if err != nil {
t.Error(err)
}
c.Request, _ = http.NewRequest("POST", "/", body)
c.Request.Header.Add("Content-Type", header)
assert.NoError(t, c.BindWith(&obj, binding.Form))
assert.Equal(t, obj.Bar, "foo")
assert.Equal(t, obj.Foo, "bar")
assert.Equal(t, w.Body.Len(), 0)
}
func TestContextGolangContext(t *testing.T) { func TestContextGolangContext(t *testing.T) {
c, _, _ := createTestContext() c, _, _ := createTestContext()
c.Request, _ = http.NewRequest("POST", "/", bytes.NewBufferString("{\"foo\":\"bar\", \"bar\":\"foo\"}")) c.Request, _ = http.NewRequest("POST", "/", bytes.NewBufferString("{\"foo\":\"bar\", \"bar\":\"foo\"}"))