frostfs-node/pkg/services/policer/check_test.go

package policer

import (
	"testing"

	netmaptest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/netmap/test"
	"github.com/stretchr/testify/require"
)

func TestNodeCache(t *testing.T) {
	cache := newNodeCache()
	node := netmaptest.NodeInfo()

	require.Negative(t, cache.processStatus(node))

	cache.SubmitSuccessfulReplication(node)
	require.Zero(t, cache.processStatus(node))

	cache.submitReplicaCandidate(node)
	require.Positive(t, cache.processStatus(node))

	cache.submitReplicaHolder(node)
	require.Zero(t, cache.processStatus(node))
}