218 lines
8.1 KiB
Go
218 lines
8.1 KiB
Go
package netmap
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neofs-api-go/v2/netmap"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestPlacementPolicy_GetPlacementVectors(t *testing.T) {
|
|
p := newPlacementPolicy(2,
|
|
[]*netmap.Replica{
|
|
newReplica(1, "SPB"),
|
|
newReplica(2, "Americas"),
|
|
},
|
|
[]*netmap.Selector{
|
|
newSelector("SPB", "City", netmap.Same, 1, "SPBSSD"),
|
|
newSelector("Americas", "City", netmap.Distinct, 2, "Americas"),
|
|
},
|
|
[]*netmap.Filter{
|
|
newFilter("SPBSSD", "", "", netmap.AND,
|
|
newFilter("", "Country", "RU", netmap.EQ),
|
|
newFilter("", "City", "St.Petersburg", netmap.EQ),
|
|
newFilter("", "SSD", "1", netmap.EQ)),
|
|
newFilter("Americas", "", "", netmap.OR,
|
|
newFilter("", "Continent", "NA", netmap.EQ),
|
|
newFilter("", "Continent", "SA", netmap.EQ)),
|
|
})
|
|
nodes := []netmap.NodeInfo{
|
|
nodeInfoFromAttributes("ID", "1", "Country", "RU", "City", "St.Petersburg", "SSD", "0"),
|
|
nodeInfoFromAttributes("ID", "2", "Country", "RU", "City", "St.Petersburg", "SSD", "1"),
|
|
nodeInfoFromAttributes("ID", "3", "Country", "RU", "City", "Moscow", "SSD", "1"),
|
|
nodeInfoFromAttributes("ID", "4", "Country", "RU", "City", "Moscow", "SSD", "1"),
|
|
nodeInfoFromAttributes("ID", "5", "Country", "RU", "City", "St.Petersburg", "SSD", "1"),
|
|
nodeInfoFromAttributes("ID", "6", "Continent", "NA", "City", "NewYork"),
|
|
nodeInfoFromAttributes("ID", "7", "Continent", "AF", "City", "Cairo"),
|
|
nodeInfoFromAttributes("ID", "8", "Continent", "AF", "City", "Cairo"),
|
|
nodeInfoFromAttributes("ID", "9", "Continent", "SA", "City", "Lima"),
|
|
nodeInfoFromAttributes("ID", "10", "Continent", "AF", "City", "Cairo"),
|
|
nodeInfoFromAttributes("ID", "11", "Continent", "NA", "City", "NewYork"),
|
|
nodeInfoFromAttributes("ID", "12", "Continent", "NA", "City", "LosAngeles"),
|
|
nodeInfoFromAttributes("ID", "13", "Continent", "SA", "City", "Lima"),
|
|
}
|
|
|
|
nm, err := NewNetmap(NodesFromV2(nodes))
|
|
require.NoError(t, err)
|
|
v, err := nm.GetContainerNodes(p, nil)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 2, len(v.Replicas()))
|
|
require.Equal(t, 6, len(v.Flatten()))
|
|
|
|
require.Equal(t, 2, len(v.Replicas()[0]))
|
|
ids := map[string]struct{}{}
|
|
for _, ni := range v.Replicas()[0] {
|
|
require.Equal(t, "RU", ni.Attribute("Country"))
|
|
require.Equal(t, "St.Petersburg", ni.Attribute("City"))
|
|
require.Equal(t, "1", ni.Attribute("SSD"))
|
|
ids[ni.Attribute("ID")] = struct{}{}
|
|
}
|
|
require.Equal(t, len(v.Replicas()[0]), len(ids), "not all nodes we distinct")
|
|
|
|
require.Equal(t, 4, len(v.Replicas()[1])) // 2 cities * 2 HRWB
|
|
ids = map[string]struct{}{}
|
|
for _, ni := range v.Replicas()[1] {
|
|
require.Contains(t, []string{"NA", "SA"}, ni.Attribute("Continent"))
|
|
ids[ni.Attribute("ID")] = struct{}{}
|
|
}
|
|
require.Equal(t, len(v.Replicas()[1]), len(ids), "not all nodes we distinct")
|
|
}
|
|
|
|
func TestPlacementPolicy_ProcessSelectors(t *testing.T) {
|
|
p := newPlacementPolicy(2, nil,
|
|
[]*netmap.Selector{
|
|
newSelector("SameRU", "City", netmap.Same, 2, "FromRU"),
|
|
newSelector("DistinctRU", "City", netmap.Distinct, 2, "FromRU"),
|
|
newSelector("Good", "Country", netmap.Distinct, 2, "Good"),
|
|
newSelector("Main", "Country", netmap.Distinct, 3, "*"),
|
|
},
|
|
[]*netmap.Filter{
|
|
newFilter("FromRU", "Country", "Russia", netmap.EQ),
|
|
newFilter("Good", "Rating", "4", netmap.GE),
|
|
})
|
|
nodes := []netmap.NodeInfo{
|
|
nodeInfoFromAttributes("Country", "Russia", "Rating", "1", "City", "SPB"),
|
|
nodeInfoFromAttributes("Country", "Germany", "Rating", "5", "City", "Berlin"),
|
|
nodeInfoFromAttributes("Country", "Russia", "Rating", "6", "City", "Moscow"),
|
|
nodeInfoFromAttributes("Country", "France", "Rating", "4", "City", "Paris"),
|
|
nodeInfoFromAttributes("Country", "France", "Rating", "1", "City", "Lyon"),
|
|
nodeInfoFromAttributes("Country", "Russia", "Rating", "5", "City", "SPB"),
|
|
nodeInfoFromAttributes("Country", "Russia", "Rating", "7", "City", "Moscow"),
|
|
nodeInfoFromAttributes("Country", "Germany", "Rating", "3", "City", "Darmstadt"),
|
|
nodeInfoFromAttributes("Country", "Germany", "Rating", "7", "City", "Frankfurt"),
|
|
nodeInfoFromAttributes("Country", "Russia", "Rating", "9", "City", "SPB"),
|
|
nodeInfoFromAttributes("Country", "Russia", "Rating", "9", "City", "SPB"),
|
|
}
|
|
|
|
nm, err := NewNetmap(NodesFromV2(nodes))
|
|
require.NoError(t, err)
|
|
c := NewContext(nm)
|
|
require.NoError(t, c.processFilters(p))
|
|
require.NoError(t, c.processSelectors(p))
|
|
|
|
for _, s := range p.GetSelectors() {
|
|
sel := c.Selections[s.GetName()]
|
|
s := c.Selectors[s.GetName()]
|
|
bucketCount, nodesInBucket := GetNodesCount(p, s)
|
|
targ := fmt.Sprintf("selector '%s'", s.GetName())
|
|
require.Equal(t, bucketCount, len(sel), targ)
|
|
for _, res := range sel {
|
|
require.Equal(t, nodesInBucket, len(res), targ)
|
|
for j := range res {
|
|
require.True(t, c.applyFilter(s.GetFilter(), res[j]), targ)
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
func TestPlacementPolicy_ProcessSelectorsHRW(t *testing.T) {
|
|
p := newPlacementPolicy(1, nil,
|
|
[]*netmap.Selector{
|
|
newSelector("Main", "Country", netmap.Distinct, 3, "*"),
|
|
}, nil)
|
|
|
|
// bucket weight order: RU > DE > FR
|
|
nodes := []netmap.NodeInfo{
|
|
nodeInfoFromAttributes("Country", "Germany", PriceAttr, "2", CapacityAttr, "10000"),
|
|
nodeInfoFromAttributes("Country", "Germany", PriceAttr, "4", CapacityAttr, "1"),
|
|
nodeInfoFromAttributes("Country", "France", PriceAttr, "3", CapacityAttr, "10"),
|
|
nodeInfoFromAttributes("Country", "Russia", PriceAttr, "2", CapacityAttr, "10000"),
|
|
nodeInfoFromAttributes("Country", "Russia", PriceAttr, "1", CapacityAttr, "10000"),
|
|
nodeInfoFromAttributes("Country", "Russia", CapacityAttr, "10000"),
|
|
nodeInfoFromAttributes("Country", "France", PriceAttr, "100", CapacityAttr, "1"),
|
|
nodeInfoFromAttributes("Country", "France", PriceAttr, "7", CapacityAttr, "10000"),
|
|
nodeInfoFromAttributes("Country", "Russia", PriceAttr, "2", CapacityAttr, "1"),
|
|
}
|
|
nm, err := NewNetmap(NodesFromV2(nodes))
|
|
require.NoError(t, err)
|
|
c := NewContext(nm)
|
|
c.setPivot([]byte("containerID"))
|
|
c.weightFunc = newWeightFunc(newMaxNorm(10000), newReverseMinNorm(1))
|
|
c.aggregator = newMaxAgg
|
|
require.NoError(t, c.processFilters(p))
|
|
require.NoError(t, c.processSelectors(p))
|
|
|
|
cnt := c.Selections["Main"]
|
|
expected := []Nodes{
|
|
{{Index: 4, Capacity: 10000, Price: 1}}, // best RU
|
|
{{Index: 0, Capacity: 10000, Price: 2}}, // best DE
|
|
{{Index: 7, Capacity: 10000, Price: 7}}, // best FR
|
|
}
|
|
require.Equal(t, len(expected), len(cnt))
|
|
for i := range expected {
|
|
require.Equal(t, len(expected[i]), len(cnt[i]))
|
|
require.Equal(t, expected[i][0].Index, cnt[i][0].Index)
|
|
require.Equal(t, expected[i][0].Capacity, cnt[i][0].Capacity)
|
|
require.Equal(t, expected[i][0].Price, cnt[i][0].Price)
|
|
}
|
|
|
|
res, err := nm.GetPlacementVectors(containerNodes(cnt), []byte("objectID"))
|
|
require.NoError(t, err)
|
|
require.Equal(t, res, cnt)
|
|
}
|
|
|
|
func TestPlacementPolicy_ProcessSelectorsInvalid(t *testing.T) {
|
|
testCases := []struct {
|
|
name string
|
|
p *netmap.PlacementPolicy
|
|
err error
|
|
}{
|
|
{
|
|
"MissingSelector",
|
|
newPlacementPolicy(2, nil,
|
|
[]*netmap.Selector{nil},
|
|
[]*netmap.Filter{}),
|
|
ErrMissingField,
|
|
},
|
|
{
|
|
"InvalidFilterReference",
|
|
newPlacementPolicy(1, nil,
|
|
[]*netmap.Selector{newSelector("MyStore", "Country", netmap.Distinct, 1, "FromNL")},
|
|
[]*netmap.Filter{newFilter("FromRU", "Country", "Russia", netmap.EQ)}),
|
|
ErrFilterNotFound,
|
|
},
|
|
{
|
|
"NotEnoughNodes (backup factor)",
|
|
newPlacementPolicy(2, nil,
|
|
[]*netmap.Selector{newSelector("MyStore", "Country", netmap.Distinct, 1, "FromRU")},
|
|
[]*netmap.Filter{newFilter("FromRU", "Country", "Russia", netmap.EQ)}),
|
|
ErrNotEnoughNodes,
|
|
},
|
|
{
|
|
"NotEnoughNodes (buckets)",
|
|
newPlacementPolicy(1, nil,
|
|
[]*netmap.Selector{newSelector("MyStore", "Country", netmap.Distinct, 2, "FromRU")},
|
|
[]*netmap.Filter{newFilter("FromRU", "Country", "Russia", netmap.EQ)}),
|
|
ErrNotEnoughNodes,
|
|
},
|
|
}
|
|
nodes := []netmap.NodeInfo{
|
|
nodeInfoFromAttributes("Country", "Russia"),
|
|
nodeInfoFromAttributes("Country", "Germany"),
|
|
nodeInfoFromAttributes(),
|
|
}
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
nm, err := NewNetmap(NodesFromV2(nodes))
|
|
require.NoError(t, err)
|
|
c := NewContext(nm)
|
|
require.NoError(t, c.processFilters(tc.p))
|
|
|
|
err = c.processSelectors(tc.p)
|
|
require.True(t, errors.Is(err, tc.err), "got: %v", err)
|
|
})
|
|
}
|
|
}
|