frostfs-node/pkg/local_object_storage/metabase/lock_test.go

package meta_test

import (
	"testing"

	objectcore "github.com/nspcc-dev/neofs-node/pkg/core/object"
	meta "github.com/nspcc-dev/neofs-node/pkg/local_object_storage/metabase"
	apistatus "github.com/nspcc-dev/neofs-sdk-go/client/status"
	cidtest "github.com/nspcc-dev/neofs-sdk-go/container/id/test"
	"github.com/nspcc-dev/neofs-sdk-go/object"
	oid "github.com/nspcc-dev/neofs-sdk-go/object/id"
	oidtest "github.com/nspcc-dev/neofs-sdk-go/object/id/test"
	objecttest "github.com/nspcc-dev/neofs-sdk-go/object/test"
	"github.com/stretchr/testify/require"
)

func TestDB_Lock(t *testing.T) {
	cnr := cidtest.ID()
	db := newDB(t)

	t.Run("empty locked list", func(t *testing.T) {
		require.Panics(t, func() { _ = db.Lock(cnr, oid.ID{}, nil) })
		require.Panics(t, func() { _ = db.Lock(cnr, oid.ID{}, []oid.ID{}) })
	})

	t.Run("(ir)regular", func(t *testing.T) {
		for _, typ := range [...]object.Type{
			object.TypeTombstone,
			object.TypeStorageGroup,
			object.TypeLock,
			object.TypeRegular,
		} {
			obj := objecttest.Object()
			obj.SetType(typ)
			obj.SetContainerID(cnr)

			// save irregular object
			err := meta.Put(db, obj, nil)
			require.NoError(t, err, typ)

			var e apistatus.LockNonRegularObject

			id, _ := obj.ID()

			// try to lock it
			err = db.Lock(cnr, oidtest.ID(), []oid.ID{id})
			if typ == object.TypeRegular {
				require.NoError(t, err, typ)
			} else {
				require.ErrorAs(t, err, &e, typ)
			}
		}
	})

	t.Run("lock-unlock scenario", func(t *testing.T) {
		cnr := cidtest.ID()

		obj := generateObjectWithCID(t, cnr)

		var err error

		err = putBig(db, obj)
		require.NoError(t, err)

		lockID := oidtest.ID()

		id, _ := obj.ID()

		// lock the object
		err = db.Lock(cnr, lockID, []oid.ID{id})
		require.NoError(t, err)

		var tombAddr oid.Address
		tombAddr.SetContainer(cnr)
		tombAddr.SetObject(lockID)

		// try to inhume locked object using tombstone
		err = meta.Inhume(db, objectcore.AddressOf(obj), tombAddr)
		require.ErrorAs(t, err, new(apistatus.ObjectLocked))

		// free locked object
		var inhumePrm meta.InhumePrm
		inhumePrm.WithAddresses(tombAddr)
		inhumePrm.WithGCMark()

		_, err = db.Inhume(inhumePrm)
		require.NoError(t, err)

		var lockAddr oid.Address
		lockAddr.SetObject(lockID)
		lockAddr.SetContainer(cnr)

		err = db.FreeLockedBy([]oid.Address{lockAddr})
		require.NoError(t, err)

		inhumePrm.WithAddresses(tombAddr)
		inhumePrm.WithGCMark()

		// now we can inhume the object
		_, err = db.Inhume(inhumePrm)
		require.NoError(t, err)
	})
}