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

Fix #1526 trailing slash to any route (#1563)

* refs #1526: Add tests for trailing slash requests with nested any routes

* refs #1526: Handle specual router case with trailing slash for non-root any route

* refs #1526: Fix accidential lookup for any route without trailing slash in request
This commit is contained in:
Roland Lammel 2020-05-06 23:01:28 +02:00 committed by GitHub
parent c08f30359b
commit 43e32ba83d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 76 additions and 1 deletions

View File

@ -355,6 +355,10 @@ func (r *Router) Find(method, path string, c Context) {
// Attempt to go back up the tree on no matching prefix or no remaining search
if l != pl || search == "" {
// Handle special case of trailing slash route with existing any route (see #1526)
if path[len(path)-1] == '/' && cn.findChildByKind(akind) != nil {
goto Any
}
if nn == nil { // Issue #1348
return // Not found
}

View File

@ -608,7 +608,6 @@ func TestRouterMatchAny(t *testing.T) {
return nil
})
c := e.NewContext(nil, nil).(*context)
r.Find(http.MethodGet, "/", c)
assert.Equal(t, "", c.Param("*"))
@ -619,6 +618,78 @@ func TestRouterMatchAny(t *testing.T) {
assert.Equal(t, "joe", c.Param("*"))
}
// TestRouterMatchAnySlash shall verify finding the best route
// for any routes with trailing slash requests
func TestRouterMatchAnySlash(t *testing.T) {
e := New()
r := e.router
handler := func(c Context) error {
c.Set("path", c.Path())
return nil
}
// Routes
r.Add(http.MethodGet, "/users", handler)
r.Add(http.MethodGet, "/users/*", handler)
r.Add(http.MethodGet, "/img/*", handler)
r.Add(http.MethodGet, "/img/load", handler)
r.Add(http.MethodGet, "/img/load/*", handler)
r.Add(http.MethodGet, "/assets/*", handler)
c := e.NewContext(nil, nil).(*context)
r.Find(http.MethodGet, "/", c)
assert.Equal(t, "", c.Param("*"))
// Test trailing slash request for simple any route (see #1526)
c = e.NewContext(nil, nil).(*context)
r.Find(http.MethodGet, "/users/", c)
c.handler(c)
assert.Equal(t, "/users/*", c.Get("path"))
assert.Equal(t, "", c.Param("*"))
c = e.NewContext(nil, nil).(*context)
r.Find(http.MethodGet, "/users/joe", c)
c.handler(c)
assert.Equal(t, "/users/*", c.Get("path"))
assert.Equal(t, "joe", c.Param("*"))
// Test trailing slash request for nested any route (see #1526)
c = e.NewContext(nil, nil).(*context)
r.Find(http.MethodGet, "/img/load", c)
c.handler(c)
assert.Equal(t, "/img/load", c.Get("path"))
assert.Equal(t, "", c.Param("*"))
c = e.NewContext(nil, nil).(*context)
r.Find(http.MethodGet, "/img/load/", c)
c.handler(c)
assert.Equal(t, "/img/load/*", c.Get("path"))
assert.Equal(t, "", c.Param("*"))
c = e.NewContext(nil, nil).(*context)
r.Find(http.MethodGet, "/img/load/ben", c)
c.handler(c)
assert.Equal(t, "/img/load/*", c.Get("path"))
assert.Equal(t, "ben", c.Param("*"))
// Test /assets/* any route
// ... without trailing slash must not match
c = e.NewContext(nil, nil).(*context)
r.Find(http.MethodGet, "/assets", c)
c.handler(c)
assert.Equal(t, nil, c.Get("path"))
assert.Equal(t, "", c.Param("*"))
// ... with trailing slash must match
c = e.NewContext(nil, nil).(*context)
r.Find(http.MethodGet, "/assets/", c)
c.handler(c)
assert.Equal(t, "/assets/*", c.Get("path"))
assert.Equal(t, "", c.Param("*"))
}
func TestRouterMatchAnyMultiLevel(t *testing.T) {
e := New()
r := e.router