forked from TrueCloudLab/frostfs-node
82 lines
1.9 KiB
Go
82 lines
1.9 KiB
Go
package meta_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neofs-node/pkg/core/object"
|
|
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_Movable(t *testing.T) {
|
|
db := newDB(t)
|
|
|
|
raw1 := generateObject(t)
|
|
raw2 := generateObject(t)
|
|
|
|
// 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(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
|
|
}
|