frostfs-node/pkg/local_object_storage/blobovnicza/blobovnicza_test.go

package blobovnicza

import (
	"context"
	"crypto/rand"
	"os"
	"testing"

	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/logger/test"
	"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/client"
	oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
	oidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id/test"
	"github.com/stretchr/testify/require"
)

func testPutGet(t *testing.T, blz *Blobovnicza, addr oid.Address, sz uint64, assertErrPut, assertErrGet func(error) bool) oid.Address {
	// create binary object
	data := make([]byte, sz)
	rand.Read(data)

	var pPut PutPrm
	pPut.SetAddress(addr)
	pPut.SetMarshaledObject(data)
	_, err := blz.Put(context.Background(), pPut)
	if assertErrPut != nil {
		require.True(t, assertErrPut(err))
	} else {
		require.NoError(t, err)
	}

	if assertErrGet != nil {
		testGet(t, blz, addr, data, assertErrGet)
	}

	return addr
}

func testGet(t *testing.T, blz *Blobovnicza, addr oid.Address, expObj []byte, assertErr func(error) bool) {
	var pGet GetPrm
	pGet.SetAddress(addr)

	// try to read object from Blobovnicza
	res, err := blz.Get(context.Background(), pGet)
	if assertErr != nil {
		require.True(t, assertErr(err))
	} else {
		require.NoError(t, err)
	}

	if assertErr == nil {
		require.Equal(t, expObj, res.Object())
	}
}

func TestBlobovnicza(t *testing.T) {
	p := "./test_blz"

	sizeLim := uint64(256 * 1 << 10) // 256KB
	objSizeLim := sizeLim / 2

	// create Blobovnicza instance
	blz := New(
		WithPath(p),
		WithObjectSizeLimit(objSizeLim),
		WithFullSizeLimit(sizeLim),
		WithLogger(test.NewLogger(t)),
	)

	defer os.Remove(p)

	// open Blobovnicza
	require.NoError(t, blz.Open(context.Background()))

	// initialize Blobovnicza
	require.NoError(t, blz.Init(context.Background()))

	// try to read non-existent address
	testGet(t, blz, oidtest.Address(), nil, client.IsErrObjectNotFound)

	filled := uint64(15 * 1 << 10)

	// test object 15KB
	addr := testPutGet(t, blz, oidtest.Address(), filled, nil, nil)

	// remove the object
	var dPrm DeletePrm
	dPrm.SetAddress(addr)

	_, err := blz.Delete(context.Background(), dPrm)
	require.NoError(t, err)

	// should return 404
	testGet(t, blz, addr, nil, client.IsErrObjectNotFound)

	// fill Blobovnicza fully
	for ; filled < sizeLim; filled += objSizeLim {
		testPutGet(t, blz, oidtest.Address(), objSizeLim, nil, nil)
	}

	// blobovnizca accepts object event if full
	testPutGet(t, blz, oidtest.Address(), 1024, func(err error) bool {
		return err == nil
	}, nil)

	require.NoError(t, blz.Close(context.Background()))
}