frostfs-node/pkg/innerring/blocktimer_test.go

package innerring

import (
	"testing"

	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/client/container"
	"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
}