127 lines
3.3 KiB
Go
127 lines
3.3 KiB
Go
// 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 (
|
|
"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, req.Method, "POST")
|
|
assert.Equal(t.T, req.URL.Path, "/path")
|
|
w.WriteHeader(500)
|
|
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, req.Method, "GET")
|
|
assert.Equal(t, req.URL.Path, "/path2")
|
|
w.WriteHeader(400)
|
|
fmt.Fprint(w, "hola!")
|
|
}))
|
|
|
|
w := performRequest(router, "POST", "/path")
|
|
assert.Equal(t, w.Code, 500)
|
|
assert.Equal(t, w.Body.String(), "hello")
|
|
|
|
w = performRequest(router, "GET", "/path2")
|
|
assert.Equal(t, w.Code, 400)
|
|
assert.Equal(t, w.Body.String(), "hola!")
|
|
}
|
|
|
|
func TestLastChar(t *testing.T) {
|
|
assert.Equal(t, lastChar("hola"), uint8('a'))
|
|
assert.Equal(t, lastChar("adios"), uint8('s'))
|
|
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, parts[0], "text/html")
|
|
assert.Equal(t, parts[1], "application/xhtml+xml")
|
|
assert.Equal(t, parts[2], "application/xml")
|
|
assert.Equal(t, parts[3], "*/*")
|
|
}
|
|
|
|
func TestChooseData(t *testing.T) {
|
|
A := "a"
|
|
B := "b"
|
|
assert.Equal(t, chooseData(A, B), A)
|
|
assert.Equal(t, chooseData(nil, B), B)
|
|
assert.Panics(t, func() { chooseData(nil, nil) })
|
|
}
|
|
|
|
func TestFilterFlags(t *testing.T) {
|
|
result := filterFlags("text/html ")
|
|
assert.Equal(t, result, "text/html")
|
|
|
|
result = filterFlags("text/html;")
|
|
assert.Equal(t, result, "text/html")
|
|
}
|
|
|
|
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, joinPaths("/a", ""), "/a")
|
|
assert.Equal(t, joinPaths("/a/", ""), "/a/")
|
|
assert.Equal(t, joinPaths("/a/", "/"), "/a/")
|
|
assert.Equal(t, joinPaths("/a", "/"), "/a/")
|
|
assert.Equal(t, joinPaths("/a", "/hola"), "/a/hola")
|
|
assert.Equal(t, joinPaths("/a/", "/hola"), "/a/hola")
|
|
assert.Equal(t, joinPaths("/a/", "/hola/"), "/a/hola/")
|
|
assert.Equal(t, joinPaths("/a/", "/hola//"), "/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, value.Foo, "hola")
|
|
assert.Equal(t, value.Bar, 10)
|
|
|
|
called = false
|
|
performRequest(router, "GET", "/?foo=hola&bar=1")
|
|
assert.False(t, called)
|
|
|
|
assert.Panics(t, func() {
|
|
Bind(&bindTestStruct{})
|
|
})
|
|
}
|