-
Notifications
You must be signed in to change notification settings - Fork 199
/
stakingDataProviderStub.go
78 lines (67 loc) · 2.32 KB
/
stakingDataProviderStub.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package mock
import (
"math/big"
"github.com/ElrondNetwork/elrond-go/state"
)
// StakingDataProviderStub -
type StakingDataProviderStub struct {
CleanCalled func()
PrepareStakingDataCalled func(keys map[uint32][][]byte) error
GetTotalStakeEligibleNodesCalled func() *big.Int
GetTotalTopUpStakeEligibleNodesCalled func() *big.Int
GetNodeStakedTopUpCalled func(blsKey []byte) (*big.Int, error)
FillValidatorInfoCalled func(blsKey []byte) error
ComputeUnQualifiedNodesCalled func(validatorInfos map[uint32][]*state.ValidatorInfo) ([][]byte, map[string][][]byte, error)
}
// FillValidatorInfo -
func (sdps *StakingDataProviderStub) FillValidatorInfo(blsKey []byte) error {
if sdps.FillValidatorInfoCalled != nil {
return sdps.FillValidatorInfoCalled(blsKey)
}
return nil
}
// ComputeUnQualifiedNodes -
func (sdps *StakingDataProviderStub) ComputeUnQualifiedNodes(validatorInfos map[uint32][]*state.ValidatorInfo) ([][]byte, map[string][][]byte, error) {
if sdps.ComputeUnQualifiedNodesCalled != nil {
return sdps.ComputeUnQualifiedNodesCalled(validatorInfos)
}
return nil, nil, nil
}
// GetTotalStakeEligibleNodes -
func (sdps *StakingDataProviderStub) GetTotalStakeEligibleNodes() *big.Int {
if sdps.GetTotalStakeEligibleNodesCalled != nil {
return sdps.GetTotalStakeEligibleNodesCalled()
}
return big.NewInt(0)
}
// GetTotalTopUpStakeEligibleNodes -
func (sdps *StakingDataProviderStub) GetTotalTopUpStakeEligibleNodes() *big.Int {
if sdps.GetTotalTopUpStakeEligibleNodesCalled != nil {
return sdps.GetTotalTopUpStakeEligibleNodesCalled()
}
return big.NewInt(0)
}
// GetNodeStakedTopUp -
func (sdps *StakingDataProviderStub) GetNodeStakedTopUp(blsKey []byte) (*big.Int, error) {
if sdps.GetNodeStakedTopUpCalled != nil {
return sdps.GetNodeStakedTopUpCalled(blsKey)
}
return big.NewInt(0), nil
}
// PrepareStakingDataForRewards -
func (sdps *StakingDataProviderStub) PrepareStakingDataForRewards(keys map[uint32][][]byte) error {
if sdps.PrepareStakingDataCalled != nil {
return sdps.PrepareStakingDataCalled(keys)
}
return nil
}
// Clean -
func (sdps *StakingDataProviderStub) Clean() {
if sdps.CleanCalled != nil {
sdps.CleanCalled()
}
}
// IsInterfaceNil -
func (sdps *StakingDataProviderStub) IsInterfaceNil() bool {
return sdps == nil
}