1
0
mirror of https://github.com/labstack/echo.git synced 2024-11-24 08:22:21 +02:00
echo/middleware/body_limit_test.go

34 lines
748 B
Go
Raw Normal View History

package middleware
import (
"io/ioutil"
"net/http"
"testing"
"bytes"
"github.com/labstack/echo"
"github.com/labstack/echo/test"
"github.com/stretchr/testify/assert"
)
func TestBodyLimit(t *testing.T) {
e := echo.New()
req := test.NewRequest(echo.POST, "/", bytes.NewReader([]byte("Hello, World!")))
rec := test.NewResponseRecorder()
c := e.NewContext(req, rec)
h := func(c echo.Context) error {
body, _ := ioutil.ReadAll(c.Request().Body())
return c.String(http.StatusOK, string(body))
}
// Within limit
BodyLimit("2M")(h)(c)
assert.Equal(t, http.StatusOK, rec.Status())
assert.Equal(t, "Hello, World!", rec.Body.String())
// Overlimit
// BodyLimit("2B")(h)(c)
// assert.Equal(t, "Hello, World!", rec.Body.String())
}