@@ -108,11 +108,23 @@ func TestHttpClientSpanStatusExtractor200(t *testing.T) {
108
108
u := codes .Code (0 )
109
109
span := testSpan {status : & u }
110
110
c .Extract (span , nil , nil , nil )
111
- if * span .status != codes .Unset {
112
- panic ("span status should be unset!" )
111
+ if * span .status != codes .Ok {
112
+ panic ("span status should be ok!" )
113
+ }
114
+ }
115
+ func TestHttpClientSpanStatusExtractor201 (t * testing.T ) {
116
+ c := HttpClientSpanStatusExtractor [any , any ]{
117
+ Getter : customizedNetHttpAttrsGetter {
118
+ code : 201 ,
119
+ },
120
+ }
121
+ u := codes .Code (0 )
122
+ span := testSpan {status : & u }
123
+ c .Extract (span , nil , nil , nil )
124
+ if * span .status != codes .Ok {
125
+ panic ("span status should be ok!" )
113
126
}
114
127
}
115
-
116
128
func TestHttpServerSpanStatusExtractor500 (t * testing.T ) {
117
129
c := HttpServerSpanStatusExtractor [any , any ]{
118
130
Getter : customizedNetHttpAttrsGetter {
@@ -150,7 +162,20 @@ func TestHttpServerSpanStatusExtractor200(t *testing.T) {
150
162
u := codes .Code (0 )
151
163
span := testSpan {status : & u }
152
164
c .Extract (span , nil , nil , nil )
153
- if * span .status != codes .Unset {
154
- panic ("span status should be unset!" )
165
+ if * span .status != codes .Ok {
166
+ panic ("span status should be ok!" )
167
+ }
168
+ }
169
+ func TestHttpServerSpanStatusExtractor201 (t * testing.T ) {
170
+ c := HttpClientSpanStatusExtractor [any , any ]{
171
+ Getter : customizedNetHttpAttrsGetter {
172
+ code : 201 ,
173
+ },
174
+ }
175
+ u := codes .Code (0 )
176
+ span := testSpan {status : & u }
177
+ c .Extract (span , nil , nil , nil )
178
+ if * span .status != codes .Ok {
179
+ panic ("span status should be ok!" )
155
180
}
156
181
}
0 commit comments