frostfs-node/pkg/local_object_storage/metabase/small_test.go

package meta_test

import (
	"testing"

	"github.com/nspcc-dev/neofs-node/pkg/core/object"
	"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/blobovnicza"
	meta "github.com/nspcc-dev/neofs-node/pkg/local_object_storage/metabase"
	oid "github.com/nspcc-dev/neofs-sdk-go/object/id"
	"github.com/stretchr/testify/require"
)

func TestDB_IsSmall(t *testing.T) {
	db := newDB(t)

	raw1 := generateObject(t)
	raw2 := generateObject(t)

	blobovniczaID := blobovnicza.ID{1, 2, 3, 4}

	// check IsSmall from empty database
	fetchedBlobovniczaID, err := metaIsSmall(db, object.AddressOf(raw1))
	require.NoError(t, err)
	require.Nil(t, fetchedBlobovniczaID)

	// put one object with blobovniczaID
	err = metaPut(db, raw1, &blobovniczaID)
	require.NoError(t, err)

	// put one object without blobovniczaID
	err = putBig(db, raw2)
	require.NoError(t, err)

	// check IsSmall for object without blobovniczaID
	fetchedBlobovniczaID, err = metaIsSmall(db, object.AddressOf(raw2))
	require.NoError(t, err)
	require.Nil(t, fetchedBlobovniczaID)

	// check IsSmall for object with blobovniczaID
	fetchedBlobovniczaID, err = metaIsSmall(db, object.AddressOf(raw1))
	require.NoError(t, err)
	require.Equal(t, &blobovniczaID, fetchedBlobovniczaID)
}

func metaIsSmall(db *meta.DB, addr oid.Address) (*blobovnicza.ID, error) {
	var isSmallPrm meta.IsSmallPrm
	isSmallPrm.WithAddress(addr)

	r, err := db.IsSmall(isSmallPrm)
	return r.BlobovniczaID(), err
}