Aleksey Shakhmatov
a90c4cd7fa
- Logging: structured slog output with method, URL, status, duration - DefaultHeaders/UserAgent: inject headers without overwriting existing - BearerAuth/BasicAuth: per-request token resolution and static credentials - Recovery: catches panics in the RoundTripper chain
108 lines
2.9 KiB
Go
108 lines
2.9 KiB
Go
package middleware_test
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
|
|
"git.codelab.vc/pkg/httpx/middleware"
|
|
)
|
|
|
|
func TestDefaultHeaders(t *testing.T) {
|
|
t.Run("adds headers without overwriting existing", func(t *testing.T) {
|
|
defaults := http.Header{
|
|
"X-Custom": {"default-value"},
|
|
"X-Untouched": {"from-middleware"},
|
|
}
|
|
|
|
var captured http.Header
|
|
base := mockTransport(func(req *http.Request) (*http.Response, error) {
|
|
captured = req.Header.Clone()
|
|
return okResponse(), nil
|
|
})
|
|
|
|
transport := middleware.DefaultHeaders(defaults)(base)
|
|
|
|
req, _ := http.NewRequest(http.MethodGet, "http://example.com", nil)
|
|
req.Header.Set("X-Custom", "request-value")
|
|
|
|
_, err := transport.RoundTrip(req)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
|
|
if got := captured.Get("X-Custom"); got != "request-value" {
|
|
t.Errorf("X-Custom = %q, want %q (should not overwrite)", got, "request-value")
|
|
}
|
|
if got := captured.Get("X-Untouched"); got != "from-middleware" {
|
|
t.Errorf("X-Untouched = %q, want %q", got, "from-middleware")
|
|
}
|
|
})
|
|
|
|
t.Run("adds headers when absent", func(t *testing.T) {
|
|
defaults := http.Header{
|
|
"Accept": {"application/json"},
|
|
}
|
|
|
|
var captured http.Header
|
|
base := mockTransport(func(req *http.Request) (*http.Response, error) {
|
|
captured = req.Header.Clone()
|
|
return okResponse(), nil
|
|
})
|
|
|
|
transport := middleware.DefaultHeaders(defaults)(base)
|
|
req, _ := http.NewRequest(http.MethodGet, "http://example.com", nil)
|
|
|
|
_, err := transport.RoundTrip(req)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
|
|
if got := captured.Get("Accept"); got != "application/json" {
|
|
t.Errorf("Accept = %q, want %q", got, "application/json")
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestUserAgent(t *testing.T) {
|
|
t.Run("sets user agent header", func(t *testing.T) {
|
|
var captured http.Header
|
|
base := mockTransport(func(req *http.Request) (*http.Response, error) {
|
|
captured = req.Header.Clone()
|
|
return okResponse(), nil
|
|
})
|
|
|
|
transport := middleware.UserAgent("httpx/1.0")(base)
|
|
req, _ := http.NewRequest(http.MethodGet, "http://example.com", nil)
|
|
|
|
_, err := transport.RoundTrip(req)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
|
|
if got := captured.Get("User-Agent"); got != "httpx/1.0" {
|
|
t.Errorf("User-Agent = %q, want %q", got, "httpx/1.0")
|
|
}
|
|
})
|
|
|
|
t.Run("does not overwrite existing user agent", func(t *testing.T) {
|
|
var captured http.Header
|
|
base := mockTransport(func(req *http.Request) (*http.Response, error) {
|
|
captured = req.Header.Clone()
|
|
return okResponse(), nil
|
|
})
|
|
|
|
transport := middleware.UserAgent("httpx/1.0")(base)
|
|
req, _ := http.NewRequest(http.MethodGet, "http://example.com", nil)
|
|
req.Header.Set("User-Agent", "custom-agent")
|
|
|
|
_, err := transport.RoundTrip(req)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
|
|
if got := captured.Get("User-Agent"); got != "custom-agent" {
|
|
t.Errorf("User-Agent = %q, want %q", got, "custom-agent")
|
|
}
|
|
})
|
|
}
|