go/echo
1
0
Fork 0
mirror of https://github.com/labstack/echo.git synced 2025-06-15 00:14:57 +02:00
Code Issues Releases Activity

Simplify code of Add/Remove trailing slash and fix bug (#1275)

Browse Source 
* Simplify code of Add/Remove trailing slash

- simplify code (more informative / understanding)
- assert collides with imported package name (in tests)
- fix unhandled errors

* add tests for https://github.com/labstack/echo/pull/1275#issuecomment-460467700
This commit is contained in:
Evgeniy Kulikov
2019-02-07 20:49:51 +03:00
committed by Vishal Rana
parent bc28fceaf3
commit 88965757af
2 changed files with 40 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
middleware/slash_test.go
View File

@ -10,6 +10,7 @@ import (
)
func TestAddTrailingSlash(t *testing.T) {
is := assert.New(t)
e := echo.New()
req := httptest.NewRequest(http.MethodGet, "/add-slash", nil)
rec := httptest.NewRecorder()
@ -17,11 +18,20 @@ func TestAddTrailingSlash(t *testing.T) {
h := AddTrailingSlash()(func(c echo.Context) error {
return nil
})
h(c)
is.NoError(h(c))
is.Equal("/add-slash/", req.URL.Path)
is.Equal("/add-slash/", req.RequestURI)
assert := assert.New(t)
assert.Equal("/add-slash/", req.URL.Path)
assert.Equal("/add-slash/", req.RequestURI)
// Method Connect must not fail:
req = httptest.NewRequest(http.MethodConnect, "", nil)
rec = httptest.NewRecorder()
c = e.NewContext(req, rec)
h = AddTrailingSlash()(func(c echo.Context) error {
return nil
})
is.NoError(h(c))
is.Equal("/", req.URL.Path)
is.Equal("/", req.RequestURI)
// With config
req = httptest.NewRequest(http.MethodGet, "/add-slash?key=value", nil)
@ -32,12 +42,13 @@ func TestAddTrailingSlash(t *testing.T) {
})(func(c echo.Context) error {
return nil
})
h(c)
assert.Equal(http.StatusMovedPermanently, rec.Code)
assert.Equal("/add-slash/?key=value", rec.Header().Get(echo.HeaderLocation))
is.NoError(h(c))
is.Equal(http.StatusMovedPermanently, rec.Code)
is.Equal("/add-slash/?key=value", rec.Header().Get(echo.HeaderLocation))
}
func TestRemoveTrailingSlash(t *testing.T) {
is := assert.New(t)
e := echo.New()
req := httptest.NewRequest(http.MethodGet, "/remove-slash/", nil)
rec := httptest.NewRecorder()
@ -45,12 +56,20 @@ func TestRemoveTrailingSlash(t *testing.T) {
h := RemoveTrailingSlash()(func(c echo.Context) error {
return nil
})
h(c)
is.NoError(h(c))
is.Equal("/remove-slash", req.URL.Path)
is.Equal("/remove-slash", req.RequestURI)
assert := assert.New(t)
assert.Equal("/remove-slash", req.URL.Path)
assert.Equal("/remove-slash", req.RequestURI)
// Method Connect must not fail:
req = httptest.NewRequest(http.MethodConnect, "", nil)
rec = httptest.NewRecorder()
c = e.NewContext(req, rec)
h = RemoveTrailingSlash()(func(c echo.Context) error {
return nil
})
is.NoError(h(c))
is.Equal("", req.URL.Path)
is.Equal("", req.RequestURI)
// With config
req = httptest.NewRequest(http.MethodGet, "/remove-slash/?key=value", nil)
@ -61,9 +80,9 @@ func TestRemoveTrailingSlash(t *testing.T) {
})(func(c echo.Context) error {
return nil
})
h(c)
assert.Equal(http.StatusMovedPermanently, rec.Code)
assert.Equal("/remove-slash?key=value", rec.Header().Get(echo.HeaderLocation))
is.NoError(h(c))
is.Equal(http.StatusMovedPermanently, rec.Code)
is.Equal("/remove-slash?key=value", rec.Header().Get(echo.HeaderLocation))
// With bare URL
req = httptest.NewRequest(http.MethodGet, "http://localhost", nil)
@ -72,6 +91,6 @@ func TestRemoveTrailingSlash(t *testing.T) {
h = RemoveTrailingSlash()(func(c echo.Context) error {
return nil
})
h(c)
assert.Equal("", req.URL.Path)
is.NoError(h(c))
is.Equal("", req.URL.Path)
}