72 lines
2.2 KiB
Go
72 lines
2.2 KiB
Go
package handler
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCopiesNumberPicker(t *testing.T) {
|
|
var locationConstraints = map[string][]uint32{}
|
|
locationConstraint1 := "one"
|
|
locationConstraint2 := "two"
|
|
locationConstraints[locationConstraint1] = []uint32{2, 3, 4}
|
|
|
|
config := &Config{
|
|
Policy: &placementPolicyMock{
|
|
copiesNumbers: locationConstraints,
|
|
defaultCopiesNumbers: []uint32{1},
|
|
},
|
|
}
|
|
h := handler{
|
|
cfg: config,
|
|
}
|
|
|
|
metadata := map[string]string{}
|
|
|
|
t.Run("pick default copies number", func(t *testing.T) {
|
|
metadata["somekey1"] = "5, 6, 7"
|
|
expectedCopiesNumbers := []uint32{1}
|
|
|
|
actualCopiesNumbers, err := h.pickCopiesNumbers(metadata, locationConstraint2)
|
|
require.NoError(t, err)
|
|
require.Equal(t, expectedCopiesNumbers, actualCopiesNumbers)
|
|
})
|
|
|
|
t.Run("pick copies number vector according to location constraint", func(t *testing.T) {
|
|
metadata["somekey2"] = "6, 7, 8"
|
|
expectedCopiesNumbers := []uint32{2, 3, 4}
|
|
|
|
actualCopiesNumbers, err := h.pickCopiesNumbers(metadata, locationConstraint1)
|
|
require.NoError(t, err)
|
|
require.Equal(t, expectedCopiesNumbers, actualCopiesNumbers)
|
|
})
|
|
|
|
t.Run("pick copies number from metadata", func(t *testing.T) {
|
|
metadata["frostfs-copies-number"] = "7, 8, 9"
|
|
expectedCopiesNumbers := []uint32{7, 8, 9}
|
|
|
|
actualCopiesNumbers, err := h.pickCopiesNumbers(metadata, locationConstraint2)
|
|
require.NoError(t, err)
|
|
require.Equal(t, expectedCopiesNumbers, actualCopiesNumbers)
|
|
})
|
|
|
|
t.Run("pick copies number from metadata with no space", func(t *testing.T) {
|
|
metadata["frostfs-copies-number"] = "7,8,9"
|
|
expectedCopiesNumbers := []uint32{7, 8, 9}
|
|
|
|
actualCopiesNumbers, err := h.pickCopiesNumbers(metadata, locationConstraint2)
|
|
require.NoError(t, err)
|
|
require.Equal(t, expectedCopiesNumbers, actualCopiesNumbers)
|
|
})
|
|
|
|
t.Run("pick copies number from metadata with trailing comma", func(t *testing.T) {
|
|
metadata["frostfs-copies-number"] = "11, 12, 13, "
|
|
expectedCopiesNumbers := []uint32{11, 12, 13}
|
|
|
|
actualCopiesNumbers, err := h.pickCopiesNumbers(metadata, locationConstraint2)
|
|
require.NoError(t, err)
|
|
require.Equal(t, expectedCopiesNumbers, actualCopiesNumbers)
|
|
})
|
|
}
|