95 lines
2 KiB
Go
95 lines
2 KiB
Go
|
package container_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/sha256"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
v2container "github.com/nspcc-dev/neofs-api-go/v2/container"
|
||
|
|
"github.com/nspcc-dev/neofs-api-go/v2/refs"
|
||
|
|
"github.com/nspcc-dev/neofs-sdk-go/container"
|
||
|
|
cid "github.com/nspcc-dev/neofs-sdk-go/container/id"
|
||
|
|
cidtest "github.com/nspcc-dev/neofs-sdk-go/container/id/test"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestSizeEstimation_Epoch(t *testing.T) {
|
||
|
|
var val container.SizeEstimation
|
||
|
|
|
||
|
|
require.Zero(t, val.Epoch())
|
||
|
|
|
||
|
|
const epoch = 123
|
||
|
|
|
||
|
|
val.SetEpoch(epoch)
|
||
|
|
require.EqualValues(t, epoch, val.Epoch())
|
||
|
|
|
||
|
|
var msg v2container.UsedSpaceAnnouncement
|
||
|
|
val.WriteToV2(&msg)
|
||
|
|
|
||
|
|
require.EqualValues(t, epoch, msg.GetEpoch())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSizeEstimation_Container(t *testing.T) {
|
||
|
|
var val container.SizeEstimation
|
||
|
|
|
||
|
|
require.Zero(t, val.Container())
|
||
|
|
|
||
|
|
cnr := cidtest.ID()
|
||
|
|
|
||
|
|
val.SetContainer(cnr)
|
||
|
|
require.True(t, val.Container().Equals(cnr))
|
||
|
|
|
||
|
|
var msg v2container.UsedSpaceAnnouncement
|
||
|
|
val.WriteToV2(&msg)
|
||
|
|
|
||
|
|
var msgCnr refs.ContainerID
|
||
|
|
cnr.WriteToV2(&msgCnr)
|
||
|
|
|
||
|
|
require.Equal(t, &msgCnr, msg.GetContainerID())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSizeEstimation_Value(t *testing.T) {
|
||
|
|
var val container.SizeEstimation
|
||
|
|
|
||
|
|
require.Zero(t, val.Value())
|
||
|
|
|
||
|
|
const value = 876
|
||
|
|
|
||
|
|
val.SetValue(value)
|
||
|
|
require.EqualValues(t, value, val.Value())
|
||
|
|
|
||
|
|
var msg v2container.UsedSpaceAnnouncement
|
||
|
|
val.WriteToV2(&msg)
|
||
|
|
|
||
|
|
require.EqualValues(t, value, msg.GetUsedSpace())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSizeEstimation_ReadFromV2(t *testing.T) {
|
||
|
|
const epoch = 654
|
||
|
|
const value = 903
|
||
|
|
var cnrMsg refs.ContainerID
|
||
|
|
|
||
|
|
var msg v2container.UsedSpaceAnnouncement
|
||
|
|
|
||
|
|
var val container.SizeEstimation
|
||
|
|
|
||
|
|
require.Error(t, val.ReadFromV2(msg))
|
||
|
|
|
||
|
|
msg.SetContainerID(&cnrMsg)
|
||
|
|
|
||
|
|
require.Error(t, val.ReadFromV2(msg))
|
||
|
|
|
||
|
|
cnrMsg.SetValue(make([]byte, sha256.Size))
|
||
|
|
|
||
|
|
var cnr cid.ID
|
||
|
|
require.NoError(t, cnr.ReadFromV2(cnrMsg))
|
||
|
|
|
||
|
|
msg.SetEpoch(epoch)
|
||
|
|
msg.SetUsedSpace(value)
|
||
|
|
|
||
|
|
require.NoError(t, val.ReadFromV2(msg))
|
||
|
|
|
||
|
|
require.EqualValues(t, epoch, val.Epoch())
|
||
|
|
require.EqualValues(t, value, val.Value())
|
||
|
|
require.EqualValues(t, cnr, val.Container())
|
||
|
|
}
|