1
0
mirror of https://github.com/labstack/echo.git synced 2024-12-24 20:14:31 +02:00

Changes based on comments for #430

Signed-off-by: Vishal Rana <vr@labstack.com>
This commit is contained in:
Vishal Rana 2016-04-16 15:53:27 -07:00
parent fd104333f2
commit 467cf05b41
11 changed files with 191 additions and 169 deletions

2
.gitignore vendored
View File

@ -11,3 +11,5 @@ node_modules
# Glide
vendor
.DS_Store

View File

@ -40,6 +40,9 @@ type (
// Path returns the registered path for the handler.
Path() string
// SetPath sets the registered path for the handler.
SetPath(string)
// P returns path parameter by index.
P(int) string
@ -49,11 +52,20 @@ type (
// ParamNames returns path parameter names.
ParamNames() []string
// SetParamNames sets path parameter names.
SetParamNames([]string)
// ParamValues returns path parameter values.
ParamValues() []string
// SetParamValues sets path parameter values.
SetParamValues([]string)
// QueryParam returns the query param for the provided name. It is an alias
// for `engine.URL#QueryParam()`.
QueryParam(string) string
// QueryParam returns the query parameters as map. It is an alias for `engine.URL#QueryParams()`.
// QueryParams returns the query parameters as map. It is an alias for `engine.URL#QueryParams()`.
QueryParams() map[string][]string
// FormValue returns the form field value for the provided name. It is an
@ -76,6 +88,9 @@ type (
// Set saves data in the context.
Set(string, interface{})
// Del data from the context
Del(string)
// Bind binds the request body into provided type `i`. The default binder
// does it based on Content-Type header.
Bind(interface{}) error
@ -125,6 +140,9 @@ type (
// Handler returns the matched handler by router.
Handler() HandlerFunc
// SetHandler sets the matched handler by router.
SetHandler(HandlerFunc)
// Logger returns the `Logger` instance.
Logger() *log.Logger
@ -136,9 +154,6 @@ type (
// and `Last-Modified` response headers.
ServeContent(io.ReadSeeker, string, time.Time) error
// Object returns the internal context implementation.
Object() *context
// Reset resets the context after request completes. It must be called along
// with `Echo#GetContext()` and `Echo#PutContext()`. See `Echo#ServeHTTP()`
Reset(engine.Request, engine.Response)
@ -163,32 +178,6 @@ const (
indexPage = "index.html"
)
// NewContext creates a Context object.
func NewContext(rq engine.Request, rs engine.Response, e *Echo) Context {
return &context{
request: rq,
response: rs,
echo: e,
pvalues: make([]string, *e.maxParam),
store: make(store),
handler: notFoundHandler,
}
}
// MockContext returns `Context` for testing purpose.
func MockContext(request engine.Request, response engine.Response, path string, paramNames []string, paramValues []string) Context {
return &context{
request: request,
response: response,
echo: new(Echo),
path: path,
pnames: paramNames,
pvalues: paramValues,
store: make(store),
handler: notFoundHandler,
}
}
func (c *context) NetContext() netContext.Context {
return c.netContext
}
@ -225,6 +214,10 @@ func (c *context) Path() string {
return c.path
}
func (c *context) SetPath(p string) {
c.path = p
}
func (c *context) P(i int) (value string) {
l := len(c.pnames)
if i < l {
@ -248,6 +241,18 @@ func (c *context) ParamNames() []string {
return c.pnames
}
func (c *context) SetParamNames(names []string) {
c.pnames = names
}
func (c *context) ParamValues() []string {
return c.pvalues
}
func (c *context) SetParamValues(values []string) {
c.pvalues = values
}
func (c *context) QueryParam(name string) string {
return c.request.URL().QueryParam(name)
}
@ -283,6 +288,10 @@ func (c *context) Get(key string) interface{} {
return c.store[key]
}
func (c *context) Del(key string) {
delete(c.store, key)
}
func (c *context) Bind(i interface{}) error {
return c.echo.binder.Bind(i, c)
}
@ -426,12 +435,12 @@ func (c *context) Handler() HandlerFunc {
return c.handler
}
func (c *context) Logger() *log.Logger {
return c.echo.logger
func (c *context) SetHandler(h HandlerFunc) {
c.handler = h
}
func (c *context) Object() *context {
return c
func (c *context) Logger() *log.Logger {
return c.echo.logger
}
func (c *context) ServeContent(content io.ReadSeeker, name string, modtime time.Time) error {

View File

@ -36,8 +36,8 @@ func TestContext(t *testing.T) {
e := New()
rq := test.NewRequest(POST, "/", strings.NewReader(userJSON))
rec := test.NewResponseRecorder()
c := NewContext(rq, rec, e)
rc := test.NewResponseRecorder()
c := e.NewContext(rq, rc).(*context)
// Request
assert.NotNil(t, c.Request())
@ -46,12 +46,12 @@ func TestContext(t *testing.T) {
assert.NotNil(t, c.Response())
// ParamNames
c.Object().pnames = []string{"uid", "fid"}
c.pnames = []string{"uid", "fid"}
assert.EqualValues(t, []string{"uid", "fid"}, c.ParamNames())
// Param by id
c.Object().pnames = []string{"id"}
c.Object().pvalues = []string{"1"}
c.pnames = []string{"id"}
c.pvalues = []string{"1"}
assert.Equal(t, "1", c.P(0))
// Param by name
@ -67,13 +67,13 @@ func TestContext(t *testing.T) {
// JSON
testBindOk(t, c, MIMEApplicationJSON)
c.Object().request = test.NewRequest(POST, "/", strings.NewReader(invalidContent))
c.request = test.NewRequest(POST, "/", strings.NewReader(invalidContent))
testBindError(t, c, MIMEApplicationJSON)
// XML
c.Object().request = test.NewRequest(POST, "/", strings.NewReader(userXML))
c.request = test.NewRequest(POST, "/", strings.NewReader(userXML))
testBindOk(t, c, MIMEApplicationXML)
c.Object().request = test.NewRequest(POST, "/", strings.NewReader(invalidContent))
c.request = test.NewRequest(POST, "/", strings.NewReader(invalidContent))
testBindError(t, c, MIMEApplicationXML)
// Unsupported
@ -86,114 +86,114 @@ func TestContext(t *testing.T) {
tpl := &Template{
templates: template.Must(template.New("hello").Parse("Hello, {{.}}!")),
}
c.Object().echo.SetRenderer(tpl)
c.echo.SetRenderer(tpl)
err := c.Render(http.StatusOK, "hello", "Joe")
if assert.NoError(t, err) {
assert.Equal(t, http.StatusOK, rec.Status())
assert.Equal(t, "Hello, Joe!", rec.Body.String())
assert.Equal(t, http.StatusOK, rc.Status())
assert.Equal(t, "Hello, Joe!", rc.Body.String())
}
c.Object().echo.renderer = nil
c.echo.renderer = nil
err = c.Render(http.StatusOK, "hello", "Joe")
assert.Error(t, err)
// JSON
rec = test.NewResponseRecorder()
c = NewContext(rq, rec, e)
rc = test.NewResponseRecorder()
c = e.NewContext(rq, rc).(*context)
err = c.JSON(http.StatusOK, user{"1", "Joe"})
if assert.NoError(t, err) {
assert.Equal(t, http.StatusOK, rec.Status())
assert.Equal(t, MIMEApplicationJSONCharsetUTF8, rec.Header().Get(HeaderContentType))
assert.Equal(t, userJSON, rec.Body.String())
assert.Equal(t, http.StatusOK, rc.Status())
assert.Equal(t, MIMEApplicationJSONCharsetUTF8, rc.Header().Get(HeaderContentType))
assert.Equal(t, userJSON, rc.Body.String())
}
// JSON (error)
rec = test.NewResponseRecorder()
c = NewContext(rq, rec, e)
rc = test.NewResponseRecorder()
c = e.NewContext(rq, rc).(*context)
err = c.JSON(http.StatusOK, make(chan bool))
assert.Error(t, err)
// JSONP
rec = test.NewResponseRecorder()
c = NewContext(rq, rec, e)
rc = test.NewResponseRecorder()
c = e.NewContext(rq, rc).(*context)
callback := "callback"
err = c.JSONP(http.StatusOK, callback, user{"1", "Joe"})
if assert.NoError(t, err) {
assert.Equal(t, http.StatusOK, rec.Status())
assert.Equal(t, MIMEApplicationJavaScriptCharsetUTF8, rec.Header().Get(HeaderContentType))
assert.Equal(t, callback+"("+userJSON+");", rec.Body.String())
assert.Equal(t, http.StatusOK, rc.Status())
assert.Equal(t, MIMEApplicationJavaScriptCharsetUTF8, rc.Header().Get(HeaderContentType))
assert.Equal(t, callback+"("+userJSON+");", rc.Body.String())
}
// XML
rec = test.NewResponseRecorder()
c = NewContext(rq, rec, e)
rc = test.NewResponseRecorder()
c = e.NewContext(rq, rc).(*context)
err = c.XML(http.StatusOK, user{"1", "Joe"})
if assert.NoError(t, err) {
assert.Equal(t, http.StatusOK, rec.Status())
assert.Equal(t, MIMEApplicationXMLCharsetUTF8, rec.Header().Get(HeaderContentType))
assert.Equal(t, xml.Header+userXML, rec.Body.String())
assert.Equal(t, http.StatusOK, rc.Status())
assert.Equal(t, MIMEApplicationXMLCharsetUTF8, rc.Header().Get(HeaderContentType))
assert.Equal(t, xml.Header+userXML, rc.Body.String())
}
// XML (error)
rec = test.NewResponseRecorder()
c = NewContext(rq, rec, e)
rc = test.NewResponseRecorder()
c = e.NewContext(rq, rc).(*context)
err = c.XML(http.StatusOK, make(chan bool))
assert.Error(t, err)
// String
rec = test.NewResponseRecorder()
c = NewContext(rq, rec, e)
rc = test.NewResponseRecorder()
c = e.NewContext(rq, rc).(*context)
err = c.String(http.StatusOK, "Hello, World!")
if assert.NoError(t, err) {
assert.Equal(t, http.StatusOK, rec.Status())
assert.Equal(t, MIMETextPlainCharsetUTF8, rec.Header().Get(HeaderContentType))
assert.Equal(t, "Hello, World!", rec.Body.String())
assert.Equal(t, http.StatusOK, rc.Status())
assert.Equal(t, MIMETextPlainCharsetUTF8, rc.Header().Get(HeaderContentType))
assert.Equal(t, "Hello, World!", rc.Body.String())
}
// HTML
rec = test.NewResponseRecorder()
c = NewContext(rq, rec, e)
rc = test.NewResponseRecorder()
c = e.NewContext(rq, rc).(*context)
err = c.HTML(http.StatusOK, "Hello, <strong>World!</strong>")
if assert.NoError(t, err) {
assert.Equal(t, http.StatusOK, rec.Status())
assert.Equal(t, MIMETextHTMLCharsetUTF8, rec.Header().Get(HeaderContentType))
assert.Equal(t, "Hello, <strong>World!</strong>", rec.Body.String())
assert.Equal(t, http.StatusOK, rc.Status())
assert.Equal(t, MIMETextHTMLCharsetUTF8, rc.Header().Get(HeaderContentType))
assert.Equal(t, "Hello, <strong>World!</strong>", rc.Body.String())
}
// Attachment
rec = test.NewResponseRecorder()
c = NewContext(rq, rec, e)
rc = test.NewResponseRecorder()
c = e.NewContext(rq, rc).(*context)
file, err := os.Open("_fixture/images/walle.png")
if assert.NoError(t, err) {
err = c.Attachment(file, "walle.png")
if assert.NoError(t, err) {
assert.Equal(t, http.StatusOK, rec.Status())
assert.Equal(t, "attachment; filename=walle.png", rec.Header().Get(HeaderContentDisposition))
assert.Equal(t, 219885, rec.Body.Len())
assert.Equal(t, http.StatusOK, rc.Status())
assert.Equal(t, "attachment; filename=walle.png", rc.Header().Get(HeaderContentDisposition))
assert.Equal(t, 219885, rc.Body.Len())
}
}
// NoContent
rec = test.NewResponseRecorder()
c = NewContext(rq, rec, e)
rc = test.NewResponseRecorder()
c = e.NewContext(rq, rc).(*context)
c.NoContent(http.StatusOK)
assert.Equal(t, http.StatusOK, rec.Status())
assert.Equal(t, http.StatusOK, rc.Status())
// Redirect
rec = test.NewResponseRecorder()
c = NewContext(rq, rec, e)
rc = test.NewResponseRecorder()
c = e.NewContext(rq, rc).(*context)
assert.Equal(t, nil, c.Redirect(http.StatusMovedPermanently, "http://labstack.github.io/echo"))
assert.Equal(t, http.StatusMovedPermanently, rec.Status())
assert.Equal(t, "http://labstack.github.io/echo", rec.Header().Get(HeaderLocation))
assert.Equal(t, http.StatusMovedPermanently, rc.Status())
assert.Equal(t, "http://labstack.github.io/echo", rc.Header().Get(HeaderLocation))
// Error
rec = test.NewResponseRecorder()
c = NewContext(rq, rec, e)
rc = test.NewResponseRecorder()
c = e.NewContext(rq, rc).(*context)
c.Error(errors.New("error"))
assert.Equal(t, http.StatusInternalServerError, rec.Status())
assert.Equal(t, http.StatusInternalServerError, rc.Status())
// Reset
c.Object().Reset(rq, test.NewResponseRecorder())
c.Reset(rq, test.NewResponseRecorder())
}
func TestContextPath(t *testing.T) {
@ -201,12 +201,12 @@ func TestContextPath(t *testing.T) {
r := e.Router()
r.Add(GET, "/users/:id", nil, e)
c := NewContext(nil, nil, e)
c := e.NewContext(nil, nil)
r.Find(GET, "/users/1", c)
assert.Equal(t, "/users/:id", c.Path())
r.Add(GET, "/users/:uid/files/:fid", nil, e)
c = NewContext(nil, nil, e)
c = e.NewContext(nil, nil)
r.Find(GET, "/users/1/files/1", c)
assert.Equal(t, "/users/:uid/files/:fid", c.Path())
}
@ -216,7 +216,8 @@ func TestContextQueryParam(t *testing.T) {
q.Set("name", "joe")
q.Set("email", "joe@labstack.com")
rq := test.NewRequest(GET, "/?"+q.Encode(), nil)
c := NewContext(rq, nil, New())
e := New()
c := e.NewContext(rq, nil)
assert.Equal(t, "joe", c.QueryParam("name"))
assert.Equal(t, "joe@labstack.com", c.QueryParam("email"))
}
@ -226,10 +227,11 @@ func TestContextFormValue(t *testing.T) {
f.Set("name", "joe")
f.Set("email", "joe@labstack.com")
e := New()
rq := test.NewRequest(POST, "/", strings.NewReader(f.Encode()))
rq.Header().Add(HeaderContentType, MIMEApplicationForm)
c := NewContext(rq, nil, New())
c := e.NewContext(rq, nil)
assert.Equal(t, "joe", c.FormValue("name"))
assert.Equal(t, "joe@labstack.com", c.FormValue("email"))
}
@ -244,7 +246,7 @@ func TestContextServeContent(t *testing.T) {
e := New()
rq := test.NewRequest(GET, "/", nil)
rc := test.NewResponseRecorder()
c := NewContext(rq, rc, e)
c := e.NewContext(rq, rc)
fs := http.Dir("_fixture/images")
f, err := fs.Open("walle.png")
@ -258,7 +260,7 @@ func TestContextServeContent(t *testing.T) {
// Cached
rc = test.NewResponseRecorder()
c = NewContext(rq, rc, e)
c = e.NewContext(rq, rc)
rq.Header().Set(HeaderIfModifiedSince, fi.ModTime().UTC().Format(http.TimeFormat))
if assert.NoError(t, c.ServeContent(f, fi.Name(), fi.ModTime())) {
assert.Equal(t, http.StatusNotModified, rc.Status())
@ -276,7 +278,7 @@ func TestContextHandler(t *testing.T) {
_, err := b.Write([]byte("handler"))
return err
}, e)
c := NewContext(nil, nil, e)
c := e.NewContext(nil, nil)
r.Find(GET, "/handler", c)
c.Handler()(c)
assert.Equal(t, "handler", b.String())

14
echo.go
View File

@ -215,7 +215,7 @@ var (
func New() (e *Echo) {
e = &Echo{maxParam: new(int)}
e.pool.New = func() interface{} {
return NewContext(nil, nil, e)
return e.NewContext(nil, nil)
}
e.router = NewRouter(e)
@ -228,6 +228,18 @@ func New() (e *Echo) {
return
}
// NewContext returns a Context instance.
func (e *Echo) NewContext(rq engine.Request, rs engine.Response) Context {
return &context{
request: rq,
response: rs,
echo: e,
pvalues: make([]string, *e.maxParam),
store: make(store),
handler: notFoundHandler,
}
}
// Router returns router.
func (e *Echo) Router() *Router {
return e.router

View File

@ -25,8 +25,8 @@ type (
func TestEcho(t *testing.T) {
e := New()
rq := test.NewRequest(GET, "/", nil)
rec := test.NewResponseRecorder()
c := NewContext(rq, rec, e)
rc := test.NewResponseRecorder()
c := e.NewContext(rq, rc)
// Router
assert.NotNil(t, e.Router())
@ -37,7 +37,7 @@ func TestEcho(t *testing.T) {
// DefaultHTTPErrorHandler
e.DefaultHTTPErrorHandler(errors.New("error"), c)
assert.Equal(t, http.StatusInternalServerError, rec.Status())
assert.Equal(t, http.StatusInternalServerError, rc.Status())
}
func TestEchoStatic(t *testing.T) {

View File

@ -22,14 +22,13 @@ type (
}
)
// MockRequest returns `Request` instance for testing purpose.
func MockRequest() *Request {
ctx := new(fasthttp.RequestCtx)
// NewRequest returns `Request` instance.
func NewRequest(c *fasthttp.RequestCtx, l *log.Logger) *Request {
return &Request{
RequestCtx: ctx,
url: &URL{URI: ctx.URI()},
header: &RequestHeader{RequestHeader: &ctx.Request.Header},
logger: log.New("test"),
RequestCtx: c,
url: &URL{URI: c.URI()},
header: &RequestHeader{RequestHeader: &c.Request.Header},
logger: l,
}
}

View File

@ -24,14 +24,13 @@ type (
}
)
// MockResponse returns `Response` instance for testing purpose.
func MockResponse() *Response {
ctx := new(fasthttp.RequestCtx)
// NewResponse returns `Response` instance.
func NewResponse(c *fasthttp.RequestCtx, l *log.Logger) *Response {
return &Response{
RequestCtx: ctx,
header: &ResponseHeader{ResponseHeader: &ctx.Response.Header},
writer: ctx,
logger: log.New("test"),
RequestCtx: c,
header: &ResponseHeader{ResponseHeader: &c.Response.Header},
writer: c,
logger: l,
}
}

View File

@ -19,14 +19,13 @@ type (
}
)
// MockRequest returns `Request` instance for testing purpose.
func MockRequest() *Request {
rq := new(http.Request)
// NewRequest returns `Request` instance.
func NewRequest(r *http.Request, l *log.Logger) *Request {
return &Request{
Request: new(http.Request),
url: &URL{URL: rq.URL},
header: &Header{Header: rq.Header},
logger: log.New("test"),
Request: r,
url: &URL{URL: r.URL},
header: &Header{Header: r.Header},
logger: l,
}
}

View File

@ -5,7 +5,6 @@ import (
"io"
"net"
"net/http"
"net/http/httptest"
"github.com/labstack/echo/engine"
"github.com/labstack/gommon/log"
@ -29,14 +28,13 @@ type (
}
)
// MockResponse returns `Response` instance for testing purpose.
func MockResponse() *Response {
rc := httptest.NewRecorder()
// NewResponse returns `Response` instance.
func NewResponse(w http.ResponseWriter, l *log.Logger) *Response {
return &Response{
ResponseWriter: rc,
header: &Header{Header: rc.Header()},
writer: rc,
logger: log.New("test"),
ResponseWriter: w,
header: &Header{Header: w.Header()},
writer: w,
logger: l,
}
}

View File

@ -287,16 +287,16 @@ func (n *node) checkMethodNotAllowed() HandlerFunc {
// - Reset it `Context#Reset()`
// - Return it `Echo#PutContext()`.
func (r *Router) Find(method, path string, context Context) {
ctx := context.Object()
cn := r.tree // Current node as root
var (
search = path
c *node // Child node
n int // Param counter
nk kind // Next kind
nn *node // Next node
ns string // Next search
search = path
c *node // Child node
n int // Param counter
nk kind // Next kind
nn *node // Next node
ns string // Next search
pvalues = context.ParamValues()
)
// Search order static > param > any
@ -356,7 +356,7 @@ func (r *Router) Find(method, path string, context Context) {
Param:
if c = cn.findChildByKind(pkind); c != nil {
// Issue #378
if len(ctx.pvalues) == n {
if len(pvalues) == n {
continue
}
@ -371,7 +371,7 @@ func (r *Router) Find(method, path string, context Context) {
i, l := 0, len(search)
for ; i < l && search[i] != '/'; i++ {
}
ctx.pvalues[n] = search[:i]
pvalues[n] = search[:i]
n++
search = search[i:]
continue
@ -393,30 +393,32 @@ func (r *Router) Find(method, path string, context Context) {
// Not found
return
}
ctx.pvalues[len(cn.pnames)-1] = search
pvalues[len(cn.pnames)-1] = search
goto End
}
End:
ctx.handler = cn.findHandler(method)
ctx.path = cn.ppath
ctx.pnames = cn.pnames
context.SetHandler(cn.findHandler(method))
context.SetPath(cn.ppath)
context.SetParamNames(cn.pnames)
// NOTE: Slow zone...
if ctx.handler == nil {
ctx.handler = cn.checkMethodNotAllowed()
if context.Handler() == nil {
context.SetHandler(cn.checkMethodNotAllowed())
// Dig further for any, might have an empty value for *, e.g.
// serving a directory. Issue #207.
if cn = cn.findChildByKind(akind); cn == nil {
return
}
if ctx.handler = cn.findHandler(method); ctx.handler == nil {
ctx.handler = cn.checkMethodNotAllowed()
if h := cn.findHandler(method); h != nil {
context.SetHandler(h)
} else {
context.SetHandler(cn.checkMethodNotAllowed())
}
ctx.path = cn.ppath
ctx.pnames = cn.pnames
ctx.pvalues[len(cn.pnames)-1] = ""
context.SetPath(cn.ppath)
context.SetParamNames(cn.pnames)
pvalues[len(cn.pnames)-1] = ""
}
return

View File

@ -281,7 +281,7 @@ func TestRouterStatic(t *testing.T) {
c.Set("path", path)
return nil
}, e)
c := NewContext(nil, nil, e).Object()
c := e.NewContext(nil, nil).(*context)
r.Find(GET, path, c)
c.handler(c)
assert.Equal(t, path, c.Get("path"))
@ -293,7 +293,7 @@ func TestRouterParam(t *testing.T) {
r.Add(GET, "/users/:id", func(c Context) error {
return nil
}, e)
c := NewContext(nil, nil, e)
c := e.NewContext(nil, nil).(*context)
r.Find(GET, "/users/1", c)
assert.Equal(t, "1", c.P(0))
}
@ -304,7 +304,7 @@ func TestRouterTwoParam(t *testing.T) {
r.Add(GET, "/users/:uid/files/:fid", func(Context) error {
return nil
}, e)
c := NewContext(nil, nil, e)
c := e.NewContext(nil, nil).(*context)
r.Find(GET, "/users/1/files/1", c)
assert.Equal(t, "1", c.P(0))
@ -324,7 +324,7 @@ func TestRouterParamWithSlash(t *testing.T) {
return nil
}, e)
c := NewContext(nil, nil, e)
c := e.NewContext(nil, nil).(*context)
assert.NotPanics(t, func() {
r.Find(GET, "/a/1/c/d/2/3", c)
})
@ -344,7 +344,7 @@ func TestRouterMatchAny(t *testing.T) {
r.Add(GET, "/users/*", func(Context) error {
return nil
}, e)
c := NewContext(nil, nil, e)
c := e.NewContext(nil, nil).(*context)
r.Find(GET, "/", c)
assert.Equal(t, "", c.P(0))
@ -362,7 +362,7 @@ func TestRouterMicroParam(t *testing.T) {
r.Add(GET, "/:a/:b/:c", func(c Context) error {
return nil
}, e)
c := NewContext(nil, nil, e)
c := e.NewContext(nil, nil).(*context)
r.Find(GET, "/1/2/3", c)
assert.Equal(t, "1", c.P(0))
assert.Equal(t, "2", c.P(1))
@ -377,7 +377,7 @@ func TestRouterMixParamMatchAny(t *testing.T) {
r.Add(GET, "/users/:id/*", func(c Context) error {
return nil
}, e)
c := NewContext(nil, nil, e).Object()
c := e.NewContext(nil, nil).(*context)
r.Find(GET, "/users/joe/comments", c)
c.handler(c)
@ -396,7 +396,7 @@ func TestRouterMultiRoute(t *testing.T) {
r.Add(GET, "/users/:id", func(c Context) error {
return nil
}, e)
c := NewContext(nil, nil, e).Object()
c := e.NewContext(nil, nil).(*context)
// Route > /users
r.Find(GET, "/users", c)
@ -408,7 +408,7 @@ func TestRouterMultiRoute(t *testing.T) {
assert.Equal(t, "1", c.P(0))
// Route > /user
c = NewContext(nil, nil, e).Object()
c = e.NewContext(nil, nil).(*context)
r.Find(GET, "/user", c)
he := c.handler(c).(*HTTPError)
assert.Equal(t, http.StatusNotFound, he.Code)
@ -447,7 +447,7 @@ func TestRouterPriority(t *testing.T) {
c.Set("g", 7)
return nil
}, e)
c := NewContext(nil, nil, e).Object()
c := e.NewContext(nil, nil).(*context)
// Route > /users
r.Find(GET, "/users", c)
@ -490,7 +490,7 @@ func TestRouterPriority(t *testing.T) {
func TestRouterPriorityNotFound(t *testing.T) {
e := New()
r := e.router
c := NewContext(nil, nil, e).Object()
c := e.NewContext(nil, nil).(*context)
// Add
r.Add(GET, "/a/foo", func(c Context) error {
@ -511,7 +511,7 @@ func TestRouterPriorityNotFound(t *testing.T) {
c.handler(c)
assert.Equal(t, 2, c.Get("b"))
c = NewContext(nil, nil, e).Object()
c = e.NewContext(nil, nil).(*context)
r.Find(GET, "/abc/def", c)
he := c.handler(c).(*HTTPError)
assert.Equal(t, http.StatusNotFound, he.Code)
@ -532,7 +532,7 @@ func TestRouterParamNames(t *testing.T) {
r.Add(GET, "/users/:uid/files/:fid", func(c Context) error {
return nil
}, e)
c := NewContext(nil, nil, e).Object()
c := e.NewContext(nil, nil).(*context)
// Route > /users
r.Find(GET, "/users", c)
@ -541,14 +541,14 @@ func TestRouterParamNames(t *testing.T) {
// Route > /users/:id
r.Find(GET, "/users/1", c)
assert.Equal(t, "id", c.Object().pnames[0])
assert.Equal(t, "id", c.pnames[0])
assert.Equal(t, "1", c.P(0))
// Route > /users/:uid/files/:fid
r.Find(GET, "/users/1/files/1", c)
assert.Equal(t, "uid", c.Object().pnames[0])
assert.Equal(t, "uid", c.pnames[0])
assert.Equal(t, "1", c.P(0))
assert.Equal(t, "fid", c.Object().pnames[1])
assert.Equal(t, "fid", c.pnames[1])
assert.Equal(t, "1", c.P(1))
}
@ -561,10 +561,10 @@ func TestRouterAPI(t *testing.T) {
return nil
}, e)
}
c := NewContext(nil, nil, e)
c := e.NewContext(nil, nil).(*context)
for _, route := range api {
r.Find(route.Method, route.Path, c)
for i, n := range c.Object().pnames {
for i, n := range c.pnames {
if assert.NotEmpty(t, n) {
assert.Equal(t, ":"+n, c.P(i))
}