@@ -99,7 +99,7 @@ func getOrFail(t *testing.T, url string, client *http.Client) []byte {
9999
100100func getCert (t * testing.T , c * tls.Conn ) []byte {
101101 t .Helper ()
102- if err := c .Handshake ( ); err != nil {
102+ if err := c .HandshakeContext ( context . Background () ); err != nil {
103103 t .Fatal ("cannot handshake" , err )
104104 }
105105 return c .ConnectionState ().PeerCertificates [0 ].Raw
@@ -318,14 +318,19 @@ func TestSimpleMitm(t *testing.T) {
318318 client , l := oneShotProxy (proxy )
319319 defer l .Close ()
320320
321- c , err := tls .Dial ("tcp" , https .Listener .Addr ().String (), & tls.Config {InsecureSkipVerify : true })
321+ ctx := context .Background ()
322+ c , err := (& tls.Dialer {
323+ Config : & tls.Config {InsecureSkipVerify : true },
324+ }).DialContext (ctx , "tcp" , https .Listener .Addr ().String ())
322325 if err != nil {
323326 t .Fatal ("cannot dial to tcp server" , err )
324327 }
325- origCert := getCert (t , c )
328+ tlsConn , ok := c .(* tls.Conn )
329+ assert .True (t , ok )
330+ origCert := getCert (t , tlsConn )
326331 _ = c .Close ()
327332
328- c2 , err := net .Dial ( "tcp" , l .Listener .Addr ().String ())
333+ c2 , err := ( & net.Dialer {}). DialContext ( ctx , "tcp" , l .Listener .Addr ().String ())
329334 if err != nil {
330335 t .Fatal ("dialing to proxy" , err )
331336 }
@@ -555,7 +560,9 @@ func TestHeadReqHasContentLength(t *testing.T) {
555560}
556561
557562func TestChunkedResponse (t * testing.T ) {
558- l , err := net .Listen ("tcp" , ":10234" )
563+ ctx := context .Background ()
564+
565+ l , err := (& net.ListenConfig {}).Listen (ctx , "tcp" , ":10234" )
559566 panicOnErr (err , "listen" )
560567 defer l .Close ()
561568 go func () {
@@ -579,10 +586,10 @@ func TestChunkedResponse(t *testing.T) {
579586 }
580587 }()
581588
582- c , err := net .Dial ( "tcp" , "localhost:10234" )
589+ c , err := ( & net.Dialer {}). DialContext ( ctx , "tcp" , "localhost:10234" )
583590 panicOnErr (err , "dial" )
584591 defer c .Close ()
585- req , _ := http .NewRequestWithContext (context . Background () , http .MethodGet , "/" , nil )
592+ req , _ := http .NewRequestWithContext (ctx , http .MethodGet , "/" , nil )
586593 _ = req .Write (c )
587594 resp , err := http .ReadResponse (bufio .NewReader (c ), req )
588595 panicOnErr (err , "readresp" )
@@ -609,7 +616,7 @@ func TestChunkedResponse(t *testing.T) {
609616 client , s := oneShotProxy (proxy )
610617 defer s .Close ()
611618
612- req , err = http .NewRequestWithContext (context . Background () , http .MethodGet , "http://localhost:10234/" , nil )
619+ req , err = http .NewRequestWithContext (ctx , http .MethodGet , "http://localhost:10234/" , nil )
613620 if err != nil {
614621 t .Fatal ("Cannot create request" , err )
615622 }
@@ -694,7 +701,7 @@ func TestGoproxyHijackConnect(t *testing.T) {
694701 client , l := oneShotProxy (proxy )
695702 defer l .Close ()
696703 proxyAddr := l .Listener .Addr ().String ()
697- conn , err := net .Dial ( "tcp" , proxyAddr )
704+ conn , err := ( & net.Dialer {}). DialContext ( context . Background (), "tcp" , proxyAddr )
698705 panicOnErr (err , "conn " + proxyAddr )
699706 buf := bufio .NewReader (conn )
700707 writeConnect (conn )
0 commit comments