package shard_test import ( "context" "testing" "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/object" "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/internal/testutil" "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/shard" apistatus "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/client/status" cidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id/test" "github.com/stretchr/testify/require" ) func TestShard_Delete(t *testing.T) { t.Parallel() t.Run("without write cache", func(t *testing.T) { t.Parallel() testShardDelete(t, false) }) t.Run("with write cache", func(t *testing.T) { t.Parallel() testShardDelete(t, true) }) } func testShardDelete(t *testing.T, hasWriteCache bool) { sh := newShard(t, hasWriteCache) defer releaseShard(sh, t) cnr := cidtest.ID() obj := testutil.GenerateObjectWithCID(cnr) testutil.AddAttribute(obj, "foo", "bar") var putPrm shard.PutPrm var getPrm shard.GetPrm t.Run("big object", func(t *testing.T) { testutil.AddPayload(obj, 1<<20) putPrm.SetObject(obj) getPrm.SetAddress(object.AddressOf(obj)) var delPrm shard.DeletePrm delPrm.SetAddresses(object.AddressOf(obj)) _, err := sh.Put(context.Background(), putPrm) require.NoError(t, err) _, err = testGet(t, sh, getPrm, hasWriteCache) require.NoError(t, err) _, err = sh.Delete(context.TODO(), delPrm) require.NoError(t, err) _, err = sh.Get(context.Background(), getPrm) require.ErrorAs(t, err, new(apistatus.ObjectNotFound)) }) t.Run("small object", func(t *testing.T) { obj := testutil.GenerateObjectWithCID(cnr) testutil.AddAttribute(obj, "foo", "bar") testutil.AddPayload(obj, 1<<5) putPrm.SetObject(obj) getPrm.SetAddress(object.AddressOf(obj)) var delPrm shard.DeletePrm delPrm.SetAddresses(object.AddressOf(obj)) _, err := sh.Put(context.Background(), putPrm) require.NoError(t, err) _, err = sh.Get(context.Background(), getPrm) require.NoError(t, err) _, err = sh.Delete(context.Background(), delPrm) require.NoError(t, err) _, err = sh.Get(context.Background(), getPrm) require.ErrorAs(t, err, new(apistatus.ObjectNotFound)) }) }