frostfs-sdk-go/object/tombstone_test.go
Evgenii Stratonikov 49ad985cad [#161] *: Do not use math/rand.Read()
Signed-off-by: Evgenii Stratonikov <e.stratonikov@yadro.com>
2023-09-08 17:17:02 +03:00

95 lines
1.9 KiB
Go

package object
import (
"crypto/rand"
"crypto/sha256"
"testing"
"git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/tombstone"
oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
"github.com/stretchr/testify/require"
)
func generateIDList(sz int) []oid.ID {
res := make([]oid.ID, sz)
cs := [sha256.Size]byte{}
for i := 0; i < sz; i++ {
var oID oid.ID
res[i] = oID
rand.Read(cs[:])
res[i].SetSHA256(cs)
}
return res
}
func TestTombstone(t *testing.T) {
ts := NewTombstone()
exp := uint64(13)
ts.SetExpirationEpoch(exp)
require.Equal(t, exp, ts.ExpirationEpoch())
splitID := NewSplitID()
ts.SetSplitID(splitID)
require.Equal(t, splitID, ts.SplitID())
members := generateIDList(3)
ts.SetMembers(members)
require.Equal(t, members, ts.Members())
}
func TestTombstoneEncoding(t *testing.T) {
ts := NewTombstone()
ts.SetExpirationEpoch(13)
ts.SetSplitID(NewSplitID())
ts.SetMembers(generateIDList(5))
t.Run("binary", func(t *testing.T) {
data, err := ts.Marshal()
require.NoError(t, err)
ts2 := NewTombstone()
require.NoError(t, ts2.Unmarshal(data))
require.Equal(t, ts, ts2)
})
t.Run("json", func(t *testing.T) {
data, err := ts.MarshalJSON()
require.NoError(t, err)
ts2 := NewTombstone()
require.NoError(t, ts2.UnmarshalJSON(data))
require.Equal(t, ts, ts2)
})
}
func TestNewTombstoneFromV2(t *testing.T) {
t.Run("from nil", func(t *testing.T) {
var x *tombstone.Tombstone
require.Nil(t, NewTombstoneFromV2(x))
})
}
func TestNewTombstone(t *testing.T) {
t.Run("default values", func(t *testing.T) {
ts := NewTombstone()
// check initial values
require.Nil(t, ts.SplitID())
require.Nil(t, ts.Members())
require.Zero(t, ts.ExpirationEpoch())
// convert to v2 message
tsV2 := ts.ToV2()
require.Nil(t, tsV2.GetSplitID())
require.Nil(t, tsV2.GetMembers())
require.Zero(t, tsV2.GetExpirationEpoch())
})
}