Function TestBodyDump_ClientGetsFullResponse

middleware/body_dump_test.go:374–404

(t *testing.T)

}

func TestBodyDump_ClientGetsFullResponse(t *testing.T) {
	e := echo.New()
	// This is critical - even when dump is limited, client gets everything
	largeResponse := strings.Repeat("DATA", 500) // 2KB
	req := httptest.NewRequest(http.MethodGet, "/", nil)
	rec := httptest.NewRecorder()
	c := e.NewContext(req, rec)

	h := func(c *echo.Context) error {
		// Write response in chunks to test incremental writes
		for range 4 {
			c.Response().Write([]byte(strings.Repeat("DATA", 125)))
		}
		return nil
	}

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

	err = mw(h)(c)
	assert.NoError(t, err)
	assert.Equal(t, 512, len(responseBodyDumped), "Dump should be limited")
	assert.Equal(t, largeResponse, rec.Body.String(), "Client must get full response")
}

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