From 5122be34e70125465ec2e7872ea566cae15bfc6d Mon Sep 17 00:00:00 2001
From: Pavel Karpy <carpawell@nspcc.ru>
Date: Mon, 6 Jun 2022 17:00:22 +0300
Subject: [PATCH] [#1461] shard: Add lock tests

Signed-off-by: Pavel Karpy <carpawell@nspcc.ru>
---
 pkg/local_object_storage/shard/lock_test.go | 128 ++++++++++++++++++++
 1 file changed, 128 insertions(+)
 create mode 100644 pkg/local_object_storage/shard/lock_test.go

diff --git a/pkg/local_object_storage/shard/lock_test.go b/pkg/local_object_storage/shard/lock_test.go
new file mode 100644
index 000000000..2ac3c6ba8
--- /dev/null
+++ b/pkg/local_object_storage/shard/lock_test.go
@@ -0,0 +1,128 @@
+package shard_test
+
+import (
+	"context"
+	"path/filepath"
+	"testing"
+
+	objectcore "github.com/nspcc-dev/neofs-node/pkg/core/object"
+	"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/blobstor"
+	meta "github.com/nspcc-dev/neofs-node/pkg/local_object_storage/metabase"
+	"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/shard"
+	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"
+	"github.com/stretchr/testify/require"
+	"go.uber.org/zap"
+)
+
+func TestShard_Lock(t *testing.T) {
+	var sh *shard.Shard
+
+	rootPath := t.TempDir()
+	opts := []shard.Option{
+		shard.WithLogger(zap.NewNop()),
+		shard.WithBlobStorOptions(
+			[]blobstor.Option{
+				blobstor.WithRootPath(filepath.Join(rootPath, "blob")),
+				blobstor.WithBlobovniczaShallowWidth(2),
+				blobstor.WithBlobovniczaShallowDepth(2),
+			}...,
+		),
+		shard.WithMetaBaseOptions(
+			meta.WithPath(filepath.Join(rootPath, "meta")),
+		),
+		shard.WitDeletedLockCallback(func(_ context.Context, addresses []oid.Address) {
+			sh.HandleDeletedLocks(addresses)
+		}),
+	}
+
+	sh = shard.New(opts...)
+	require.NoError(t, sh.Open())
+	require.NoError(t, sh.Init())
+
+	t.Cleanup(func() {
+		releaseShard(sh, t)
+	})
+
+	cnr := cidtest.ID()
+	obj := generateObjectWithCID(t, cnr)
+	objID, _ := obj.ID()
+
+	lock := generateObjectWithCID(t, cnr)
+	lock.SetType(object.TypeLock)
+	lockID, _ := lock.ID()
+
+	// put the object
+
+	var putPrm shard.PutPrm
+	putPrm.WithObject(obj)
+
+	_, err := sh.Put(putPrm)
+	require.NoError(t, err)
+
+	// lock the object
+
+	err = sh.Lock(cnr, lockID, []oid.ID{objID})
+	require.NoError(t, err)
+
+	putPrm.WithObject(lock)
+	_, err = sh.Put(putPrm)
+	require.NoError(t, err)
+
+	t.Run("inhuming locked objects", func(t *testing.T) {
+		ts := generateObjectWithCID(t, cnr)
+
+		var inhumePrm shard.InhumePrm
+		inhumePrm.WithTarget(objectcore.AddressOf(ts), objectcore.AddressOf(obj))
+
+		_, err = sh.Inhume(inhumePrm)
+		require.ErrorAs(t, err, new(apistatus.ObjectLocked))
+
+		inhumePrm.MarkAsGarbage(objectcore.AddressOf(obj))
+		_, err = sh.Inhume(inhumePrm)
+		require.ErrorAs(t, err, new(apistatus.ObjectLocked))
+	})
+
+	t.Run("inhuming lock objects", func(t *testing.T) {
+		ts := generateObjectWithCID(t, cnr)
+
+		var inhumePrm shard.InhumePrm
+		inhumePrm.WithTarget(objectcore.AddressOf(ts), objectcore.AddressOf(lock))
+
+		_, err = sh.Inhume(inhumePrm)
+		require.Error(t, err)
+
+		inhumePrm.MarkAsGarbage(objectcore.AddressOf(lock))
+		_, err = sh.Inhume(inhumePrm)
+		require.Error(t, err)
+	})
+
+	t.Run("force objects inhuming", func(t *testing.T) {
+		var inhumePrm shard.InhumePrm
+		inhumePrm.MarkAsGarbage(objectcore.AddressOf(lock))
+		inhumePrm.ForceRemoval()
+
+		_, err = sh.Inhume(inhumePrm)
+		require.NoError(t, err)
+
+		// it should be possible to remove
+		// lock object now
+
+		inhumePrm = shard.InhumePrm{}
+		inhumePrm.MarkAsGarbage(objectcore.AddressOf(obj))
+
+		_, err = sh.Inhume(inhumePrm)
+		require.NoError(t, err)
+
+		// check that object has been removed
+
+		var getPrm shard.GetPrm
+		getPrm.WithAddress(objectcore.AddressOf(obj))
+
+		_, err = sh.Get(getPrm)
+		require.ErrorAs(t, err, new(apistatus.ObjectNotFound))
+	})
+
+}