Function TestBodyDecodeWithLimitTooLarge

http_test.go:463–525

(t *testing.T)

}

func TestBodyDecodeWithLimitTooLarge(t *testing.T) {
	t.Parallel()

	body := bytes.Repeat([]byte("a"), 2*1024)
	maxBodySize := 1024

	testCases := []struct {
		name     string
		encoding string
		encode   func([]byte) []byte
	}{
		{
			name:     "gzip",
			encoding: "gzip",
			encode: func(src []byte) []byte {
				return AppendGzipBytes(nil, src)
			},
		},
		{
			name:     "deflate",
			encoding: "deflate",
			encode: func(src []byte) []byte {
				return AppendDeflateBytes(nil, src)
			},
		},
		{
			name:     "brotli",
			encoding: "br",
			encode: func(src []byte) []byte {
				return AppendBrotliBytes(nil, src)
			},
		},
		{
			name:     "zstd",
			encoding: "zstd",
			encode: func(src []byte) []byte {
				return AppendZstdBytes(nil, src)
			},
		},
	}

	for _, testCase := range testCases {
		t.Run(testCase.name+"_request_uncompressed", func(t *testing.T) {
			var req Request
			req.Header.SetContentEncoding(testCase.encoding)
			req.SetBodyRaw(testCase.encode(body))
			_, err := req.BodyUncompressedWithLimit(maxBodySize)
			if !errors.Is(err, ErrBodyTooLarge) {
				t.Fatalf("unexpected error: %v", err)
			}
		})

		t.Run(testCase.name+"_response_uncompressed", func(t *testing.T) {
			var resp Response
			resp.Header.SetContentEncoding(testCase.encoding)
			resp.SetBodyRaw(testCase.encode(body))
			_, err := resp.BodyUncompressedWithLimit(maxBodySize)
			if !errors.Is(err, ErrBodyTooLarge) {