prysm-pulse/shared/slotutil/slotticker_test.go
Ivan Martinez 5afee4ba3a
Use SlotTickerWithOffset for StreamIndexedAttestations (#5999)
* Change streamindexed to slot ticker with offset
* Make 2/3rds
* Add test
* Merge branch 'master' of github.com:prysmaticlabs/prysm into ticker-offset
* Add check for offset
* Merge branch 'master' of github.com:prysmaticlabs/prysm into ticker-offset
* Fix test
* Merge refs/heads/master into ticker-offset
2020-05-26 19:34:09 +00:00

140 lines
3.0 KiB
Go

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)
}
}
func TestGetSlotTickerWithOffset_OK(t *testing.T) {
genesisTime := time.Now()
secondsPerSlot := uint64(4)
offset := time.Duration(secondsPerSlot/2) * time.Second
offsetTicker := GetSlotTickerWithOffset(genesisTime, offset, secondsPerSlot)
normalTicker := GetSlotTicker(genesisTime, secondsPerSlot)
firstTicked := 0
for {
select {
case _ = <-offsetTicker.C():
if firstTicked != 1 {
t.Fatal("Expected other ticker to tick first")
}
firstTicked = 2
return
case _ = <-normalTicker.C():
if firstTicked != 0 {
t.Fatal("Expected normal ticker to tick first")
}
firstTicked = 1
break
}
}
}