test

pkg/services/rpcsrv/server_test.go

@@ -2,13 +2,15 @@ package rpcsrv
 
 import (
 	"bytes"
+	"context"
 	"encoding/base64"
 	"encoding/binary"
 	"encoding/json"
 	"fmt"
 	gio "io"
 	"math"
 	"math/big"
+	"net"
 	"net/http"
 	"net/http/httptest"
 	"reflect"
@@ -3459,7 +3461,14 @@ func checkErrGetBatchResult(t *testing.T, body []byte, expectingFail bool, expec
 }
 
 func doRPCCallOverWS(rpcCall string, url string, t *testing.T) []byte {
-	dialer := websocket.Dialer{HandshakeTimeout: time.Second}
+	dialer := websocket.Dialer{
+		HandshakeTimeout: time.Second,
+		NetDial: func(network, addr string) (net.Conn, error) {
+			return net.Dial("tcp4", addr)
+		}, NetDialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
+			netDialer := &net.Dialer{}
+			return netDialer.DialContext(ctx, "tcp4", addr)
+		}}
 	url = "ws" + strings.TrimPrefix(url, "http")
 	c, r, err := dialer.Dial(url+"/ws", nil)
 	require.NoError(t, err)
@@ -3476,7 +3485,12 @@ func doRPCCallOverWS(rpcCall string, url string, t *testing.T) []byte {
 }
 
 func doRPCCallOverHTTP(rpcCall string, url string, t *testing.T) []byte {
-	cl := http.Client{Timeout: time.Second}
+	tr := http.DefaultTransport.(*http.Transport).Clone()
+	tr.DialContext = func(ctx context.Context, network, addr string) (net.Conn, error) {
+		netDialer := &net.Dialer{}
+		return netDialer.DialContext(ctx, "tcp4", addr)
+	}
+	cl := http.Client{Timeout: time.Second, Transport: tr}
 	resp, err := cl.Post(url, "application/json", strings.NewReader(rpcCall))
 	require.NoErrorf(t, err, "could not make a POST request")
 	body, err := gio.ReadAll(resp.Body)