@@ -20,21 +20,40 @@ import (
20
20
func TestOTLPWriteHandler (t * testing.T ) {
21
21
exportRequest := generateOTLPWriteRequest (t )
22
22
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 )
25
26
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" )
29
30
30
- push := verifyOTLPWriteRequestHandler (t , cortexpb .API )
31
- handler := OTLPHandler (nil , push )
31
+ push := verifyOTLPWriteRequestHandler (t , cortexpb .API )
32
+ handler := OTLPHandler (nil , push )
32
33
33
- recorder := httptest .NewRecorder ()
34
- handler .ServeHTTP (recorder , req )
34
+ recorder := httptest .NewRecorder ()
35
+ handler .ServeHTTP (recorder , req )
35
36
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
+ })
38
57
}
39
58
40
59
func generateOTLPWriteRequest (t * testing.T ) pmetricotlp.ExportRequest {
0 commit comments