mirror of
https://github.com/pomerium/pomerium.git
synced 2025-05-28 08:27:26 +02:00
internal/httputil: add HTTPStatsRoundTripper to DefaultClient (#828)
This commit is contained in:
parent
7abe3a3b02
commit
9e711b4612
2 changed files with 41 additions and 6 deletions
25
internal/httputil/client_test.go
Normal file
25
internal/httputil/client_test.go
Normal 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)
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue