frostfs-node/pkg/local_object_storage/shard/shutdown_test.go

package shard_test

import (
	"context"
	"math/rand"
	"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"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/writecache"
	cidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id/test"
	objectSDK "git.frostfs.info/TrueCloudLab/frostfs-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] = testutil.GenerateObjectWithCIDWithPayload(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(context.Background(), 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(context.Background(), getPrm)
		require.NoError(t, err, i)
	}
}