frostfs-node/pkg/services/policer/check_test.go
Leonard Lyubich e6f8904040 [#1680] policer: Refactor tracking the processed nodes
Add clear methods with docs. Use the methods instead of direct map
and bool instructions.

Signed-off-by: Leonard Lyubich <ctulhurider@gmail.com>
2022-10-11 12:54:27 +03:00

24 lines
538 B
Go

package policer
import (
"testing"
netmaptest "github.com/nspcc-dev/neofs-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.Hash())
require.Zero(t, cache.processStatus(node))
cache.submitReplicaCandidate(node)
require.Positive(t, cache.processStatus(node))
cache.submitReplicaHolder(node)
require.Zero(t, cache.processStatus(node))
}