/
slotticker_test.go
111 lines (92 loc) · 2.37 KB
/
slotticker_test.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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
package slotutil
import (
"testing"
"time"
)
var _ = Ticker(&SlotTicker{})
func TestSlotTicker(t *testing.T) {
ticker := &SlotTicker{
c: make(chan uint64),
done: make(chan struct{}),
}
defer ticker.Done()
var sinceDuration time.Duration
since := func(time.Time) time.Duration {
return sinceDuration
}
var untilDuration time.Duration
until := func(time.Time) time.Duration {
return untilDuration
}
var tick chan time.Time
after := func(time.Duration) <-chan time.Time {
return tick
}
genesisTime := time.Date(2018, 1, 1, 0, 0, 0, 0, time.UTC)
secondsPerSlot := uint64(8)
// Test when the ticker starts immediately after genesis time.
sinceDuration = 1 * time.Second
untilDuration = 7 * time.Second
// Make this a buffered channel to prevent a deadlock since
// the other goroutine calls a function in this goroutine.
tick = make(chan time.Time, 2)
ticker.start(genesisTime, secondsPerSlot, since, until, after)
// Tick once.
tick <- time.Now()
slot := <-ticker.C()
if slot != 0 {
t.Fatalf("Expected %d, got %d", 0, slot)
}
// Tick twice.
tick <- time.Now()
slot = <-ticker.C()
if slot != 1 {
t.Fatalf("Expected %d, got %d", 1, slot)
}
// Tick thrice.
tick <- time.Now()
slot = <-ticker.C()
if slot != 2 {
t.Fatalf("Expected %d, got %d", 2, slot)
}
}
func TestSlotTickerGenesis(t *testing.T) {
ticker := &SlotTicker{
c: make(chan uint64),
done: make(chan struct{}),
}
defer ticker.Done()
var sinceDuration time.Duration
since := func(time.Time) time.Duration {
return sinceDuration
}
var untilDuration time.Duration
until := func(time.Time) time.Duration {
return untilDuration
}
var tick chan time.Time
after := func(time.Duration) <-chan time.Time {
return tick
}
genesisTime := time.Date(2018, 1, 1, 0, 0, 0, 0, time.UTC)
secondsPerSlot := uint64(8)
// Test when the ticker starts before genesis time.
sinceDuration = -1 * time.Second
untilDuration = 1 * time.Second
// Make this a buffered channel to prevent a deadlock since
// the other goroutine calls a function in this goroutine.
tick = make(chan time.Time, 2)
ticker.start(genesisTime, secondsPerSlot, since, until, after)
// Tick once.
tick <- time.Now()
slot := <-ticker.C()
if slot != 0 {
t.Fatalf("Expected %d, got %d", 0, slot)
}
// Tick twice.
tick <- time.Now()
slot = <-ticker.C()
if slot != 1 {
t.Fatalf("Expected %d, got %d", 1, slot)
}
}