Add OTLP compression check tests

Signed-off-by: SungJin1212 <[email protected]>

pkg/util/push/otlp_test.go

@@ -2,6 +2,7 @@ package push
 
 import (
 	"bytes"
+	"compress/gzip"
 	"context"
 	"io"
 	"net/http"
@@ -233,6 +234,8 @@ func TestOTLPWriteHandler(t *testing.T) {
 		format             string
 		expectedStatusCode int
 		expectedErrMsg     string
+		compression        bool
+		encodingType       string
 	}{
 		{
 			description:        "Test proto format write",
@@ -260,6 +263,21 @@ func TestOTLPWriteHandler(t *testing.T) {
 			expectedStatusCode: http.StatusBadRequest,
 			expectedErrMsg:     "received message larger than max",
 		},
+		{
+			description:        "valid encoding type: gzip",
+			maxRecvMsgSize:     10000,
+			format:             jsonContentType,
+			expectedStatusCode: http.StatusOK,
+			encodingType:       "gzip",
+			compression:        true,
+		},
+		{
+			description:        "invalid encoding type: snappy",
+			maxRecvMsgSize:     10000,
+			format:             jsonContentType,
+			expectedStatusCode: http.StatusBadRequest,
+			encodingType:       "snappy",
+		},
 	}
 
 	for _, test := range tests {
@@ -268,20 +286,40 @@ func TestOTLPWriteHandler(t *testing.T) {
 			ctx = user.InjectOrgID(ctx, "user-1")
 			var req *http.Request
 
+			compressionFunc := func(t *testing.T, body []byte) []byte {
+				var b bytes.Buffer
+				gz := gzip.NewWriter(&b)
+				_, err := gz.Write(body)
+				require.NoError(t, err)
+				require.NoError(t, gz.Close())
+
+				return b.Bytes()
+			}
+
 			if test.format == pbContentType {
 				buf, err := exportRequest.MarshalProto()
 				require.NoError(t, err)
 
+				if test.compression {
+					buf = compressionFunc(t, buf)
+				}
+
 				req, err = http.NewRequestWithContext(ctx, "", "", bytes.NewReader(buf))
 				require.NoError(t, err)
 				req.Header.Set("Content-Type", pbContentType)
+				req.Header.Set("Content-Encoding", test.encodingType)
 			} else {
 				buf, err := exportRequest.MarshalJSON()
 				require.NoError(t, err)
 
+				if test.compression {
+					buf = compressionFunc(t, buf)
+				}
+
 				req, err = http.NewRequestWithContext(ctx, "", "", bytes.NewReader(buf))
 				require.NoError(t, err)
 				req.Header.Set("Content-Type", jsonContentType)
+				req.Header.Set("Content-Encoding", test.encodingType)
 			}
 
 			push := verifyOTLPWriteRequestHandler(t, cortexpb.API)