package container_test

import (
	"crypto/sha256"
	"testing"

	containergrpc "git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/container/grpc"
	refsgrpc "git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/refs/grpc"
	"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container"
	cid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id"
	cidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id/test"
	"github.com/stretchr/testify/require"
	"google.golang.org/protobuf/proto"
)

func TestSizeEstimation_Epoch(t *testing.T) {
	val := container.NewSizeEstimation()

	require.Zero(t, val.Epoch())

	const epoch = 123

	val.SetEpoch(epoch)
	require.EqualValues(t, epoch, val.Epoch())

	var msg containergrpc.AnnounceUsedSpaceRequest_Body_Announcement
	val.WriteToV2(&msg)

	require.EqualValues(t, epoch, msg.GetEpoch())
}

func TestSizeEstimation_Container(t *testing.T) {
	val := container.NewSizeEstimation()

	require.Zero(t, val.Container())

	cnr := cidtest.ID()

	val.SetContainer(cnr)
	require.True(t, val.Container().Equals(cnr))

	var msg containergrpc.AnnounceUsedSpaceRequest_Body_Announcement
	val.WriteToV2(&msg)

	var msgCnr refsgrpc.ContainerID
	cnr.WriteToV2(&msgCnr)

	require.True(t, proto.Equal(&msgCnr, msg.GetContainerId()))
}

func TestSizeEstimation_Value(t *testing.T) {
	val := container.NewSizeEstimation()

	require.Zero(t, val.Value())

	const value = 876

	val.SetValue(value)
	require.EqualValues(t, value, val.Value())

	var msg containergrpc.AnnounceUsedSpaceRequest_Body_Announcement
	val.WriteToV2(&msg)

	require.EqualValues(t, value, msg.GetUsedSpace())
}

func TestSizeEstimation_ReadFromV2(t *testing.T) {
	const epoch = 654
	const value = 903
	var cnrMsg refsgrpc.ContainerID

	msg := new(containergrpc.AnnounceUsedSpaceRequest_Body_Announcement)

	val := container.NewSizeEstimation()

	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())
}