frostfs-api-go/container/types_test.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())
	})
}