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.ID() 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.Result() 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)) }) }