| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738 |
- // 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.
- package gin
- import (
- "bytes"
- "errors"
- "html/template"
- "mime/multipart"
- "net/http"
- "net/http/httptest"
- "strings"
- "testing"
- "time"
- "github.com/manucorporat/sse"
- "github.com/stretchr/testify/assert"
- )
- // Unit tests TODO
- // func (c *Context) File(filepath string) {
- // func (c *Context) Negotiate(code int, config Negotiate) {
- // BAD case: func (c *Context) Render(code int, render render.Render, obj ...interface{}) {
- // test that information is not leaked when reusing Contexts (using the Pool)
- func createMultipartRequest() *http.Request {
- boundary := "--testboundary"
- body := new(bytes.Buffer)
- mw := multipart.NewWriter(body)
- defer mw.Close()
- must(mw.SetBoundary(boundary))
- must(mw.WriteField("foo", "bar"))
- must(mw.WriteField("bar", "10"))
- must(mw.WriteField("bar", "foo2"))
- must(mw.WriteField("array", "first"))
- must(mw.WriteField("array", "second"))
- must(mw.WriteField("id", ""))
- req, err := http.NewRequest("POST", "/", body)
- must(err)
- req.Header.Set("Content-Type", MIMEMultipartPOSTForm+"; boundary="+boundary)
- return req
- }
- func must(err error) {
- if err != nil {
- panic(err.Error())
- }
- }
- func TestContextReset(t *testing.T) {
- router := New()
- c := router.allocateContext()
- assert.Equal(t, c.engine, router)
- c.index = 2
- c.Writer = &responseWriter{ResponseWriter: httptest.NewRecorder()}
- c.Params = Params{Param{}}
- c.Error(errors.New("test"))
- c.Set("foo", "bar")
- c.reset()
- assert.False(t, c.IsAborted())
- assert.Nil(t, c.Keys)
- assert.Nil(t, c.Accepted)
- assert.Len(t, c.Errors, 0)
- assert.Empty(t, c.Errors.Errors())
- assert.Empty(t, c.Errors.ByType(ErrorTypeAny))
- assert.Len(t, c.Params, 0)
- assert.EqualValues(t, c.index, -1)
- assert.Equal(t, c.Writer.(*responseWriter), &c.writermem)
- }
- func TestContextHandlers(t *testing.T) {
- c, _, _ := CreateTestContext()
- assert.Nil(t, c.handlers)
- assert.Nil(t, c.handlers.Last())
- c.handlers = HandlersChain{}
- assert.NotNil(t, c.handlers)
- assert.Nil(t, c.handlers.Last())
- f := func(c *Context) {}
- g := func(c *Context) {}
- c.handlers = HandlersChain{f}
- compareFunc(t, f, c.handlers.Last())
- c.handlers = HandlersChain{f, g}
- compareFunc(t, g, c.handlers.Last())
- }
- // TestContextSetGet tests that a parameter is set correctly on the
- // current context and can be retrieved using Get.
- func TestContextSetGet(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.Set("foo", "bar")
- value, err := c.Get("foo")
- assert.Equal(t, value, "bar")
- assert.True(t, err)
- value, err = c.Get("foo2")
- assert.Nil(t, value)
- assert.False(t, err)
- assert.Equal(t, c.MustGet("foo"), "bar")
- assert.Panics(t, func() { c.MustGet("no_exist") })
- }
- func TestContextSetGetValues(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.Set("string", "this is a string")
- c.Set("int32", int32(-42))
- c.Set("int64", int64(42424242424242))
- c.Set("uint64", uint64(42))
- c.Set("float32", float32(4.2))
- c.Set("float64", 4.2)
- var a interface{} = 1
- c.Set("intInterface", a)
- assert.Exactly(t, c.MustGet("string").(string), "this is a string")
- assert.Exactly(t, c.MustGet("int32").(int32), int32(-42))
- assert.Exactly(t, c.MustGet("int64").(int64), int64(42424242424242))
- assert.Exactly(t, c.MustGet("uint64").(uint64), uint64(42))
- assert.Exactly(t, c.MustGet("float32").(float32), float32(4.2))
- assert.Exactly(t, c.MustGet("float64").(float64), 4.2)
- assert.Exactly(t, c.MustGet("intInterface").(int), 1)
- }
- func TestContextCopy(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.index = 2
- c.Request, _ = http.NewRequest("POST", "/hola", nil)
- c.handlers = HandlersChain{func(c *Context) {}}
- c.Params = Params{Param{Key: "foo", Value: "bar"}}
- c.Set("foo", "bar")
- cp := c.Copy()
- assert.Nil(t, cp.handlers)
- assert.Nil(t, cp.writermem.ResponseWriter)
- assert.Equal(t, &cp.writermem, cp.Writer.(*responseWriter))
- assert.Equal(t, cp.Request, c.Request)
- assert.Equal(t, cp.index, abortIndex)
- assert.Equal(t, cp.Keys, c.Keys)
- assert.Equal(t, cp.engine, c.engine)
- assert.Equal(t, cp.Params, c.Params)
- }
- func TestContextHandlerName(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.handlers = HandlersChain{func(c *Context) {}, handlerNameTest}
- assert.Regexp(t, "^(.*/vendor/)?github.com/gin-gonic/gin.handlerNameTest$", c.HandlerName())
- }
- func handlerNameTest(c *Context) {
- }
- func TestContextQuery(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.Request, _ = http.NewRequest("GET", "http://example.com/?foo=bar&page=10&id=", nil)
- value, ok := c.GetQuery("foo")
- assert.True(t, ok)
- assert.Equal(t, value, "bar")
- assert.Equal(t, c.DefaultQuery("foo", "none"), "bar")
- assert.Equal(t, c.Query("foo"), "bar")
- value, ok = c.GetQuery("page")
- assert.True(t, ok)
- assert.Equal(t, value, "10")
- assert.Equal(t, c.DefaultQuery("page", "0"), "10")
- assert.Equal(t, c.Query("page"), "10")
- value, ok = c.GetQuery("id")
- assert.True(t, ok)
- assert.Empty(t, value)
- assert.Equal(t, c.DefaultQuery("id", "nada"), "")
- assert.Empty(t, c.Query("id"))
- value, ok = c.GetQuery("NoKey")
- assert.False(t, ok)
- assert.Empty(t, value)
- assert.Equal(t, c.DefaultQuery("NoKey", "nada"), "nada")
- assert.Empty(t, c.Query("NoKey"))
- // postform should not mess
- value, ok = c.GetPostForm("page")
- assert.False(t, ok)
- assert.Empty(t, value)
- assert.Empty(t, c.PostForm("foo"))
- }
- func TestContextQueryAndPostForm(t *testing.T) {
- c, _, _ := CreateTestContext()
- body := bytes.NewBufferString("foo=bar&page=11&both=&foo=second")
- c.Request, _ = http.NewRequest("POST", "/?both=GET&id=main&id=omit&array[]=first&array[]=second", body)
- c.Request.Header.Add("Content-Type", MIMEPOSTForm)
- assert.Equal(t, c.DefaultPostForm("foo", "none"), "bar")
- assert.Equal(t, c.PostForm("foo"), "bar")
- assert.Empty(t, c.Query("foo"))
- value, ok := c.GetPostForm("page")
- assert.True(t, ok)
- assert.Equal(t, value, "11")
- assert.Equal(t, c.DefaultPostForm("page", "0"), "11")
- assert.Equal(t, c.PostForm("page"), "11")
- assert.Equal(t, c.Query("page"), "")
- value, ok = c.GetPostForm("both")
- assert.True(t, ok)
- assert.Empty(t, value)
- assert.Empty(t, c.PostForm("both"))
- assert.Equal(t, c.DefaultPostForm("both", "nothing"), "")
- assert.Equal(t, c.Query("both"), "GET")
- value, ok = c.GetQuery("id")
- assert.True(t, ok)
- assert.Equal(t, value, "main")
- assert.Equal(t, c.DefaultPostForm("id", "000"), "000")
- assert.Equal(t, c.Query("id"), "main")
- assert.Empty(t, c.PostForm("id"))
- value, ok = c.GetQuery("NoKey")
- assert.False(t, ok)
- assert.Empty(t, value)
- value, ok = c.GetPostForm("NoKey")
- assert.False(t, ok)
- assert.Empty(t, value)
- assert.Equal(t, c.DefaultPostForm("NoKey", "nada"), "nada")
- assert.Equal(t, c.DefaultQuery("NoKey", "nothing"), "nothing")
- assert.Empty(t, c.PostForm("NoKey"))
- assert.Empty(t, c.Query("NoKey"))
- var obj struct {
- Foo string `form:"foo"`
- ID string `form:"id"`
- Page int `form:"page"`
- Both string `form:"both"`
- Array []string `form:"array[]"`
- }
- assert.NoError(t, c.Bind(&obj))
- assert.Equal(t, obj.Foo, "bar")
- assert.Equal(t, obj.ID, "main")
- assert.Equal(t, obj.Page, 11)
- assert.Equal(t, obj.Both, "")
- assert.Equal(t, obj.Array, []string{"first", "second"})
- values, ok := c.GetQueryArray("array[]")
- assert.True(t, ok)
- assert.Equal(t, "first", values[0])
- assert.Equal(t, "second", values[1])
- values = c.QueryArray("array[]")
- assert.Equal(t, "first", values[0])
- assert.Equal(t, "second", values[1])
- values = c.QueryArray("nokey")
- assert.Equal(t, 0, len(values))
- values = c.QueryArray("both")
- assert.Equal(t, 1, len(values))
- assert.Equal(t, "GET", values[0])
- }
- func TestContextPostFormMultipart(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.Request = createMultipartRequest()
- var obj struct {
- Foo string `form:"foo"`
- Bar string `form:"bar"`
- BarAsInt int `form:"bar"`
- Array []string `form:"array"`
- ID string `form:"id"`
- }
- assert.NoError(t, c.Bind(&obj))
- assert.Equal(t, obj.Foo, "bar")
- assert.Equal(t, obj.Bar, "10")
- assert.Equal(t, obj.BarAsInt, 10)
- assert.Equal(t, obj.Array, []string{"first", "second"})
- assert.Equal(t, obj.ID, "")
- value, ok := c.GetQuery("foo")
- assert.False(t, ok)
- assert.Empty(t, value)
- assert.Empty(t, c.Query("bar"))
- assert.Equal(t, c.DefaultQuery("id", "nothing"), "nothing")
- value, ok = c.GetPostForm("foo")
- assert.True(t, ok)
- assert.Equal(t, value, "bar")
- assert.Equal(t, c.PostForm("foo"), "bar")
- value, ok = c.GetPostForm("array")
- assert.True(t, ok)
- assert.Equal(t, value, "first")
- assert.Equal(t, c.PostForm("array"), "first")
- assert.Equal(t, c.DefaultPostForm("bar", "nothing"), "10")
- value, ok = c.GetPostForm("id")
- assert.True(t, ok)
- assert.Empty(t, value)
- assert.Empty(t, c.PostForm("id"))
- assert.Empty(t, c.DefaultPostForm("id", "nothing"))
- value, ok = c.GetPostForm("nokey")
- assert.False(t, ok)
- assert.Empty(t, value)
- assert.Equal(t, c.DefaultPostForm("nokey", "nothing"), "nothing")
- values, ok := c.GetPostFormArray("array")
- assert.True(t, ok)
- assert.Equal(t, "first", values[0])
- assert.Equal(t, "second", values[1])
- values = c.PostFormArray("array")
- assert.Equal(t, "first", values[0])
- assert.Equal(t, "second", values[1])
- values = c.PostFormArray("nokey")
- assert.Equal(t, 0, len(values))
- values = c.PostFormArray("foo")
- assert.Equal(t, 1, len(values))
- assert.Equal(t, "bar", values[0])
- }
- func TestContextSetCookie(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.SetCookie("user", "gin", 1, "/", "localhost", true, true)
- assert.Equal(t, c.Writer.Header().Get("Set-Cookie"), "user=gin; Path=/; Domain=localhost; Max-Age=1; HttpOnly; Secure")
- }
- func TestContextGetCookie(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.Request, _ = http.NewRequest("GET", "/get", nil)
- c.Request.Header.Set("Cookie", "user=gin")
- cookie, _ := c.Cookie("user")
- assert.Equal(t, cookie, "gin")
- }
- // Tests that the response is serialized as JSON
- // and Content-Type is set to application/json
- func TestContextRenderJSON(t *testing.T) {
- c, w, _ := CreateTestContext()
- c.JSON(201, H{"foo": "bar"})
- assert.Equal(t, w.Code, 201)
- assert.Equal(t, w.Body.String(), "{\"foo\":\"bar\"}\n")
- assert.Equal(t, w.HeaderMap.Get("Content-Type"), "application/json; charset=utf-8")
- }
- // Tests that the response is serialized as JSON
- // we change the content-type before
- func TestContextRenderAPIJSON(t *testing.T) {
- c, w, _ := CreateTestContext()
- c.Header("Content-Type", "application/vnd.api+json")
- c.JSON(201, H{"foo": "bar"})
- assert.Equal(t, w.Code, 201)
- assert.Equal(t, w.Body.String(), "{\"foo\":\"bar\"}\n")
- assert.Equal(t, w.HeaderMap.Get("Content-Type"), "application/vnd.api+json")
- }
- // Tests that the response is serialized as JSON
- // and Content-Type is set to application/json
- func TestContextRenderIndentedJSON(t *testing.T) {
- c, w, _ := CreateTestContext()
- c.IndentedJSON(201, H{"foo": "bar", "bar": "foo", "nested": H{"foo": "bar"}})
- assert.Equal(t, w.Code, 201)
- assert.Equal(t, w.Body.String(), "{\n \"bar\": \"foo\",\n \"foo\": \"bar\",\n \"nested\": {\n \"foo\": \"bar\"\n }\n}")
- assert.Equal(t, w.HeaderMap.Get("Content-Type"), "application/json; charset=utf-8")
- }
- // Tests that the response executes the templates
- // and responds with Content-Type set to text/html
- func TestContextRenderHTML(t *testing.T) {
- c, w, router := CreateTestContext()
- templ := template.Must(template.New("t").Parse(`Hello {{.name}}`))
- router.SetHTMLTemplate(templ)
- c.HTML(201, "t", H{"name": "alexandernyquist"})
- assert.Equal(t, w.Code, 201)
- assert.Equal(t, w.Body.String(), "Hello alexandernyquist")
- assert.Equal(t, w.HeaderMap.Get("Content-Type"), "text/html; charset=utf-8")
- }
- // TestContextXML tests that the response is serialized as XML
- // and Content-Type is set to application/xml
- func TestContextRenderXML(t *testing.T) {
- c, w, _ := CreateTestContext()
- c.XML(201, H{"foo": "bar"})
- assert.Equal(t, w.Code, 201)
- assert.Equal(t, w.Body.String(), "<map><foo>bar</foo></map>")
- assert.Equal(t, w.HeaderMap.Get("Content-Type"), "application/xml; charset=utf-8")
- }
- // TestContextString tests that the response is returned
- // with Content-Type set to text/plain
- func TestContextRenderString(t *testing.T) {
- c, w, _ := CreateTestContext()
- c.String(201, "test %s %d", "string", 2)
- assert.Equal(t, w.Code, 201)
- assert.Equal(t, w.Body.String(), "test string 2")
- assert.Equal(t, w.HeaderMap.Get("Content-Type"), "text/plain; charset=utf-8")
- }
- // TestContextString tests that the response is returned
- // with Content-Type set to text/html
- func TestContextRenderHTMLString(t *testing.T) {
- c, w, _ := CreateTestContext()
- c.Header("Content-Type", "text/html; charset=utf-8")
- c.String(201, "<html>%s %d</html>", "string", 3)
- assert.Equal(t, w.Code, 201)
- assert.Equal(t, w.Body.String(), "<html>string 3</html>")
- assert.Equal(t, w.HeaderMap.Get("Content-Type"), "text/html; charset=utf-8")
- }
- // TestContextData tests that the response can be written from `bytesting`
- // with specified MIME type
- func TestContextRenderData(t *testing.T) {
- c, w, _ := CreateTestContext()
- c.Data(201, "text/csv", []byte(`foo,bar`))
- assert.Equal(t, w.Code, 201)
- assert.Equal(t, w.Body.String(), "foo,bar")
- assert.Equal(t, w.HeaderMap.Get("Content-Type"), "text/csv")
- }
- func TestContextRenderSSE(t *testing.T) {
- c, w, _ := CreateTestContext()
- c.SSEvent("float", 1.5)
- c.Render(-1, sse.Event{
- Id: "123",
- Data: "text",
- })
- c.SSEvent("chat", H{
- "foo": "bar",
- "bar": "foo",
- })
- assert.Equal(t, strings.Replace(w.Body.String(), " ", "", -1), strings.Replace("event:float\ndata:1.5\n\nid:123\ndata:text\n\nevent:chat\ndata:{\"bar\":\"foo\",\"foo\":\"bar\"}\n\n", " ", "", -1))
- }
- func TestContextRenderFile(t *testing.T) {
- c, w, _ := CreateTestContext()
- c.Request, _ = http.NewRequest("GET", "/", nil)
- c.File("./gin.go")
- assert.Equal(t, w.Code, 200)
- assert.Contains(t, w.Body.String(), "func New() *Engine {")
- assert.Equal(t, w.HeaderMap.Get("Content-Type"), "text/plain; charset=utf-8")
- }
- // TestContextRenderYAML tests that the response is serialized as YAML
- // and Content-Type is set to application/x-yaml
- func TestContextRenderYAML(t *testing.T) {
- c, w, _ := CreateTestContext()
- c.YAML(201, H{"foo": "bar"})
- assert.Equal(t, w.Code, 201)
- assert.Equal(t, w.Body.String(), "foo: bar\n")
- assert.Equal(t, w.HeaderMap.Get("Content-Type"), "application/x-yaml; charset=utf-8")
- }
- func TestContextHeaders(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.Header("Content-Type", "text/plain")
- c.Header("X-Custom", "value")
- assert.Equal(t, c.Writer.Header().Get("Content-Type"), "text/plain")
- assert.Equal(t, c.Writer.Header().Get("X-Custom"), "value")
- c.Header("Content-Type", "text/html")
- c.Header("X-Custom", "")
- assert.Equal(t, c.Writer.Header().Get("Content-Type"), "text/html")
- _, exist := c.Writer.Header()["X-Custom"]
- assert.False(t, exist)
- }
- // TODO
- func TestContextRenderRedirectWithRelativePath(t *testing.T) {
- c, w, _ := CreateTestContext()
- c.Request, _ = http.NewRequest("POST", "http://example.com", nil)
- assert.Panics(t, func() { c.Redirect(299, "/new_path") })
- assert.Panics(t, func() { c.Redirect(309, "/new_path") })
- c.Redirect(301, "/path")
- c.Writer.WriteHeaderNow()
- assert.Equal(t, w.Code, 301)
- assert.Equal(t, w.Header().Get("Location"), "/path")
- }
- func TestContextRenderRedirectWithAbsolutePath(t *testing.T) {
- c, w, _ := CreateTestContext()
- c.Request, _ = http.NewRequest("POST", "http://example.com", nil)
- c.Redirect(302, "http://google.com")
- c.Writer.WriteHeaderNow()
- assert.Equal(t, w.Code, 302)
- assert.Equal(t, w.Header().Get("Location"), "http://google.com")
- }
- func TestContextRenderRedirectWith201(t *testing.T) {
- c, w, _ := CreateTestContext()
- c.Request, _ = http.NewRequest("POST", "http://example.com", nil)
- c.Redirect(201, "/resource")
- c.Writer.WriteHeaderNow()
- assert.Equal(t, w.Code, 201)
- assert.Equal(t, w.Header().Get("Location"), "/resource")
- }
- func TestContextRenderRedirectAll(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.Request, _ = http.NewRequest("POST", "http://example.com", nil)
- assert.Panics(t, func() { c.Redirect(200, "/resource") })
- assert.Panics(t, func() { c.Redirect(202, "/resource") })
- assert.Panics(t, func() { c.Redirect(299, "/resource") })
- assert.Panics(t, func() { c.Redirect(309, "/resource") })
- assert.NotPanics(t, func() { c.Redirect(300, "/resource") })
- assert.NotPanics(t, func() { c.Redirect(308, "/resource") })
- }
- func TestContextNegotiationFormat(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.Request, _ = http.NewRequest("POST", "", nil)
- assert.Panics(t, func() { c.NegotiateFormat() })
- assert.Equal(t, c.NegotiateFormat(MIMEJSON, MIMEXML), MIMEJSON)
- assert.Equal(t, c.NegotiateFormat(MIMEHTML, MIMEJSON), MIMEHTML)
- }
- func TestContextNegotiationFormatWithAccept(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.Request, _ = http.NewRequest("POST", "/", nil)
- c.Request.Header.Add("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8")
- assert.Equal(t, c.NegotiateFormat(MIMEJSON, MIMEXML), MIMEXML)
- assert.Equal(t, c.NegotiateFormat(MIMEXML, MIMEHTML), MIMEHTML)
- assert.Equal(t, c.NegotiateFormat(MIMEJSON), "")
- }
- func TestContextNegotiationFormatCustum(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.Request, _ = http.NewRequest("POST", "/", nil)
- c.Request.Header.Add("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8")
- c.Accepted = nil
- c.SetAccepted(MIMEJSON, MIMEXML)
- assert.Equal(t, c.NegotiateFormat(MIMEJSON, MIMEXML), MIMEJSON)
- assert.Equal(t, c.NegotiateFormat(MIMEXML, MIMEHTML), MIMEXML)
- assert.Equal(t, c.NegotiateFormat(MIMEJSON), MIMEJSON)
- }
- func TestContextIsAborted(t *testing.T) {
- c, _, _ := CreateTestContext()
- assert.False(t, c.IsAborted())
- c.Abort()
- assert.True(t, c.IsAborted())
- c.Next()
- assert.True(t, c.IsAborted())
- c.index++
- assert.True(t, c.IsAborted())
- }
- // TestContextData tests that the response can be written from `bytesting`
- // with specified MIME type
- func TestContextAbortWithStatus(t *testing.T) {
- c, w, _ := CreateTestContext()
- c.index = 4
- c.AbortWithStatus(401)
- assert.Equal(t, c.index, abortIndex)
- assert.Equal(t, c.Writer.Status(), 401)
- assert.Equal(t, w.Code, 401)
- assert.True(t, c.IsAborted())
- }
- func TestContextError(t *testing.T) {
- c, _, _ := CreateTestContext()
- assert.Empty(t, c.Errors)
- c.Error(errors.New("first error"))
- assert.Len(t, c.Errors, 1)
- assert.Equal(t, c.Errors.String(), "Error #01: first error\n")
- c.Error(&Error{
- Err: errors.New("second error"),
- Meta: "some data 2",
- Type: ErrorTypePublic,
- })
- assert.Len(t, c.Errors, 2)
- assert.Equal(t, c.Errors[0].Err, errors.New("first error"))
- assert.Nil(t, c.Errors[0].Meta)
- assert.Equal(t, c.Errors[0].Type, ErrorTypePrivate)
- assert.Equal(t, c.Errors[1].Err, errors.New("second error"))
- assert.Equal(t, c.Errors[1].Meta, "some data 2")
- assert.Equal(t, c.Errors[1].Type, ErrorTypePublic)
- assert.Equal(t, c.Errors.Last(), c.Errors[1])
- }
- func TestContextTypedError(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.Error(errors.New("externo 0")).SetType(ErrorTypePublic)
- c.Error(errors.New("interno 0")).SetType(ErrorTypePrivate)
- for _, err := range c.Errors.ByType(ErrorTypePublic) {
- assert.Equal(t, err.Type, ErrorTypePublic)
- }
- for _, err := range c.Errors.ByType(ErrorTypePrivate) {
- assert.Equal(t, err.Type, ErrorTypePrivate)
- }
- assert.Equal(t, c.Errors.Errors(), []string{"externo 0", "interno 0"})
- }
- func TestContextAbortWithError(t *testing.T) {
- c, w, _ := CreateTestContext()
- c.AbortWithError(401, errors.New("bad input")).SetMeta("some input")
- assert.Equal(t, w.Code, 401)
- assert.Equal(t, c.index, abortIndex)
- assert.True(t, c.IsAborted())
- }
- func TestContextClientIP(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.Request, _ = http.NewRequest("POST", "/", nil)
- c.Request.Header.Set("X-Real-IP", " 10.10.10.10 ")
- c.Request.Header.Set("X-Forwarded-For", " 20.20.20.20, 30.30.30.30")
- c.Request.RemoteAddr = " 40.40.40.40:42123 "
- assert.Equal(t, c.ClientIP(), "10.10.10.10")
- c.Request.Header.Del("X-Real-IP")
- assert.Equal(t, c.ClientIP(), "20.20.20.20")
- c.Request.Header.Set("X-Forwarded-For", "30.30.30.30 ")
- assert.Equal(t, c.ClientIP(), "30.30.30.30")
- c.Request.Header.Del("X-Forwarded-For")
- assert.Equal(t, c.ClientIP(), "40.40.40.40")
- }
- func TestContextContentType(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.Request, _ = http.NewRequest("POST", "/", nil)
- c.Request.Header.Set("Content-Type", "application/json; charset=utf-8")
- assert.Equal(t, c.ContentType(), "application/json")
- }
- func TestContextAutoBindJSON(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.Request, _ = http.NewRequest("POST", "/", bytes.NewBufferString("{\"foo\":\"bar\", \"bar\":\"foo\"}"))
- c.Request.Header.Add("Content-Type", MIMEJSON)
- var obj struct {
- Foo string `json:"foo"`
- Bar string `json:"bar"`
- }
- assert.NoError(t, c.Bind(&obj))
- assert.Equal(t, obj.Bar, "foo")
- assert.Equal(t, obj.Foo, "bar")
- assert.Empty(t, c.Errors)
- }
- func TestContextBindWithJSON(t *testing.T) {
- c, w, _ := CreateTestContext()
- c.Request, _ = http.NewRequest("POST", "/", bytes.NewBufferString("{\"foo\":\"bar\", \"bar\":\"foo\"}"))
- c.Request.Header.Add("Content-Type", MIMEXML) // set fake content-type
- var obj struct {
- Foo string `json:"foo"`
- Bar string `json:"bar"`
- }
- assert.NoError(t, c.BindJSON(&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) {
- c, w, _ := CreateTestContext()
- c.Request, _ = http.NewRequest("POST", "http://example.com", bytes.NewBufferString("\"foo\":\"bar\", \"bar\":\"foo\"}"))
- c.Request.Header.Add("Content-Type", MIMEJSON)
- var obj struct {
- Foo string `json:"foo"`
- Bar string `json:"bar"`
- }
- assert.False(t, c.IsAborted())
- assert.Error(t, c.Bind(&obj))
- c.Writer.WriteHeaderNow()
- assert.Empty(t, obj.Bar)
- assert.Empty(t, obj.Foo)
- assert.Equal(t, w.Code, 400)
- assert.True(t, c.IsAborted())
- }
- func TestContextGolangContext(t *testing.T) {
- c, _, _ := CreateTestContext()
- c.Request, _ = http.NewRequest("POST", "/", bytes.NewBufferString("{\"foo\":\"bar\", \"bar\":\"foo\"}"))
- assert.NoError(t, c.Err())
- assert.Nil(t, c.Done())
- ti, ok := c.Deadline()
- assert.Equal(t, ti, time.Time{})
- assert.False(t, ok)
- assert.Equal(t, c.Value(0), c.Request)
- assert.Nil(t, c.Value("foo"))
- c.Set("foo", "bar")
- assert.Equal(t, c.Value("foo"), "bar")
- assert.Nil(t, c.Value(1))
- }
|