forked from TrueCloudLab/frostfs-node
60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
package shard_test
|
|
|
|
import (
|
|
"math/rand"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neofs-node/pkg/core/object"
|
|
"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/shard"
|
|
"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/writecache"
|
|
cidtest "github.com/nspcc-dev/neofs-sdk-go/container/id/test"
|
|
objectSDK "github.com/nspcc-dev/neofs-sdk-go/object"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestWriteCacheObjectLoss(t *testing.T) {
|
|
const (
|
|
smallSize = 1024
|
|
objCount = 100
|
|
)
|
|
|
|
objects := make([]*objectSDK.Object, objCount)
|
|
for i := range objects {
|
|
size := smallSize
|
|
// if i%2 == 0 {
|
|
size = smallSize / 2
|
|
// }
|
|
data := make([]byte, size)
|
|
rand.Read(data)
|
|
|
|
objects[i] = generateObjectWithPayload(cidtest.ID(), data)
|
|
}
|
|
|
|
dir := t.TempDir()
|
|
wcOpts := []writecache.Option{
|
|
writecache.WithSmallObjectSize(smallSize),
|
|
writecache.WithMaxObjectSize(smallSize * 2)}
|
|
|
|
sh := newCustomShard(t, dir, true, wcOpts, nil)
|
|
|
|
var putPrm shard.PutPrm
|
|
|
|
for i := range objects {
|
|
putPrm.SetObject(objects[i])
|
|
_, err := sh.Put(putPrm)
|
|
require.NoError(t, err)
|
|
}
|
|
require.NoError(t, sh.Close())
|
|
|
|
sh = newCustomShard(t, dir, true, wcOpts, nil)
|
|
defer releaseShard(sh, t)
|
|
|
|
var getPrm shard.GetPrm
|
|
|
|
for i := range objects {
|
|
getPrm.SetAddress(object.AddressOf(objects[i]))
|
|
|
|
_, err := sh.Get(getPrm)
|
|
require.NoError(t, err, i)
|
|
}
|
|
}
|