forked from TrueCloudLab/frostfs-node
128 lines
3.3 KiB
Go
128 lines
3.3 KiB
Go
package meta_test
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
meta "github.com/nspcc-dev/neofs-node/pkg/local_object_storage/metabase"
|
|
cidtest "github.com/nspcc-dev/neofs-sdk-go/container/id/test"
|
|
objectSDK "github.com/nspcc-dev/neofs-sdk-go/object"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDB_Exists(t *testing.T) {
|
|
db := newDB(t)
|
|
|
|
t.Run("no object", func(t *testing.T) {
|
|
nonExist := generateRawObject(t)
|
|
exists, err := meta.Exists(db, nonExist.Object().Address())
|
|
require.NoError(t, err)
|
|
require.False(t, exists)
|
|
})
|
|
|
|
t.Run("regular object", func(t *testing.T) {
|
|
regular := generateRawObject(t)
|
|
err := putBig(db, regular.Object())
|
|
require.NoError(t, err)
|
|
|
|
exists, err := meta.Exists(db, regular.Object().Address())
|
|
require.NoError(t, err)
|
|
require.True(t, exists)
|
|
})
|
|
|
|
t.Run("tombstone object", func(t *testing.T) {
|
|
ts := generateRawObject(t)
|
|
ts.SetType(objectSDK.TypeTombstone)
|
|
|
|
err := putBig(db, ts.Object())
|
|
require.NoError(t, err)
|
|
|
|
exists, err := meta.Exists(db, ts.Object().Address())
|
|
require.NoError(t, err)
|
|
require.True(t, exists)
|
|
})
|
|
|
|
t.Run("storage group object", func(t *testing.T) {
|
|
sg := generateRawObject(t)
|
|
sg.SetType(objectSDK.TypeStorageGroup)
|
|
|
|
err := putBig(db, sg.Object())
|
|
require.NoError(t, err)
|
|
|
|
exists, err := meta.Exists(db, sg.Object().Address())
|
|
require.NoError(t, err)
|
|
require.True(t, exists)
|
|
})
|
|
|
|
t.Run("virtual object", func(t *testing.T) {
|
|
cid := cidtest.GenerateID()
|
|
parent := generateRawObjectWithCID(t, cid)
|
|
|
|
child := generateRawObjectWithCID(t, cid)
|
|
child.SetParent(parent.Object().SDK())
|
|
child.SetParentID(parent.ID())
|
|
|
|
err := putBig(db, child.Object())
|
|
require.NoError(t, err)
|
|
|
|
_, err = meta.Exists(db, parent.Object().Address())
|
|
|
|
var expectedErr *objectSDK.SplitInfoError
|
|
require.True(t, errors.As(err, &expectedErr))
|
|
})
|
|
|
|
t.Run("merge split info", func(t *testing.T) {
|
|
cid := cidtest.GenerateID()
|
|
splitID := objectSDK.NewSplitID()
|
|
|
|
parent := generateRawObjectWithCID(t, cid)
|
|
addAttribute(parent, "foo", "bar")
|
|
|
|
child := generateRawObjectWithCID(t, cid)
|
|
child.SetParent(parent.Object().SDK())
|
|
child.SetParentID(parent.ID())
|
|
child.SetSplitID(splitID)
|
|
|
|
link := generateRawObjectWithCID(t, cid)
|
|
link.SetParent(parent.Object().SDK())
|
|
link.SetParentID(parent.ID())
|
|
link.SetChildren(child.ID())
|
|
link.SetSplitID(splitID)
|
|
|
|
t.Run("direct order", func(t *testing.T) {
|
|
err := putBig(db, child.Object())
|
|
require.NoError(t, err)
|
|
|
|
err = putBig(db, link.Object())
|
|
require.NoError(t, err)
|
|
|
|
_, err = meta.Exists(db, parent.Object().Address())
|
|
require.Error(t, err)
|
|
|
|
si, ok := err.(*objectSDK.SplitInfoError)
|
|
require.True(t, ok)
|
|
|
|
require.Equal(t, splitID, si.SplitInfo().SplitID())
|
|
require.Equal(t, child.ID(), si.SplitInfo().LastPart())
|
|
require.Equal(t, link.ID(), si.SplitInfo().Link())
|
|
})
|
|
|
|
t.Run("reverse order", func(t *testing.T) {
|
|
err := meta.Put(db, link.Object(), nil)
|
|
require.NoError(t, err)
|
|
|
|
err = putBig(db, child.Object())
|
|
require.NoError(t, err)
|
|
|
|
_, err = meta.Exists(db, parent.Object().Address())
|
|
require.Error(t, err)
|
|
|
|
si, ok := err.(*objectSDK.SplitInfoError)
|
|
require.True(t, ok)
|
|
|
|
require.Equal(t, splitID, si.SplitInfo().SplitID())
|
|
require.Equal(t, child.ID(), si.SplitInfo().LastPart())
|
|
require.Equal(t, link.ID(), si.SplitInfo().Link())
|
|
})
|
|
})
|
|
}
|