131 lines
3.6 KiB
Go
131 lines
3.6 KiB
Go
package innerring
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/client/container"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/event"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/logger/test"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestEpochTimer(t *testing.T) {
|
|
t.Parallel()
|
|
alphaState := &testAlphabetState{isAlphabet: true}
|
|
neh := &testNewEpochHandler{}
|
|
cnrStopper := &testContainerEstStopper{}
|
|
epochState := &testEpochState{
|
|
counter: 99,
|
|
duration: 10,
|
|
}
|
|
|
|
args := &epochTimerArgs{
|
|
l: test.NewLogger(t, true),
|
|
alphabetState: alphaState,
|
|
newEpochHandlers: []newEpochHandler{neh.Handle},
|
|
cnrWrapper: cnrStopper,
|
|
epoch: epochState,
|
|
stopEstimationDMul: 2,
|
|
stopEstimationDDiv: 10,
|
|
}
|
|
et := newEpochTimer(args)
|
|
err := et.Reset()
|
|
require.NoError(t, err, "failed to reset timer")
|
|
|
|
et.Tick(100)
|
|
require.Equal(t, 0, neh.called, "invalid new epoch handler calls")
|
|
require.Equal(t, 0, cnrStopper.called, "invalid container stop handler calls")
|
|
|
|
et.Tick(101)
|
|
require.Equal(t, 0, neh.called, "invalid new epoch handler calls")
|
|
require.Equal(t, 1, cnrStopper.called, "invalid container stop handler calls")
|
|
|
|
et.Tick(102)
|
|
require.Equal(t, 0, neh.called, "invalid new epoch handler calls")
|
|
require.Equal(t, 1, cnrStopper.called, "invalid container stop handler calls")
|
|
|
|
et.Tick(103)
|
|
require.Equal(t, 0, neh.called, "invalid new epoch handler calls")
|
|
require.Equal(t, 1, cnrStopper.called, "invalid container stop handler calls")
|
|
|
|
var h uint32
|
|
for h = 104; h < 109; h++ {
|
|
et.Tick(h)
|
|
require.Equal(t, 0, neh.called, "invalid new epoch handler calls")
|
|
require.Equal(t, 1, cnrStopper.called, "invalid container stop handler calls")
|
|
}
|
|
|
|
et.Tick(109)
|
|
require.Equal(t, 1, neh.called, "invalid new epoch handler calls")
|
|
require.Equal(t, 1, cnrStopper.called, "invalid container stop handler calls")
|
|
|
|
et.Tick(110)
|
|
require.Equal(t, 1, neh.called, "invalid new epoch handler calls")
|
|
require.Equal(t, 1, cnrStopper.called, "invalid container stop handler calls")
|
|
|
|
et.Tick(111)
|
|
require.Equal(t, 1, neh.called, "invalid new epoch handler calls")
|
|
require.Equal(t, 2, cnrStopper.called, "invalid container stop handler calls")
|
|
|
|
et.Tick(112)
|
|
require.Equal(t, 1, neh.called, "invalid new epoch handler calls")
|
|
require.Equal(t, 2, cnrStopper.called, "invalid container stop handler calls")
|
|
|
|
et.Tick(113)
|
|
require.Equal(t, 1, neh.called, "invalid new epoch handler calls")
|
|
require.Equal(t, 2, cnrStopper.called, "invalid container stop handler calls")
|
|
|
|
for h = 114; h < 119; h++ {
|
|
et.Tick(h)
|
|
require.Equal(t, 1, neh.called, "invalid new epoch handler calls")
|
|
require.Equal(t, 2, cnrStopper.called, "invalid container stop handler calls")
|
|
}
|
|
et.Tick(120)
|
|
require.Equal(t, 2, neh.called, "invalid new epoch handler calls")
|
|
require.Equal(t, 2, cnrStopper.called, "invalid container stop handler calls")
|
|
}
|
|
|
|
type testAlphabetState struct {
|
|
isAlphabet bool
|
|
}
|
|
|
|
func (s *testAlphabetState) IsAlphabet() bool {
|
|
return s.isAlphabet
|
|
}
|
|
|
|
type testNewEpochHandler struct {
|
|
called int
|
|
}
|
|
|
|
func (h *testNewEpochHandler) Handle() {
|
|
h.called++
|
|
}
|
|
|
|
type testContainerEstStopper struct {
|
|
called int
|
|
}
|
|
|
|
func (s *testContainerEstStopper) StopEstimation(_ container.StopEstimationPrm) error {
|
|
s.called++
|
|
return nil
|
|
}
|
|
|
|
type testEpochState struct {
|
|
counter uint64
|
|
duration uint64
|
|
}
|
|
|
|
func (s *testEpochState) EpochCounter() uint64 {
|
|
return s.counter
|
|
}
|
|
func (s *testEpochState) EpochDuration() uint64 {
|
|
return s.duration
|
|
}
|
|
|
|
type testEventHandler struct {
|
|
called int
|
|
}
|
|
|
|
func (h *testEventHandler) Handle(e event.Event) {
|
|
h.called++
|
|
}
|