forked from TrueCloudLab/frostfs-node
[#668] shard/test: Add dontRelease options
Most of the time we would like to close shard with minor exceptions. Signed-off-by: Evgenii Stratonikov <e.stratonikov@yadro.com>
This commit is contained in:
parent
c661ba1312
commit
39879fa868
4 changed files with 7 additions and 14 deletions
|
@ -27,10 +27,6 @@ func Test_GCDropsLockedExpiredSimpleObject(t *testing.T) {
|
|||
|
||||
sh := newCustomShard(t, false, shardOptions{metaOptions: []meta.Option{meta.WithEpochState(epoch)}})
|
||||
|
||||
t.Cleanup(func() {
|
||||
releaseShard(sh, t)
|
||||
})
|
||||
|
||||
cnr := cidtest.ID()
|
||||
|
||||
var objExpirationAttr objectSDK.Attribute
|
||||
|
@ -124,10 +120,6 @@ func Test_GCDropsLockedExpiredComplexObject(t *testing.T) {
|
|||
|
||||
sh := newCustomShard(t, false, shardOptions{metaOptions: []meta.Option{meta.WithEpochState(epoch)}})
|
||||
|
||||
t.Cleanup(func() {
|
||||
releaseShard(sh, t)
|
||||
})
|
||||
|
||||
lock := testutil.GenerateObjectWithCID(cnr)
|
||||
lock.SetType(objectSDK.TypeLock)
|
||||
lock.SetAttributes(lockExpirationAttr)
|
||||
|
|
|
@ -97,7 +97,6 @@ func testShardGetRange(t *testing.T, hasWriteCache bool) {
|
|||
}),
|
||||
},
|
||||
})
|
||||
defer releaseShard(sh, t)
|
||||
|
||||
for _, tc := range testCases {
|
||||
t.Run(tc.name, func(t *testing.T) {
|
||||
|
|
|
@ -32,15 +32,14 @@ func (s epochState) CurrentEpoch() uint64 {
|
|||
|
||||
type shardOptions struct {
|
||||
rootPath string
|
||||
dontRelease bool
|
||||
wcOpts writecacheconfig.Options
|
||||
bsOpts []blobstor.Option
|
||||
metaOptions []meta.Option
|
||||
}
|
||||
|
||||
func newShard(t testing.TB, enableWriteCache bool) *Shard {
|
||||
sh := newCustomShard(t, enableWriteCache, shardOptions{})
|
||||
t.Cleanup(func() { releaseShard(sh, t) })
|
||||
return sh
|
||||
return newCustomShard(t, enableWriteCache, shardOptions{})
|
||||
}
|
||||
|
||||
func newCustomShard(t testing.TB, enableWriteCache bool, o shardOptions) *Shard {
|
||||
|
@ -121,6 +120,10 @@ func newCustomShard(t testing.TB, enableWriteCache bool, o shardOptions) *Shard
|
|||
require.NoError(t, sh.Open())
|
||||
require.NoError(t, sh.Init(context.Background()))
|
||||
|
||||
if !o.dontRelease {
|
||||
t.Cleanup(func() { releaseShard(sh, t) })
|
||||
}
|
||||
|
||||
return sh
|
||||
}
|
||||
|
||||
|
|
|
@ -43,7 +43,7 @@ func TestWriteCacheObjectLoss(t *testing.T) {
|
|||
},
|
||||
}
|
||||
|
||||
sh := newCustomShard(t, true, shardOptions{rootPath: dir, wcOpts: wcOpts})
|
||||
sh := newCustomShard(t, true, shardOptions{dontRelease: true, rootPath: dir, wcOpts: wcOpts})
|
||||
|
||||
var errG errgroup.Group
|
||||
for i := range objects {
|
||||
|
@ -59,7 +59,6 @@ func TestWriteCacheObjectLoss(t *testing.T) {
|
|||
require.NoError(t, sh.Close())
|
||||
|
||||
sh = newCustomShard(t, true, shardOptions{rootPath: dir, wcOpts: wcOpts})
|
||||
defer releaseShard(sh, t)
|
||||
|
||||
var getPrm GetPrm
|
||||
|
||||
|
|
Loading…
Reference in a new issue