fix(test): use DefaultTransport.RoundTrip and preserve query string in rewriteHostTransport

Address Copilot review feedback:
- Switch from http.DefaultClient.Do to http.DefaultTransport.RoundTrip
  to avoid unexpected redirect handling in test helper
- Preserve req.URL.RawQuery when rewriting URLs

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
I Putu Eddy Irawan 2026-03-05 20:49:57 +07:00
parent 310c644e09
commit 45f21bca10

View file

@ -883,12 +883,15 @@ func TestExaSearchProvider_MaxResultsCapping(t *testing.T) {
func rewriteHostTransport(target string) http.RoundTripper { func rewriteHostTransport(target string) http.RoundTripper {
return roundTripFunc(func(req *http.Request) (*http.Response, error) { return roundTripFunc(func(req *http.Request) (*http.Response, error) {
newURL := target + req.URL.Path newURL := target + req.URL.Path
if req.URL.RawQuery != "" {
newURL += "?" + req.URL.RawQuery
}
newReq, err := http.NewRequestWithContext(req.Context(), req.Method, newURL, req.Body) newReq, err := http.NewRequestWithContext(req.Context(), req.Method, newURL, req.Body)
if err != nil { if err != nil {
return nil, err return nil, err
} }
newReq.Header = req.Header newReq.Header = req.Header
return http.DefaultClient.Do(newReq) return http.DefaultTransport.RoundTrip(newReq)
}) })
} }