154 lines
3.6 KiB
Go
154 lines
3.6 KiB
Go
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"
|
|
oid "github.com/nspcc-dev/neofs-sdk-go/object/id"
|
|
oidtest "github.com/nspcc-dev/neofs-sdk-go/object/id/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 := []*oid.ID{oidtest.ID(), oidtest.ID()}
|
|
r.SetPassSG(passSG)
|
|
require.Equal(t, passSG, r.PassSG())
|
|
|
|
failSG := []*oid.ID{oidtest.ID(), oidtest.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))
|
|
})
|
|
}
|