package meta_test import ( "errors" "testing" "github.com/nspcc-dev/neofs-node/pkg/core/object" 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 := generateObject(t) exists, err := meta.Exists(db, object.AddressOf(nonExist)) require.NoError(t, err) require.False(t, exists) }) t.Run("regular object", func(t *testing.T) { regular := generateObject(t) err := putBig(db, regular) require.NoError(t, err) exists, err := meta.Exists(db, object.AddressOf(regular)) require.NoError(t, err) require.True(t, exists) }) t.Run("tombstone object", func(t *testing.T) { ts := generateObject(t) ts.SetType(objectSDK.TypeTombstone) err := putBig(db, ts) require.NoError(t, err) exists, err := meta.Exists(db, object.AddressOf(ts)) require.NoError(t, err) require.True(t, exists) }) t.Run("storage group object", func(t *testing.T) { sg := generateObject(t) sg.SetType(objectSDK.TypeStorageGroup) err := putBig(db, sg) require.NoError(t, err) exists, err := meta.Exists(db, object.AddressOf(sg)) require.NoError(t, err) require.True(t, exists) }) t.Run("lock object", func(t *testing.T) { lock := generateObject(t) lock.SetType(objectSDK.TypeLock) err := putBig(db, lock) require.NoError(t, err) exists, err := meta.Exists(db, object.AddressOf(lock)) require.NoError(t, err) require.True(t, exists) }) t.Run("virtual object", func(t *testing.T) { cid := cidtest.ID() parent := generateObjectWithCID(t, cid) child := generateObjectWithCID(t, cid) child.SetParent(parent) idParent, _ := parent.ID() child.SetParentID(idParent) err := putBig(db, child) require.NoError(t, err) _, err = meta.Exists(db, object.AddressOf(parent)) var expectedErr *objectSDK.SplitInfoError require.True(t, errors.As(err, &expectedErr)) }) t.Run("merge split info", func(t *testing.T) { cid := cidtest.ID() splitID := objectSDK.NewSplitID() parent := generateObjectWithCID(t, cid) addAttribute(parent, "foo", "bar") child := generateObjectWithCID(t, cid) child.SetParent(parent) idParent, _ := parent.ID() child.SetParentID(idParent) child.SetSplitID(splitID) link := generateObjectWithCID(t, cid) link.SetParent(parent) link.SetParentID(idParent) idChild, _ := child.ID() link.SetChildren(idChild) link.SetSplitID(splitID) t.Run("direct order", func(t *testing.T) { err := putBig(db, child) require.NoError(t, err) err = putBig(db, link) require.NoError(t, err) _, err = meta.Exists(db, object.AddressOf(parent)) require.Error(t, err) si, ok := err.(*objectSDK.SplitInfoError) require.True(t, ok) require.Equal(t, splitID, si.SplitInfo().SplitID()) id1, _ := child.ID() id2, _ := si.SplitInfo().LastPart() require.Equal(t, id1, id2) id1, _ = link.ID() id2, _ = si.SplitInfo().Link() require.Equal(t, id1, id2) }) t.Run("reverse order", func(t *testing.T) { err := meta.Put(db, link, nil) require.NoError(t, err) err = putBig(db, child) require.NoError(t, err) _, err = meta.Exists(db, object.AddressOf(parent)) require.Error(t, err) si, ok := err.(*objectSDK.SplitInfoError) require.True(t, ok) require.Equal(t, splitID, si.SplitInfo().SplitID()) id1, _ := child.ID() id2, _ := si.SplitInfo().LastPart() require.Equal(t, id1, id2) id1, _ = link.ID() id2, _ = si.SplitInfo().Link() require.Equal(t, id1, id2) }) }) }