test: fix test cases

Author: qianhh

eth/protocols/dbft/handler_test.go

@@ -24,14 +24,18 @@ func (t *testBC) BlockNumber() hexutil.Uint64 {
 	return t.height
 }
 
+func (t *testBC) IsAddressAllowed(addr common.Address) error {
+	return nil
+}
+
 func TestHandling(t *testing.T) {
 	var (
 		key, _         = crypto.GenerateKey()
 		addr           = crypto.PubkeyToAddress(key.PublicKey)
 		bc             = &testBC{height: 10}
-		s1             = New(bc, nil, func(u uint64, address common.Address) error { return nil })
-		s2             = New(bc, nil, func(u uint64, address common.Address) error { return nil })
-		s3             = New(bc, nil, func(u uint64, address common.Address) error { return nil })
+		s1             = New(bc, nil)
+		s2             = New(bc, nil)
+		s3             = New(bc, nil)
 		p1             = p2p.NewPeer(enode.ID{1}, "peer1", nil)
 		p2             = p2p.NewPeer(enode.ID{2}, "peer2", nil)
 		p3             = p2p.NewPeer(enode.ID{3}, "peer3", nil)

eth/protocols/dbft/pool_test.go

@@ -9,6 +9,7 @@ import (
 	"testing"
 
 	"github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/common/hexutil"
 	"github.com/ethereum/go-ethereum/crypto"
 	"github.com/stretchr/testify/require"
 )
@@ -140,6 +141,8 @@ func (c *testChain) IsAddressAllowed(u common.Address) error {
 }
 func (c *testChain) BlockHeight() uint64 { return c.height }
 
+func (c *testChain) BlockNumber() hexutil.Uint64 { return hexutil.Uint64(c.height) }
+
 func (c *testChain) goodMessage(t *testing.T, height uint64) *Message {
 	return someMessage(t, height, c.goodAddrs[0], c.goodKey)
 }

internal/ethapi/api_test.go

@@ -631,6 +631,18 @@ func (b testBackend) BloomStatus() (uint64, uint64) { panic("implement me") }
 func (b testBackend) ServiceFilter(ctx context.Context, session *bloombits.MatcherSession) {
 	panic("implement me")
 }
+func (b testBackend) IsExtensibleAllowed(blockNum uint64, addr common.Address) error {
+	return nil
+}
+func (b testBackend) GetValidatorsSortedByBlockNumber(blockNum uint64) ([]common.Address, error) {
+	return nil, nil
+}
+func (b testBackend) GetValidatorsSortedByState(state *state.StateDB, header *types.Header) ([]common.Address, error) {
+	return nil, nil
+}
+func (b testBackend) GetDKGIndex(blockNum uint64, validatorIndex int) (int, error) {
+	return 0, nil
+}
 
 func TestEstimateGas(t *testing.T) {
 	t.Parallel()

internal/ethapi/transaction_args_test.go

@@ -413,3 +413,15 @@ func (b *backendMock) SubscribeRemovedLogsEvent(ch chan<- core.RemovedLogsEvent)
 }
 
 func (b *backendMock) Engine() consensus.Engine { return nil }
+func (b *backendMock) IsExtensibleAllowed(blockNum uint64, addr common.Address) error {
+	return nil
+}
+func (b *backendMock) GetValidatorsSortedByBlockNumber(blockNum uint64) ([]common.Address, error) {
+	return nil, nil
+}
+func (b *backendMock) GetValidatorsSortedByState(state *state.StateDB, header *types.Header) ([]common.Address, error) {
+	return nil, nil
+}
+func (b *backendMock) GetDKGIndex(blockNum uint64, validatorIndex int) (int, error) {
+	return 0, nil
+}