refactor: use opentelemetry's standard api to track http status code (#2760)

This commit is contained in:
chen quan
2023-01-06 23:27:54 +08:00
committed by GitHub
parent e71c505e94
commit fc9b3ffdc1
3 changed files with 6 additions and 98 deletions

View File

@@ -148,46 +148,3 @@ func TestTraceResponseWriter(t *testing.T) {
})
}
}
func TestTraceHandler_traceResponseWriter(t *testing.T) {
writer := &traceResponseWriter{
w: httptest.NewRecorder(),
}
assert.NotPanics(t, func() {
writer.Hijack()
})
writer = &traceResponseWriter{
w: mockedHijackable{httptest.NewRecorder()},
}
assert.NotPanics(t, func() {
writer.Hijack()
})
writer = &traceResponseWriter{
w: httptest.NewRecorder(),
}
writer.WriteHeader(http.StatusBadRequest)
assert.NotNil(t, writer.Header())
writer = &traceResponseWriter{
w: httptest.NewRecorder(),
}
assert.NotPanics(t, func() {
writer.Flush()
})
writer = &traceResponseWriter{
w: mockedFlusher{httptest.NewRecorder()},
}
assert.NotPanics(t, func() {
writer.Flush()
})
}
type mockedFlusher struct {
*httptest.ResponseRecorder
}
func (m mockedFlusher) Flush() {
}