Skip to content

Commit 184ead9

Browse files
authored
Add json format oltp ingestion test (#6286)
Signed-off-by: SungJin1212 <[email protected]>
1 parent 3b5011b commit 184ead9

File tree

1 file changed

+30
-11
lines changed

1 file changed

+30
-11
lines changed

pkg/util/push/otlp_test.go

Lines changed: 30 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -20,21 +20,40 @@ import (
2020
func TestOTLPWriteHandler(t *testing.T) {
2121
exportRequest := generateOTLPWriteRequest(t)
2222

23-
buf, err := exportRequest.MarshalProto()
24-
require.NoError(t, err)
23+
t.Run("Test proto format write", func(t *testing.T) {
24+
buf, err := exportRequest.MarshalProto()
25+
require.NoError(t, err)
2526

26-
req, err := http.NewRequest("", "", bytes.NewReader(buf))
27-
require.NoError(t, err)
28-
req.Header.Set("Content-Type", "application/x-protobuf")
27+
req, err := http.NewRequest("", "", bytes.NewReader(buf))
28+
require.NoError(t, err)
29+
req.Header.Set("Content-Type", "application/x-protobuf")
2930

30-
push := verifyOTLPWriteRequestHandler(t, cortexpb.API)
31-
handler := OTLPHandler(nil, push)
31+
push := verifyOTLPWriteRequestHandler(t, cortexpb.API)
32+
handler := OTLPHandler(nil, push)
3233

33-
recorder := httptest.NewRecorder()
34-
handler.ServeHTTP(recorder, req)
34+
recorder := httptest.NewRecorder()
35+
handler.ServeHTTP(recorder, req)
3536

36-
resp := recorder.Result()
37-
require.Equal(t, http.StatusOK, resp.StatusCode)
37+
resp := recorder.Result()
38+
require.Equal(t, http.StatusOK, resp.StatusCode)
39+
})
40+
t.Run("Test json format write", func(t *testing.T) {
41+
buf, err := exportRequest.MarshalJSON()
42+
require.NoError(t, err)
43+
44+
req, err := http.NewRequest("", "", bytes.NewReader(buf))
45+
require.NoError(t, err)
46+
req.Header.Set("Content-Type", "application/json")
47+
48+
push := verifyOTLPWriteRequestHandler(t, cortexpb.API)
49+
handler := OTLPHandler(nil, push)
50+
51+
recorder := httptest.NewRecorder()
52+
handler.ServeHTTP(recorder, req)
53+
54+
resp := recorder.Result()
55+
require.Equal(t, http.StatusOK, resp.StatusCode)
56+
})
3857
}
3958

4059
func generateOTLPWriteRequest(t *testing.T) pmetricotlp.ExportRequest {

0 commit comments

Comments
 (0)