package audit_test

import (
	"testing"

	auditv2 "github.com/nspcc-dev/neofs-api-go/v2/audit"
	"github.com/nspcc-dev/neofs-sdk-go/audit"
	audittest "github.com/nspcc-dev/neofs-sdk-go/audit/test"
	cidtest "github.com/nspcc-dev/neofs-sdk-go/container/id/test"
	"github.com/nspcc-dev/neofs-sdk-go/object"
	objecttest "github.com/nspcc-dev/neofs-sdk-go/object/test"
	"github.com/nspcc-dev/neofs-sdk-go/version"
	"github.com/stretchr/testify/require"
)

func TestResult(t *testing.T) {
	r := audit.NewResult()
	require.Equal(t, version.Current(), r.Version())

	epoch := uint64(13)
	r.SetAuditEpoch(epoch)
	require.Equal(t, epoch, r.AuditEpoch())

	cid := cidtest.GenerateID()
	r.SetContainerID(cid)
	require.Equal(t, cid, r.ContainerID())

	key := []byte{1, 2, 3}
	r.SetPublicKey(key)
	require.Equal(t, key, r.PublicKey())

	r.SetComplete(true)
	require.True(t, r.Complete())

	requests := uint32(2)
	r.SetRequests(requests)
	require.Equal(t, requests, r.Requests())

	retries := uint32(1)
	r.SetRetries(retries)
	require.Equal(t, retries, r.Retries())

	passSG := []*object.ID{objecttest.ID(), objecttest.ID()}
	r.SetPassSG(passSG)
	require.Equal(t, passSG, r.PassSG())

	failSG := []*object.ID{objecttest.ID(), objecttest.ID()}
	r.SetFailSG(failSG)
	require.Equal(t, failSG, r.FailSG())

	hit := uint32(1)
	r.SetHit(hit)
	require.Equal(t, hit, r.Hit())

	miss := uint32(2)
	r.SetMiss(miss)
	require.Equal(t, miss, r.Miss())

	fail := uint32(3)
	r.SetFail(fail)
	require.Equal(t, fail, r.Fail())

	passNodes := [][]byte{{1}, {2}}
	r.SetPassNodes(passNodes)
	require.Equal(t, passNodes, r.PassNodes())

	failNodes := [][]byte{{3}, {4}}
	r.SetFailNodes(failNodes)
	require.Equal(t, failNodes, r.FailNodes())
}

func TestStorageGroupEncoding(t *testing.T) {
	r := audittest.Generate()

	t.Run("binary", func(t *testing.T) {
		data, err := r.Marshal()
		require.NoError(t, err)

		r2 := audit.NewResult()
		require.NoError(t, r2.Unmarshal(data))

		require.Equal(t, r, r2)
	})

	t.Run("json", func(t *testing.T) {
		data, err := r.MarshalJSON()
		require.NoError(t, err)

		r2 := audit.NewResult()
		require.NoError(t, r2.UnmarshalJSON(data))

		require.Equal(t, r, r2)
	})
}

func TestResult_ToV2(t *testing.T) {
	t.Run("nil", func(t *testing.T) {
		var x *audit.Result

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

	t.Run("default values", func(t *testing.T) {
		result := audit.NewResult()

		// check initial values
		require.Equal(t, version.Current(), result.Version())

		require.False(t, result.Complete())

		require.Nil(t, result.ContainerID())
		require.Nil(t, result.PublicKey())
		require.Nil(t, result.PassSG())
		require.Nil(t, result.FailSG())
		require.Nil(t, result.PassNodes())
		require.Nil(t, result.FailNodes())

		require.Zero(t, result.Hit())
		require.Zero(t, result.Miss())
		require.Zero(t, result.Fail())
		require.Zero(t, result.Requests())
		require.Zero(t, result.Retries())
		require.Zero(t, result.AuditEpoch())

		// convert to v2 message
		resultV2 := result.ToV2()

		require.Equal(t, version.Current().ToV2(), resultV2.GetVersion())

		require.False(t, resultV2.GetComplete())

		require.Nil(t, resultV2.GetContainerID())
		require.Nil(t, resultV2.GetPublicKey())
		require.Nil(t, resultV2.GetPassSG())
		require.Nil(t, resultV2.GetFailSG())
		require.Nil(t, resultV2.GetPassNodes())
		require.Nil(t, resultV2.GetFailNodes())

		require.Zero(t, resultV2.GetHit())
		require.Zero(t, resultV2.GetMiss())
		require.Zero(t, resultV2.GetFail())
		require.Zero(t, resultV2.GetRequests())
		require.Zero(t, resultV2.GetRetries())
		require.Zero(t, resultV2.GetAuditEpoch())
	})
}

func TestNewResultFromV2(t *testing.T) {
	t.Run("from nil", func(t *testing.T) {
		var x *auditv2.DataAuditResult

		require.Nil(t, audit.NewResultFromV2(x))
	})
}