From 45f21bca10db1e39e259bdbbe05bdee40a3aaa90 Mon Sep 17 00:00:00 2001 From: I Putu Eddy Irawan Date: Thu, 5 Mar 2026 20:49:57 +0700 Subject: [PATCH] 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 --- pkg/tools/web_test.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pkg/tools/web_test.go b/pkg/tools/web_test.go index 3ca106112..b9382484a 100644 --- a/pkg/tools/web_test.go +++ b/pkg/tools/web_test.go @@ -883,12 +883,15 @@ func TestExaSearchProvider_MaxResultsCapping(t *testing.T) { func rewriteHostTransport(target string) http.RoundTripper { return roundTripFunc(func(req *http.Request) (*http.Response, error) { newURL := target + req.URL.Path + if req.URL.RawQuery != "" { + newURL += "?" + req.URL.RawQuery + } newReq, err := http.NewRequestWithContext(req.Context(), req.Method, newURL, req.Body) if err != nil { return nil, err } newReq.Header = req.Header - return http.DefaultClient.Do(newReq) + return http.DefaultTransport.RoundTrip(newReq) }) }