internal/httputil: add HTTPStatsRoundTripper to DefaultClient (#828)

This commit is contained in:
Cuong Manh Le 2020-06-09 01:34:32 +07:00 committed by GitHub
parent 7abe3a3b02
commit 9e711b4612
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 41 additions and 6 deletions

View file

@ -0,0 +1,25 @@
package httputil
import (
"context"
"net/http"
"net/http/httptest"
"testing"
"github.com/pomerium/pomerium/internal/telemetry/requestid"
)
func TestDefaultClient(t *testing.T) {
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
for _, header := range []string{"X-B3-Sampled", "X-B3-Spanid", "X-B3-Traceid", "X-Request-Id"} {
if _, ok := r.Header[header]; !ok {
t.Errorf("header %s is not set", header)
}
}
w.WriteHeader(http.StatusOK)
}))
defer ts.Close()
req, _ := http.NewRequest(http.MethodGet, ts.URL, nil)
req = req.WithContext(requestid.WithValue(context.Background(), "foo"))
_, _ = DefaultClient.Do(req)
}