123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139 |
- // 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"
- "encoding/xml"
- "fmt"
- "net/http"
- "testing"
- "github.com/stretchr/testify/assert"
- )
- func init() {
- SetMode(TestMode)
- }
- type testStruct struct {
- T *testing.T
- }
- func (t *testStruct) ServeHTTP(w http.ResponseWriter, req *http.Request) {
- assert.Equal(t.T, "POST", req.Method)
- assert.Equal(t.T, "/path", req.URL.Path)
- w.WriteHeader(http.StatusInternalServerError)
- fmt.Fprint(w, "hello")
- }
- func TestWrap(t *testing.T) {
- router := New()
- router.POST("/path", WrapH(&testStruct{t}))
- router.GET("/path2", WrapF(func(w http.ResponseWriter, req *http.Request) {
- assert.Equal(t, "GET", req.Method)
- assert.Equal(t, "/path2", req.URL.Path)
- w.WriteHeader(http.StatusBadRequest)
- fmt.Fprint(w, "hola!")
- }))
- w := performRequest(router, "POST", "/path")
- assert.Equal(t, http.StatusInternalServerError, w.Code)
- assert.Equal(t, "hello", w.Body.String())
- w = performRequest(router, "GET", "/path2")
- assert.Equal(t, http.StatusBadRequest, w.Code)
- assert.Equal(t, "hola!", w.Body.String())
- }
- func TestLastChar(t *testing.T) {
- assert.Equal(t, uint8('a'), lastChar("hola"))
- assert.Equal(t, uint8('s'), lastChar("adios"))
- assert.Panics(t, func() { lastChar("") })
- }
- func TestParseAccept(t *testing.T) {
- parts := parseAccept("text/html , application/xhtml+xml,application/xml;q=0.9, */* ;q=0.8")
- assert.Len(t, parts, 4)
- assert.Equal(t, "text/html", parts[0])
- assert.Equal(t, "application/xhtml+xml", parts[1])
- assert.Equal(t, "application/xml", parts[2])
- assert.Equal(t, "*/*", parts[3])
- }
- func TestChooseData(t *testing.T) {
- A := "a"
- B := "b"
- assert.Equal(t, A, chooseData(A, B))
- assert.Equal(t, B, chooseData(nil, B))
- assert.Panics(t, func() { chooseData(nil, nil) })
- }
- func TestFilterFlags(t *testing.T) {
- result := filterFlags("text/html ")
- assert.Equal(t, "text/html", result)
- result = filterFlags("text/html;")
- assert.Equal(t, "text/html", result)
- }
- func TestFunctionName(t *testing.T) {
- assert.Regexp(t, `^(.*/vendor/)?github.com/gin-gonic/gin.somefunction$`, nameOfFunction(somefunction))
- }
- func somefunction() {
- // this empty function is used by TestFunctionName()
- }
- func TestJoinPaths(t *testing.T) {
- assert.Equal(t, "", joinPaths("", ""))
- assert.Equal(t, "/", joinPaths("", "/"))
- assert.Equal(t, "/a", joinPaths("/a", ""))
- assert.Equal(t, "/a/", joinPaths("/a/", ""))
- assert.Equal(t, "/a/", joinPaths("/a/", "/"))
- assert.Equal(t, "/a/", joinPaths("/a", "/"))
- assert.Equal(t, "/a/hola", joinPaths("/a", "/hola"))
- assert.Equal(t, "/a/hola", joinPaths("/a/", "/hola"))
- assert.Equal(t, "/a/hola/", joinPaths("/a/", "/hola/"))
- assert.Equal(t, "/a/hola/", joinPaths("/a/", "/hola//"))
- }
- type bindTestStruct struct {
- Foo string `form:"foo" binding:"required"`
- Bar int `form:"bar" binding:"min=4"`
- }
- func TestBindMiddleware(t *testing.T) {
- var value *bindTestStruct
- var called bool
- router := New()
- router.GET("/", Bind(bindTestStruct{}), func(c *Context) {
- called = true
- value = c.MustGet(BindKey).(*bindTestStruct)
- })
- performRequest(router, "GET", "/?foo=hola&bar=10")
- assert.True(t, called)
- assert.Equal(t, "hola", value.Foo)
- assert.Equal(t, 10, value.Bar)
- called = false
- performRequest(router, "GET", "/?foo=hola&bar=1")
- assert.False(t, called)
- assert.Panics(t, func() {
- Bind(&bindTestStruct{})
- })
- }
- func TestMarshalXMLforH(t *testing.T) {
- h := H{
- "": "test",
- }
- var b bytes.Buffer
- enc := xml.NewEncoder(&b)
- var x xml.StartElement
- e := h.MarshalXML(enc, x)
- assert.Error(t, e)
- }
|