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

package meta_test

import (
	"context"
	"testing"

	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/object"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/internal/testutil"
	meta "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/metabase"
	oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
	"github.com/stretchr/testify/require"
)

func TestDB_Movable(t *testing.T) {
	db := newDB(t)
	defer func() { require.NoError(t, db.Close()) }()

	raw1 := testutil.GenerateObject()
	raw2 := testutil.GenerateObject()

	// put two objects in metabase
	err := putBig(db, raw1)
	require.NoError(t, err)

	err = putBig(db, raw2)
	require.NoError(t, err)

	// check if toMoveIt index empty
	toMoveList, err := metaMovable(db)
	require.NoError(t, err)
	require.Len(t, toMoveList, 0)

	// mark to move object2
	err = metaToMoveIt(db, object.AddressOf(raw2))
	require.NoError(t, err)

	// check if toMoveIt index contains address of object 2
	toMoveList, err = metaMovable(db)
	require.NoError(t, err)
	require.Len(t, toMoveList, 1)
	require.Contains(t, toMoveList, object.AddressOf(raw2))

	// remove from toMoveIt index non existing address
	err = metaDoNotMove(db, object.AddressOf(raw1))
	require.NoError(t, err)

	// check if toMoveIt index hasn't changed
	toMoveList, err = metaMovable(db)
	require.NoError(t, err)
	require.Len(t, toMoveList, 1)

	// remove from toMoveIt index existing address
	err = metaDoNotMove(db, object.AddressOf(raw2))
	require.NoError(t, err)

	// check if toMoveIt index is empty now
	toMoveList, err = metaMovable(db)
	require.NoError(t, err)
	require.Len(t, toMoveList, 0)
}

func metaToMoveIt(db *meta.DB, addr oid.Address) error {
	var toMovePrm meta.ToMoveItPrm
	toMovePrm.SetAddress(addr)

	_, err := db.ToMoveIt(context.Background(), toMovePrm)
	return err
}

func metaMovable(db *meta.DB) ([]oid.Address, error) {
	r, err := db.Movable(meta.MovablePrm{})
	if err != nil {
		return nil, err
	}

	return r.AddressList(), nil
}

func metaDoNotMove(db *meta.DB, addr oid.Address) error {
	var doNotMovePrm meta.DoNotMovePrm
	doNotMovePrm.SetAddress(addr)

	_, err := db.DoNotMove(doNotMovePrm)
	return err
}