Function TestBodyDump_RequestExceedsLimit

middleware/body_dump_test.go:255–285

(t *testing.T)

}

func TestBodyDump_RequestExceedsLimit(t *testing.T) {
	e := echo.New()
	// Create 2KB of data but limit to 1KB
	largeData := strings.Repeat("A", 2*1024)
	req := httptest.NewRequest(http.MethodPost, "/", strings.NewReader(largeData))
	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) // 1KB limit
	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), "Dumped request should be truncated to limit")
	assert.Equal(t, strings.Repeat("A", 1024), requestBodyDumped, "Dumped data should match first N bytes")
	// Handler should receive truncated data (what was dumped)
	assert.Equal(t, strings.Repeat("A", 1024), rec.Body.String())
}

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