forked from TrueCloudLab/frostfs-node
f7a9e43165
In previous implementation handler was not called more the one time after Reset. It was caused by tick counter not being reset inside Reset method. Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
112 lines
2.1 KiB
Go
112 lines
2.1 KiB
Go
package timers_test
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neofs-node/pkg/innerring/timers"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func tickN(t *timers.BlockTimer, n uint32) {
|
|
for i := uint32(0); i < n; i++ {
|
|
t.Tick()
|
|
}
|
|
}
|
|
|
|
func TestBlockTimer(t *testing.T) {
|
|
blockDur := uint32(10)
|
|
baseCallCounter := uint32(0)
|
|
|
|
bt := timers.NewBlockTimer(timers.StaticBlockMeter(blockDur), func() {
|
|
baseCallCounter++
|
|
})
|
|
|
|
require.NoError(t, bt.Reset())
|
|
|
|
intervalNum := uint32(7)
|
|
|
|
tickN(bt, intervalNum*blockDur)
|
|
|
|
require.Equal(t, intervalNum, uint32(baseCallCounter))
|
|
|
|
// add half-interval handler
|
|
halfCallCounter := uint32(0)
|
|
|
|
bt.OnDelta(1, 2, func() {
|
|
halfCallCounter++
|
|
})
|
|
|
|
// add double interval handler
|
|
doubleCallCounter := uint32(0)
|
|
|
|
bt.OnDelta(2, 1, func() {
|
|
doubleCallCounter++
|
|
})
|
|
|
|
require.NoError(t, bt.Reset())
|
|
|
|
baseCallCounter = 0
|
|
intervalNum = 20
|
|
|
|
tickN(bt, intervalNum*blockDur)
|
|
|
|
require.Equal(t, intervalNum, uint32(halfCallCounter))
|
|
require.Equal(t, intervalNum, uint32(baseCallCounter))
|
|
require.Equal(t, intervalNum/2, uint32(doubleCallCounter))
|
|
}
|
|
|
|
func TestDeltaPulse(t *testing.T) {
|
|
blockDur := uint32(9)
|
|
baseCallCounter := uint32(0)
|
|
|
|
bt := timers.NewBlockTimer(timers.StaticBlockMeter(blockDur), func() {
|
|
baseCallCounter++
|
|
})
|
|
|
|
deltaCallCounter := uint32(0)
|
|
|
|
div := uint32(3)
|
|
|
|
bt.OnDelta(1, div, func() {
|
|
deltaCallCounter++
|
|
}, timers.WithPulse())
|
|
|
|
require.NoError(t, bt.Reset())
|
|
|
|
intervalNum := uint32(7)
|
|
|
|
tickN(bt, intervalNum*blockDur)
|
|
|
|
fmt.Println(baseCallCounter, deltaCallCounter)
|
|
require.Equal(t, intervalNum, uint32(baseCallCounter))
|
|
require.Equal(t, intervalNum*div, uint32(deltaCallCounter))
|
|
}
|
|
|
|
func TestDeltaReset(t *testing.T) {
|
|
blockDur := uint32(6)
|
|
baseCallCounter := 0
|
|
|
|
bt := timers.NewBlockTimer(timers.StaticBlockMeter(blockDur), func() {
|
|
baseCallCounter++
|
|
})
|
|
|
|
detlaCallCounter := 0
|
|
|
|
bt.OnDelta(1, 3, func() {
|
|
detlaCallCounter++
|
|
})
|
|
|
|
require.NoError(t, bt.Reset())
|
|
|
|
tickN(bt, 6)
|
|
|
|
require.Equal(t, 1, baseCallCounter)
|
|
require.Equal(t, 1, detlaCallCounter)
|
|
|
|
require.NoError(t, bt.Reset())
|
|
|
|
tickN(bt, 3)
|
|
|
|
require.Equal(t, 2, detlaCallCounter)
|
|
}
|