frostfs-node/pkg/innerring/processors/audit/scheduler_test.go

package audit_test

import (
	"testing"

	"github.com/nspcc-dev/neofs-node/pkg/innerring/processors/audit"
	cid "github.com/nspcc-dev/neofs-sdk-go/container/id"
	cidtest "github.com/nspcc-dev/neofs-sdk-go/container/id/test"
	"github.com/stretchr/testify/require"
)

func TestSelect(t *testing.T) {
	cids := generateContainers(10)

	t.Run("invalid input", func(t *testing.T) {
		require.Empty(t, audit.Select(cids, 0, 0, 0))
	})

	t.Run("even split", func(t *testing.T) {
		const irSize = 5 // every node takes two audit nodes

		m := hitMap(cids)

		for i := 0; i < irSize; i++ {
			s := audit.Select(cids, 0, uint64(i), irSize)
			require.Equal(t, len(cids)/irSize, len(s))

			for _, id := range s {
				n, ok := m[id.String()]
				require.True(t, ok)
				require.Equal(t, 0, n)
				m[id.String()] = 1
			}
		}

		require.True(t, allHit(m))
	})

	t.Run("odd split", func(t *testing.T) {
		const irSize = 3

		m := hitMap(cids)

		for i := 0; i < irSize; i++ {
			s := audit.Select(cids, 0, uint64(i), irSize)

			for _, id := range s {
				n, ok := m[id.String()]
				require.True(t, ok)
				require.Equal(t, 0, n)
				m[id.String()] = 1
			}
		}

		require.True(t, allHit(m))
	})

	t.Run("epoch shift", func(t *testing.T) {
		const irSize = 4

		m := hitMap(cids)

		for i := 0; i < irSize; i++ {
			s := audit.Select(cids, uint64(i), 0, irSize)

			for _, id := range s {
				n, ok := m[id.String()]
				require.True(t, ok)
				require.Equal(t, 0, n)
				m[id.String()] = 1
			}
		}

		require.True(t, allHit(m))
	})
}

func generateContainers(n int) []*cid.ID {
	result := make([]*cid.ID, 0, n)

	for i := 0; i < n; i++ {
		result = append(result, cidtest.GenerateID())
	}

	return result
}

func hitMap(ids []*cid.ID) map[string]int {
	result := make(map[string]int, len(ids))

	for _, id := range ids {
		result[id.String()] = 0
	}

	return result
}

func allHit(m map[string]int) bool {
	for _, v := range m {
		if v == 0 {
			return false
		}
	}

	return true
}