Function TestBodyDump_RequestAtExactLimit

middleware/body_dump_test.go:287–314

(t *testing.T)

}

func TestBodyDump_RequestAtExactLimit(t *testing.T) {
	e := echo.New()
	exactData := strings.Repeat("B", 1024)
	req := httptest.NewRequest(http.MethodPost, "/", strings.NewReader(exactData))
	rec := httptest.NewRecorder()
	c := e.NewContext(req, rec)

	h := func(c *echo.Context) error {
		body, _ := io.ReadAll(c.Request().Body)
		return c.String(http.StatusOK, string(body))
	}

	requestBodyDumped := ""
	limit := int64(1024)
	mw, err := BodyDumpConfig{
		Handler: func(c *echo.Context, reqBody, resBody []byte, err error) {
			requestBodyDumped = string(reqBody)
		},
		MaxRequestBytes:  limit,
		MaxResponseBytes: 1 * MB,
	}.ToMiddleware()
	assert.NoError(t, err)

	err = mw(h)(c)
	assert.NoError(t, err)
	assert.Equal(t, int(limit), len(requestBodyDumped), "Exact limit should dump full data")
	assert.Equal(t, exactData, requestBodyDumped)
}

func TestBodyDump_ResponseWithinLimit(t *testing.T) {
	e := echo.New()