Function TestBodyDump_BothLimitsSimultaneous

middleware/body_dump_test.go:406–436

(t *testing.T)

}

func TestBodyDump_BothLimitsSimultaneous(t *testing.T) {
	e := echo.New()
	largeRequest := strings.Repeat("Q", 2*1024)
	largeResponse := strings.Repeat("R", 2*1024)
	req := httptest.NewRequest(http.MethodPost, "/", strings.NewReader(largeRequest))
	rec := httptest.NewRecorder()
	c := e.NewContext(req, rec)

	h := func(c *echo.Context) error {
		io.ReadAll(c.Request().Body) // Consume request
		return c.String(http.StatusOK, largeResponse)
	}

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

	err = mw(h)(c)
	assert.NoError(t, err)
	assert.Equal(t, int(limit), len(requestBodyDumped), "Request dump should be limited")
	assert.Equal(t, int(limit), len(responseBodyDumped), "Response dump should be limited")
}

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