forked from TrueCloudLab/frostfs-node
40 lines
511 B
Go
40 lines
511 B
Go
|
package netmap
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestNetMap_Nodes(t *testing.T) {
|
||
|
|
nm := New()
|
||
|
|
|
||
|
|
info1 := Info{}
|
||
|
|
info1.SetPublicKey([]byte{1, 2, 3})
|
||
|
|
|
||
|
|
info2 := Info{}
|
||
|
|
info2.SetPublicKey([]byte{4, 5, 6})
|
||
|
|
|
||
|
|
nodes := []Info{
|
||
|
|
info1,
|
||
|
|
info2,
|
||
|
|
}
|
||
|
|
|
||
|
|
nm.SetNodes(nodes)
|
||
|
|
|
||
|
|
require.Equal(t, nodes, nm.Nodes())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNetMap_Root(t *testing.T) {
|
||
|
|
nm := New()
|
||
|
|
|
||
|
|
bucket := &Bucket{
|
||
|
|
Key: "key",
|
||
|
|
Value: "value",
|
||
|
|
}
|
||
|
|
|
||
|
|
nm.SetRoot(bucket)
|
||
|
|
|
||
|
|
require.Equal(t, bucket, nm.Root())
|
||
|
|
}
|