prysm-pulse/time/slots/slotticker_test.go
Raul Jordan d077483577
Add V3 Suffix to All Prysm Packages (#11083)
* v3 import renamings

* tidy

* fmt

* rev

* Update beacon-chain/core/epoch/precompute/reward_penalty_test.go

* Update beacon-chain/core/helpers/validators_test.go

* Update beacon-chain/db/alias.go

* Update beacon-chain/db/alias.go

* Update beacon-chain/db/alias.go

* Update beacon-chain/db/iface/BUILD.bazel

* Update beacon-chain/db/kv/kv.go

* Update beacon-chain/db/kv/state.go

* Update beacon-chain/rpc/prysm/v1alpha1/validator/attester_test.go

* Update beacon-chain/rpc/prysm/v1alpha1/validator/attester_test.go

* Update beacon-chain/sync/initial-sync/service.go

* fix deps

* fix bad replacements

* fix bad replacements

* change back

* gohashtree version

* fix deps

Co-authored-by: Nishant Das <nishdas93@gmail.com>
Co-authored-by: Potuz <potuz@prysmaticlabs.com>
2022-08-16 12:20:13 +00:00

139 lines
3.0 KiB
Go

package slots
import (
"testing"
"time"
types "github.com/prysmaticlabs/prysm/v3/consensus-types/primitives"
)
var _ Ticker = (*SlotTicker)(nil)
func TestSlotTicker(t *testing.T) {
ticker := &SlotTicker{
c: make(chan types.Slot),
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 types.Slot),
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 := NewSlotTickerWithOffset(genesisTime, offset, secondsPerSlot)
normalTicker := NewSlotTicker(genesisTime, secondsPerSlot)
firstTicked := 0
for {
select {
case <-offsetTicker.C():
if firstTicked != 1 {
t.Fatal("Expected other ticker to tick first")
}
return
case <-normalTicker.C():
if firstTicked != 0 {
t.Fatal("Expected normal ticker to tick first")
}
firstTicked = 1
}
}
}