-
Notifications
You must be signed in to change notification settings - Fork 199
/
rounderMock.go
76 lines (61 loc) · 1.62 KB
/
rounderMock.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
package mock
import (
"time"
)
// RounderMock -
type RounderMock struct {
RoundIndex int64
IndexCalled func() int64
TimeDurationCalled func() time.Duration
TimeStampCalled func() time.Time
UpdateRoundCalled func(time.Time, time.Time)
RemainingTimeCalled func(startTime time.Time, maxTime time.Duration) time.Duration
BeforeGenesisCalled func() bool
}
// BeforeGenesis -
func (rndm *RounderMock) BeforeGenesis() bool {
if rndm.BeforeGenesisCalled != nil {
return rndm.BeforeGenesisCalled()
}
return false
}
// Index -
func (rndm *RounderMock) Index() int64 {
if rndm.IndexCalled != nil {
return rndm.IndexCalled()
}
return rndm.RoundIndex
}
// TimeDuration -
func (rndm *RounderMock) TimeDuration() time.Duration {
if rndm.TimeDurationCalled != nil {
return rndm.TimeDurationCalled()
}
return 4000 * time.Millisecond
}
// TimeStamp -
func (rndm *RounderMock) TimeStamp() time.Time {
if rndm.TimeStampCalled != nil {
return rndm.TimeStampCalled()
}
return time.Unix(0, 0)
}
// UpdateRound -
func (rndm *RounderMock) UpdateRound(genesisRoundTimeStamp time.Time, timeStamp time.Time) {
if rndm.UpdateRoundCalled != nil {
rndm.UpdateRoundCalled(genesisRoundTimeStamp, timeStamp)
return
}
rndm.RoundIndex++
}
// RemainingTime -
func (rndm *RounderMock) RemainingTime(startTime time.Time, maxTime time.Duration) time.Duration {
if rndm.RemainingTimeCalled != nil {
return rndm.RemainingTimeCalled(startTime, maxTime)
}
return 4000 * time.Millisecond
}
// IsInterfaceNil returns true if there is no value under the interface
func (rndm *RounderMock) IsInterfaceNil() bool {
return rndm == nil
}