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

package meta_test

import (
	"bytes"
	"testing"

	objectSDK "github.com/nspcc-dev/neofs-api-go/pkg/object"
	"github.com/nspcc-dev/neofs-node/pkg/core/object"
	meta "github.com/nspcc-dev/neofs-node/pkg/local_object_storage/metabase"
	"github.com/stretchr/testify/require"
)

func TestDB_Get(t *testing.T) {
	db := newDB(t)
	defer releaseDB(db)

	raw := generateRawObject(t)

	// equal fails on diff of <nil> attributes and <{}> attributes,
	/* so we make non empty attribute slice in parent*/
	addAttribute(raw, "foo", "bar")

	t.Run("object not found", func(t *testing.T) {
		_, err := meta.Get(db, raw.Object().Address())
		require.Error(t, err)
	})

	t.Run("put regular object", func(t *testing.T) {
		err := putBig(db, raw.Object())
		require.NoError(t, err)

		newObj, err := meta.Get(db, raw.Object().Address())
		require.NoError(t, err)
		require.Equal(t, raw.Object(), newObj)
	})

	t.Run("put tombstone object", func(t *testing.T) {
		raw.SetType(objectSDK.TypeTombstone)
		raw.SetID(testOID())

		err := putBig(db, raw.Object())
		require.NoError(t, err)

		newObj, err := meta.Get(db, raw.Object().Address())
		require.NoError(t, err)
		require.Equal(t, raw.Object(), newObj)
	})

	t.Run("put storage group object", func(t *testing.T) {
		raw.SetType(objectSDK.TypeStorageGroup)
		raw.SetID(testOID())

		err := putBig(db, raw.Object())
		require.NoError(t, err)

		newObj, err := meta.Get(db, raw.Object().Address())
		require.NoError(t, err)
		require.Equal(t, raw.Object(), newObj)
	})

	t.Run("put virtual object", func(t *testing.T) {
		cid := testCID()
		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)

		err := putBig(db, child.Object())
		require.NoError(t, err)

		t.Run("raw is true", func(t *testing.T) {
			_, err = meta.GetRaw(db, parent.Object().Address(), true)
			require.Error(t, err)

			siErr, ok := err.(*objectSDK.SplitInfoError)
			require.True(t, ok)

			require.Equal(t, splitID, siErr.SplitInfo().SplitID())
			require.Equal(t, child.ID(), siErr.SplitInfo().LastPart())
			require.Nil(t, siErr.SplitInfo().Link())
		})

		newParent, err := meta.GetRaw(db, parent.Object().Address(), false)
		require.NoError(t, err)
		require.True(t, binaryEqual(parent.Object(), newParent))

		newChild, err := meta.GetRaw(db, child.Object().Address(), true)
		require.NoError(t, err)
		require.True(t, binaryEqual(child.Object(), newChild))
	})
}

// binary equal is used when object contains empty lists in the structure and
// requre.Equal fails on comparing <nil> and []{} lists.
func binaryEqual(a, b *object.Object) bool {
	binaryA, err := a.Marshal()
	if err != nil {
		return false
	}

	binaryB, err := b.Marshal()
	if err != nil {
		return false
	}

	return bytes.Equal(binaryA, binaryB)
}