Add writecache inhume tests #1028

Merged
fyrchik merged 2 commits from dstepanov-yadro/frostfs-node:fix/writecache_consistency into master 2024-09-04 19:51:07 +00:00
3 changed files with 129 additions and 3 deletions

2
go.mod
View file

@ -42,7 +42,7 @@ require (
golang.org/x/sys v0.16.0
golang.org/x/term v0.16.0
google.golang.org/grpc v1.61.0
google.golang.org/protobuf v1.32.0
google.golang.org/protobuf v1.33.0
gopkg.in/yaml.v3 v3.0.1
)

4
go.sum
View file

@ -422,8 +422,8 @@ google.golang.org/protobuf v1.23.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2
google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
google.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=
google.golang.org/protobuf v1.27.1/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=
google.golang.org/protobuf v1.32.0 h1:pPC6BG5ex8PDFnkbrGU3EixyhKcQ2aDuBS36lqK/C7I=
google.golang.org/protobuf v1.32.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
google.golang.org/protobuf v1.33.0 h1:uNO2rsAINq/JlFpSdYEKIZ0uKD/R9cpdv0T+yoGwGmI=
google.golang.org/protobuf v1.33.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=

View file

@ -7,10 +7,13 @@ import (
objectV2 "git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/object"
objectCore "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/object"
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/common"
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/internal/testutil"
meta "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/metabase"
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/writecache"
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util"
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/client"
apistatus "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/client/status"
cidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id/test"
objectSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
@ -167,3 +170,126 @@ func Test_GCDropsLockedExpiredComplexObject(t *testing.T) {
_, err = sh.Get(context.Background(), getPrm)
require.True(t, client.IsErrObjectNotFound(err), "expired complex object must be deleted on epoch after lock expires")
}
func TestGCDropsObjectInhumedFromWritecache(t *testing.T) {
t.Parallel()
t.Run("flush write-cache before inhume", func(t *testing.T) {
t.Parallel()
testGCDropsObjectInhumedFromWritecache(t, true)
})
t.Run("don't flush write-cache before inhume", func(t *testing.T) {
t.Parallel()
testGCDropsObjectInhumedFromWritecache(t, false)
})
}
func testGCDropsObjectInhumedFromWritecache(t *testing.T, flushbeforeInhume bool) {
sh := newCustomShard(t, true, shardOptions{
additionalShardOptions: []Option{WithDisabledGC()},
wcOpts: []writecache.Option{writecache.WithDisableBackgroundFlush()},
})
defer func() { require.NoError(t, sh.Close()) }()
obj := testutil.GenerateObjectWithSize(1024)
var putPrm PutPrm
putPrm.SetObject(obj)
_, err := sh.Put(context.Background(), putPrm)
require.NoError(t, err)
// writecache stores object
wcObj, err := sh.writeCache.Head(context.Background(), objectCore.AddressOf(obj))
require.NoError(t, err)
require.Equal(t, objectCore.AddressOf(obj), objectCore.AddressOf(wcObj))
// blobstore doesn't store object
bsRes, err := sh.blobStor.Get(context.Background(), common.GetPrm{
Address: objectCore.AddressOf(obj),
})
require.ErrorAs(t, err, new(*apistatus.ObjectNotFound))
require.Nil(t, bsRes.Object)
require.Nil(t, bsRes.RawData)
if flushbeforeInhume {
sh.writeCache.Flush(context.Background(), false, false)
}
var inhumePrm InhumePrm
inhumePrm.MarkAsGarbage(objectCore.AddressOf(obj))
_, err = sh.Inhume(context.Background(), inhumePrm)
require.NoError(t, err)
// writecache doesn't store object
wcObj, err = sh.writeCache.Head(context.Background(), objectCore.AddressOf(obj))
require.Error(t, err)
require.Nil(t, wcObj)
if flushbeforeInhume {
// blobstore store object
bsRes, err = sh.blobStor.Get(context.Background(), common.GetPrm{
Address: objectCore.AddressOf(obj),
})
require.NoError(t, err)
require.Equal(t, objectCore.AddressOf(obj), objectCore.AddressOf(bsRes.Object))
} else {
// blobstore doesn't store object
bsRes, err = sh.blobStor.Get(context.Background(), common.GetPrm{
Address: objectCore.AddressOf(obj),
})
require.ErrorAs(t, err, new(*apistatus.ObjectNotFound))
require.Nil(t, bsRes.Object)
require.Nil(t, bsRes.RawData)
}
gcRes := sh.removeGarbage(context.Background())
require.True(t, gcRes.success)
require.Equal(t, uint64(1), gcRes.deleted)
}
func TestGCDontDeleteObjectFromWritecache(t *testing.T) {
sh := newCustomShard(t, true, shardOptions{
additionalShardOptions: []Option{WithDisabledGC()},
wcOpts: []writecache.Option{writecache.WithDisableBackgroundFlush()},
})
defer func() { require.NoError(t, sh.Close()) }()
obj := testutil.GenerateObjectWithSize(1024)
var putPrm PutPrm
putPrm.SetObject(obj)
_, err := sh.Put(context.Background(), putPrm)
require.NoError(t, err)
// writecache stores object
wcObj, err := sh.writeCache.Head(context.Background(), objectCore.AddressOf(obj))
require.NoError(t, err)
require.Equal(t, objectCore.AddressOf(obj), objectCore.AddressOf(wcObj))
// blobstore doesn't store object
bsRes, err := sh.blobStor.Get(context.Background(), common.GetPrm{
Address: objectCore.AddressOf(obj),
})
require.ErrorAs(t, err, new(*apistatus.ObjectNotFound))
require.Nil(t, bsRes.Object)
require.Nil(t, bsRes.RawData)
var metaInhumePrm meta.InhumePrm
metaInhumePrm.SetAddresses(objectCore.AddressOf(obj))
metaInhumePrm.SetLockObjectHandling()
metaInhumePrm.SetGCMark()
_, err = sh.metaBase.Inhume(context.Background(), metaInhumePrm)
require.NoError(t, err)
// logs: WARN shard/delete.go:98 can't remove object: object must be flushed from writecache
gcRes := sh.removeGarbage(context.Background())
require.True(t, gcRes.success)
require.Equal(t, uint64(0), gcRes.deleted)
// writecache stores object
wcObj, err = sh.writeCache.Head(context.Background(), objectCore.AddressOf(obj))
require.NoError(t, err)
require.Equal(t, objectCore.AddressOf(obj), objectCore.AddressOf(wcObj))
}