diff --git a/les/peer_test.go b/les/peer_test.go index 0ba9cbfd52..392528a448 100644 --- a/les/peer_test.go +++ b/les/peer_test.go @@ -5,6 +5,7 @@ import ( "testing" "github.com/ethereum/go-ethereum/common" + "github.com/ethereum/go-ethereum/common/mclock" "github.com/ethereum/go-ethereum/les/flowcontrol" "github.com/ethereum/go-ethereum/p2p" "github.com/ethereum/go-ethereum/p2p/enode" @@ -36,7 +37,7 @@ func TestPeerHandshakeSetAnnounceTypeToAnnounceTypeSignedForTrustedPeer(t *testi rw: &rwStub{ WriteHook: func(recvList keyValueList) { //checking that ulc sends to peer allowedRequests=onlyAnnounceRequests and announceType = announceTypeSigned - recv := recvList.decode() + recv, _ := recvList.decode() var reqType uint64 err := recv.get("announceType", &reqType) @@ -81,7 +82,7 @@ func TestPeerHandshakeAnnounceTypeSignedForTrustedPeersPeerNotInTrusted(t *testi rw: &rwStub{ WriteHook: func(recvList keyValueList) { //checking that ulc sends to peer allowedRequests=noRequests and announceType != announceTypeSigned - recv := recvList.decode() + recv, _ := recvList.decode() var reqType uint64 err := recv.get("announceType", &reqType) @@ -239,17 +240,11 @@ func TestPeerHandshakeClientReturnErrorOnUselessPeer(t *testing.T) { func generateLesServer() *LesServer { s := &LesServer{ - defParams: &flowcontrol.ServerParams{ + defParams: flowcontrol.ServerParams{ BufLimit: uint64(300000000), MinRecharge: uint64(50000), }, - fcManager: flowcontrol.NewClientManager(1, 2, 3), - fcCostStats: &requestCostStats{ - stats: make(map[uint64]*linReg, len(reqList)), - }, - } - for _, code := range reqList { - s.fcCostStats.stats[code] = &linReg{cnt: 100} + fcManager: flowcontrol.NewClientManager(nil, &mclock.System{}), } return s } diff --git a/les/ulc_test.go b/les/ulc_test.go index 3b95e6368e..69ea62059d 100644 --- a/les/ulc_test.go +++ b/les/ulc_test.go @@ -11,6 +11,7 @@ import ( "crypto/ecdsa" "math/big" + "github.com/ethereum/go-ethereum/common/mclock" "github.com/ethereum/go-ethereum/core" "github.com/ethereum/go-ethereum/crypto" "github.com/ethereum/go-ethereum/eth" @@ -217,7 +218,7 @@ func newFullPeerPair(t *testing.T, index int, numberOfblocks int, chainGen func( // newLightPeer creates node with light sync mode func newLightPeer(t *testing.T, ulcConfig *eth.ULCConfig) pairPeer { peers := newPeerSet() - dist := newRequestDistributor(peers, make(chan struct{})) + dist := newRequestDistributor(peers, make(chan struct{}), &mclock.System{}) rm := newRetrieveManager(peers, dist, nil) ldb := ethdb.NewMemDatabase()