-
Notifications
You must be signed in to change notification settings - Fork 199
/
raterMock.go
71 lines (59 loc) · 2.29 KB
/
raterMock.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
package mock
// RaterMock -
type RaterMock struct {
GetRatingCalled func(string) uint32
GetStartRatingCalled func() uint32
GetSignedBlocksThresholdCalled func() float32
ComputeIncreaseProposerCalled func(shardId uint32, rating uint32) uint32
ComputeDecreaseProposerCalled func(shardId uint32, rating uint32, consecutiveMissedBlocks uint32) uint32
RevertIncreaseProposerCalled func(shardId uint32, rating uint32, nrReverts uint32) uint32
ComputeIncreaseValidatorCalled func(shardId uint32, rating uint32) uint32
ComputeDecreaseValidatorCalled func(shardId uint32, rating uint32) uint32
GetChanceCalled func(rating uint32) uint32
}
// GetRating -
func (rm *RaterMock) GetRating(pk string) uint32 {
return rm.GetRatingCalled(pk)
}
// GetStartRating -
func (rm *RaterMock) GetStartRating() uint32 {
if rm.GetStartRatingCalled != nil {
return rm.GetStartRatingCalled()
}
return 10
}
// GetSignedBlocksThreshold -
func (rm *RaterMock) GetSignedBlocksThreshold() float32 {
return rm.GetSignedBlocksThresholdCalled()
}
// ComputeIncreaseProposer -
func (rm *RaterMock) ComputeIncreaseProposer(shardId uint32, currentRating uint32) uint32 {
return rm.ComputeIncreaseProposerCalled(shardId, currentRating)
}
// ComputeDecreaseProposer -
func (rm *RaterMock) ComputeDecreaseProposer(shardId uint32, currentRating uint32, consecutiveMisses uint32) uint32 {
return rm.ComputeDecreaseProposerCalled(shardId, currentRating, consecutiveMisses)
}
// RevertIncreaseValidator -
func (rm *RaterMock) RevertIncreaseValidator(shardId uint32, currentRating uint32, nrReverts uint32) uint32 {
return rm.RevertIncreaseProposerCalled(shardId, currentRating, nrReverts)
}
// ComputeIncreaseValidator -
func (rm *RaterMock) ComputeIncreaseValidator(shardId uint32, currentRating uint32) uint32 {
return rm.ComputeIncreaseValidatorCalled(shardId, currentRating)
}
// ComputeDecreaseValidator -
func (rm *RaterMock) ComputeDecreaseValidator(shardId uint32, currentRating uint32) uint32 {
return rm.ComputeDecreaseValidatorCalled(shardId, currentRating)
}
// GetChance -
func (rm *RaterMock) GetChance(rating uint32) uint32 {
if rm.GetChanceCalled != nil {
return rm.GetChanceCalled(rating)
}
return 80
}
// IsInterfaceNil -
func (rm *RaterMock) IsInterfaceNil() bool {
return rm == nil
}