Function TestDecompress_MultipleReads

middleware/decompress_test.go:420–455

(t *testing.T)

}

func TestDecompress_MultipleReads(t *testing.T) {
	e := echo.New()
	// Test that limit is enforced across multiple Read() calls
	largeBody := strings.Repeat("M", 2*1024) // 2KB
	gz, _ := gzipString(largeBody)

	req := httptest.NewRequest(http.MethodPost, "/", bytes.NewReader(gz))
	req.Header.Set(echo.HeaderContentEncoding, GZIPEncoding)
	rec := httptest.NewRecorder()
	c := e.NewContext(req, rec)

	h, err := DecompressConfig{MaxDecompressedSize: 1024}.ToMiddleware() // 1KB limit
	assert.NoError(t, err)

	err = h(func(c *echo.Context) error {
		// Read in small chunks
		buf := make([]byte, 256)
		total := 0
		for {
			n, readErr := c.Request().Body.Read(buf)
			total += n
			if readErr != nil {
				if readErr == io.EOF {
					return nil
				}
				return readErr
			}
		}
	})(c)

	// Should return 413 error from cumulative reads
	assert.Error(t, err)
	he, ok := err.(echo.HTTPStatusCoder)
	assert.True(t, ok)
	assert.Equal(t, http.StatusRequestEntityTooLarge, he.StatusCode())
}

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