frostfs-sdk-go/object/splitinfo_test.go

package object_test

import (
	"crypto/rand"
	"encoding/json"
	"testing"

	objv2 "git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/object/grpc"
	"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
	oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
	"github.com/golang/protobuf/proto"
	"github.com/stretchr/testify/require"
)

func TestSplitInfo(t *testing.T) {
	s := object.NewSplitInfo()
	splitID := object.NewSplitID()
	lastPart := generateID()
	link := generateID()

	s.SetSplitID(splitID)
	require.Equal(t, splitID, s.SplitID())

	s.SetLastPart(lastPart)
	lp, set := s.LastPart()
	require.True(t, set)
	require.Equal(t, lastPart, lp)

	s.SetLink(link)
	l, set := s.Link()
	require.True(t, set)
	require.Equal(t, link, l)
}

func TestSplitInfoMarshal(t *testing.T) {
	testToV2 := func(t *testing.T, s *object.SplitInfo) {
		v2 := s.ToV2()
		newS := object.NewSplitInfoFromV2(v2)

		require.True(t, proto.Equal(s.ToV2(), newS.ToV2()))
	}
	testMarshal := func(t *testing.T, s *object.SplitInfo) {
		data, err := s.Marshal()
		require.NoError(t, err)

		newS := object.NewSplitInfo()

		err = newS.Unmarshal(data)
		require.NoError(t, err)
		require.True(t, proto.Equal(s.ToV2(), newS.ToV2()))
	}

	t.Run("good, both fields are set", func(t *testing.T) {
		s := object.NewSplitInfo()
		s.SetSplitID(object.NewSplitID())
		s.SetLink(generateID())
		s.SetLastPart(generateID())

		testToV2(t, s)
		testMarshal(t, s)
	})
	t.Run("good, only link is set", func(t *testing.T) {
		s := object.NewSplitInfo()
		s.SetSplitID(object.NewSplitID())
		s.SetLink(generateID())

		testToV2(t, s)
		testMarshal(t, s)
	})
	t.Run("good, only last part is set", func(t *testing.T) {
		s := object.NewSplitInfo()
		s.SetSplitID(object.NewSplitID())
		s.SetLastPart(generateID())

		testToV2(t, s)
		testMarshal(t, s)
	})
	t.Run("bad, no fields are set", func(t *testing.T) {
		s := object.NewSplitInfo()
		s.SetSplitID(object.NewSplitID())

		data, err := s.Marshal()
		require.NoError(t, err)
		require.Error(t, object.NewSplitInfo().Unmarshal(data))
	})
}

func generateID() oid.ID {
	var buf [32]byte
	_, _ = rand.Read(buf[:])

	var id oid.ID
	id.SetSHA256(buf)

	return id
}

func TestNewSplitInfoFromV2(t *testing.T) {
	t.Run("from nil", func(t *testing.T) {
		var x *objv2.SplitInfo

		require.Nil(t, object.NewSplitInfoFromV2(x))
	})
}

func TestSplitInfo_ToV2(t *testing.T) {
	t.Run("nil", func(t *testing.T) {
		var x *object.SplitInfo

		require.Nil(t, x.ToV2())
	})
}

func TestNewSplitInfo(t *testing.T) {
	t.Run("default values", func(t *testing.T) {
		si := object.NewSplitInfo()

		// check initial values
		require.Nil(t, si.SplitID())
		_, set := si.LastPart()
		require.False(t, set)
		_, set = si.Link()
		require.False(t, set)

		// convert to v2 message
		siV2 := si.ToV2()

		require.Nil(t, siV2.GetSplitId())
		require.Nil(t, siV2.GetLastPart())
		require.Nil(t, siV2.GetLink())
	})
}

func TestSplitInfoMarshalJSON(t *testing.T) {
	t.Run("good", func(t *testing.T) {
		s := object.NewSplitInfo()
		s.SetSplitID(object.NewSplitID())
		s.SetLastPart(generateID())
		s.SetLink(generateID())

		data, err := s.MarshalJSON()
		require.NoError(t, err)

		actual := object.NewSplitInfo()
		require.NoError(t, json.Unmarshal(data, actual))
		require.Equal(t, s, actual)
	})
	t.Run("bad link", func(t *testing.T) {
		data := `{"splitId":"Sn707289RrqDyJOrZMbMoQ==","lastPart":{"value":"Y7baWE0UdUOBr1ELKX3Q5v1LKRubQUbI81Q5UxCVeow="},"link":{"value":"bad"}}`
		require.Error(t, json.Unmarshal([]byte(data), object.NewSplitInfo()))
	})
	t.Run("bad last part", func(t *testing.T) {
		data := `{"splitId":"Sn707289RrqDyJOrZMbMoQ==","lastPart":{"value":"bad"},"link":{"value":"eRyPNCNNxHfxPcjijlv05HEcdoep/b7eHNLRSmDlnts="}}`
		require.Error(t, json.Unmarshal([]byte(data), object.NewSplitInfo()))
	})
}