forked from TrueCloudLab/frostfs-api-go
131 lines
2.4 KiB
Go
131 lines
2.4 KiB
Go
package container
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
"github.com/nspcc-dev/neofs-api-go/refs"
|
|
"github.com/nspcc-dev/neofs-crypto/test"
|
|
"github.com/nspcc-dev/netmap"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCID(t *testing.T) {
|
|
t.Run("check that marshal/unmarshal works like expected", func(t *testing.T) {
|
|
var (
|
|
c2 Container
|
|
cid2 CID
|
|
key = test.DecodeKey(0)
|
|
)
|
|
|
|
rules := netmap.PlacementRule{
|
|
ReplFactor: 2,
|
|
SFGroups: []netmap.SFGroup{
|
|
{
|
|
Selectors: []netmap.Select{
|
|
{Key: "Country", Count: 1},
|
|
{Key: netmap.NodesBucket, Count: 2},
|
|
},
|
|
Filters: []netmap.Filter{
|
|
{Key: "Country", F: netmap.FilterIn("USA")},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
owner, err := refs.NewOwnerID(&key.PublicKey)
|
|
require.NoError(t, err)
|
|
|
|
c1, err := New(10, owner, 0xDEADBEEF, rules)
|
|
require.NoError(t, err)
|
|
|
|
data, err := proto.Marshal(c1)
|
|
require.NoError(t, err)
|
|
|
|
require.NoError(t, c2.Unmarshal(data))
|
|
require.Equal(t, c1, &c2)
|
|
|
|
cid1, err := c1.ID()
|
|
require.NoError(t, err)
|
|
|
|
data, err = proto.Marshal(&cid1)
|
|
require.NoError(t, err)
|
|
require.NoError(t, cid2.Unmarshal(data))
|
|
|
|
require.Equal(t, cid1, cid2)
|
|
})
|
|
}
|
|
|
|
func TestPutRequestGettersSetters(t *testing.T) {
|
|
t.Run("owner", func(t *testing.T) {
|
|
owner := OwnerID{1, 2, 3}
|
|
m := new(PutRequest)
|
|
|
|
m.SetOwnerID(owner)
|
|
|
|
require.Equal(t, owner, m.GetOwnerID())
|
|
})
|
|
|
|
t.Run("capacity", func(t *testing.T) {
|
|
cp := uint64(3)
|
|
m := new(PutRequest)
|
|
|
|
m.SetCapacity(cp)
|
|
|
|
require.Equal(t, cp, m.GetCapacity())
|
|
})
|
|
|
|
t.Run("message ID", func(t *testing.T) {
|
|
id, err := refs.NewUUID()
|
|
require.NoError(t, err)
|
|
|
|
m := new(PutRequest)
|
|
|
|
m.SetMessageID(id)
|
|
|
|
require.Equal(t, id, m.GetMessageID())
|
|
})
|
|
|
|
t.Run("rules", func(t *testing.T) {
|
|
rules := netmap.PlacementRule{
|
|
ReplFactor: 1,
|
|
}
|
|
|
|
m := new(PutRequest)
|
|
|
|
m.SetRules(rules)
|
|
|
|
require.Equal(t, rules, m.GetRules())
|
|
})
|
|
|
|
t.Run("basic ACL", func(t *testing.T) {
|
|
bACL := uint32(5)
|
|
m := new(PutRequest)
|
|
|
|
m.SetBasicACL(bACL)
|
|
|
|
require.Equal(t, bACL, m.GetBasicACL())
|
|
})
|
|
}
|
|
|
|
func TestDeleteRequestGettersSetters(t *testing.T) {
|
|
t.Run("cid", func(t *testing.T) {
|
|
cid := CID{1, 2, 3}
|
|
m := new(DeleteRequest)
|
|
|
|
m.SetCID(cid)
|
|
|
|
require.Equal(t, cid, m.GetCID())
|
|
})
|
|
}
|
|
|
|
func TestGetRequestGettersSetters(t *testing.T) {
|
|
t.Run("cid", func(t *testing.T) {
|
|
cid := CID{1, 2, 3}
|
|
m := new(GetRequest)
|
|
|
|
m.SetCID(cid)
|
|
|
|
require.Equal(t, cid, m.GetCID())
|
|
})
|
|
}
|