113 lines
2 KiB
Go
113 lines
2 KiB
Go
package audit_test
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"crypto/sha256"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neofs-api-go/pkg/container"
|
|
"github.com/nspcc-dev/neofs-node/pkg/innerring/processors/audit"
|
|
"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) []*container.ID {
|
|
var buf [sha256.Size]byte
|
|
result := make([]*container.ID, 0, n)
|
|
|
|
for i := 0; i < n; i++ {
|
|
_, _ = rand.Read(buf[:])
|
|
|
|
cid := container.NewID()
|
|
cid.SetSHA256(buf)
|
|
|
|
result = append(result, cid)
|
|
}
|
|
|
|
return result
|
|
}
|
|
|
|
func hitMap(ids []*container.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
|
|
}
|