99 "io"
1010 "math/rand/v2"
1111 "mime/multipart"
12- "net"
1312 "net/http"
1413 "net/url"
1514 "os"
@@ -86,7 +85,6 @@ func GetBotClient(t *testing.T) client.Client {
8685 t .Helper ()
8786 rc := createBot (t , "test-bot" )
8887 client , err := client .NewClient (rc ,
89- client .WithHTTPClient (newHTTPClientForTestSite ()),
9088 client .SkipWarnOnInsecureTLS (),
9189 client .EnableStatistics ())
9290 require .NoError (t , err )
@@ -193,49 +191,14 @@ func getTestClient(t *testing.T, username string) (*z.RC, client.Client) {
193191
194192 apiClient , err := client .NewTestClient (rc ,
195193 client .SkipWarnOnInsecureTLS (),
196- client .EnableStatistics (),
197- client .WithHTTPClient (newHTTPClientForTestSite ()))
194+ client .EnableStatistics ())
198195 if err != nil {
199196 t .Fatalf ("Failed to create z.client: %v" , err )
200197 }
201198
202199 return rc , apiClient
203200}
204201
205- func newHTTPClientForTestSite () * http.Client {
206- const timeout = 30 * time .Second
207- const keepAlive = 30 * time .Second
208-
209- baseTransport , ok := http .DefaultTransport .(* http.Transport )
210- if ! ok {
211- panic ("default transport is not an *http.Transport" )
212- }
213-
214- transport := baseTransport .Clone ()
215- dialer := & net.Dialer {
216- Timeout : timeout ,
217- KeepAlive : keepAlive ,
218- }
219-
220- transport .DialContext = func (ctx context.Context , network , address string ) (net.Conn , error ) {
221- host , port , err := net .SplitHostPort (address )
222- if err != nil {
223- return nil , err
224- }
225- if host == "localhost" || host == "::1" {
226- host = "127.0.0.1"
227- }
228- return dialer .DialContext (ctx , network , net .JoinHostPort (host , port ))
229- }
230-
231- return & http.Client {
232- Transport : transport ,
233- Timeout : http .DefaultClient .Timeout ,
234- CheckRedirect : http .DefaultClient .CheckRedirect ,
235- Jar : http .DefaultClient .Jar ,
236- }
237- }
238-
239202type UserInfo struct {
240203 APIKey string `json:"api_key"`
241204 EMail string `json:"email"`
@@ -251,10 +214,7 @@ func fetchUserInfo(t *testing.T, username string) UserInfo {
251214 attemptDelay = 3 * time .Second
252215 )
253216
254- httpClient := newHTTPClientForTestSite ()
255- if httpClient == nil {
256- httpClient = http .DefaultClient
257- }
217+ httpClient := http .DefaultClient
258218
259219 for attempt := range numAttempts {
260220 info , done := getUserInfo (t , httpClient , username , attempt )
@@ -303,7 +263,7 @@ func getUserInfo(t *testing.T, client *http.Client, username string, attempt int
303263 t .Fatalf ("Empty API key received for user %s" , username )
304264 }
305265 if result .EMail != username {
306- t .Fatalf ("Unexpected email in API key Response : got %s, want %s" , result .EMail , username )
266+ t .Fatalf ("Unexpected email in API key response : got %s, want %s" , result .EMail , username )
307267 }
308268 return result , true
309269 }
@@ -436,9 +396,7 @@ func createBot(t *testing.T, botName string) *z.RC {
436396 req .SetBasicAuth (rc .Email , rc .APIKey )
437397 req .Header .Set ("Content-Type" , w .FormDataContentType ())
438398
439- httpClient := newHTTPClientForTestSite ()
440-
441- resp , err := httpClient .Do (req )
399+ resp , err := http .DefaultClient .Do (req )
442400 require .NoError (t , err )
443401 defer resp .Body .Close ()
444402
0 commit comments