forked from TrueCloudLab/frostfs-node
66 lines
2 KiB
Go
66 lines
2 KiB
Go
|
package writecachebadger
|
||
|
|
|
||
|
|
import (
|
||
|
|
"path/filepath"
|
||
|
|
"sync/atomic"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor"
|
||
|
|
"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/local_object_storage/writecache/writecachetest"
|
||
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/logger"
|
||
|
|
oidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id/test"
|
||
|
|
"github.com/dgraph-io/badger/v4"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
"go.uber.org/zap/zaptest"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestFlush(t *testing.T) {
|
||
|
|
createCacheFn := func(t *testing.T, smallSize uint64, mb *meta.DB, bs *blobstor.BlobStor, opts ...Option) writecache.Cache {
|
||
|
|
return New(
|
||
|
|
append([]Option{
|
||
|
|
WithLogger(&logger.Logger{Logger: zaptest.NewLogger(t)}),
|
||
|
|
WithPath(filepath.Join(t.TempDir(), "writecache")),
|
||
|
|
WithMetabase(mb),
|
||
|
|
WithBlobstor(bs),
|
||
|
|
WithGCInterval(1 * time.Second),
|
||
|
|
}, opts...)...)
|
||
|
|
}
|
||
|
|
|
||
|
|
errCountOpt := func() (Option, *atomic.Uint32) {
|
||
|
|
cnt := &atomic.Uint32{}
|
||
|
|
return WithReportErrorFunc(func(string, error) {
|
||
|
|
cnt.Add(1)
|
||
|
|
}), cnt
|
||
|
|
}
|
||
|
|
|
||
|
|
failures := []writecachetest.TestFailureInjector[Option]{
|
||
|
|
{
|
||
|
|
Desc: "db, invalid address",
|
||
|
|
InjectFn: func(t *testing.T, wc writecache.Cache) {
|
||
|
|
c := wc.(*cache)
|
||
|
|
obj := testutil.GenerateObject()
|
||
|
|
data, err := obj.Marshal()
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NoError(t, c.db.Update(func(tx *badger.Txn) error {
|
||
|
|
return tx.Set([]byte{1, 2, 3}, data)
|
||
|
|
}))
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Desc: "db, invalid object",
|
||
|
|
InjectFn: func(t *testing.T, wc writecache.Cache) {
|
||
|
|
c := wc.(*cache)
|
||
|
|
require.NoError(t, c.db.Update(func(tx *badger.Txn) error {
|
||
|
|
return tx.Set([]byte(oidtest.Address().EncodeToString()), []byte{1, 2, 3})
|
||
|
|
}))
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
writecachetest.TestFlush(t, createCacheFn, errCountOpt, failures...)
|
||
|
|
}
|