[#293] pkg/audit: Implement and use generator of Result
Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
This commit is contained in:
parent
00778cc9ba
commit
cd32722d25
2 changed files with 41 additions and 15 deletions
|
@ -5,6 +5,7 @@ import (
|
|||
|
||||
"github.com/nspcc-dev/neofs-api-go/pkg"
|
||||
"github.com/nspcc-dev/neofs-api-go/pkg/audit"
|
||||
audittest "github.com/nspcc-dev/neofs-api-go/pkg/audit/test"
|
||||
cidtest "github.com/nspcc-dev/neofs-api-go/pkg/container/id/test"
|
||||
"github.com/nspcc-dev/neofs-api-go/pkg/object"
|
||||
objecttest "github.com/nspcc-dev/neofs-api-go/pkg/object/test"
|
||||
|
@ -38,11 +39,11 @@ func TestResult(t *testing.T) {
|
|||
r.SetRetries(retries)
|
||||
require.Equal(t, retries, r.Retries())
|
||||
|
||||
passSG := []*object.ID{objecttest.GenerateID(), objecttest.GenerateID()}
|
||||
passSG := []*object.ID{objecttest.ID(), objecttest.ID()}
|
||||
r.SetPassSG(passSG)
|
||||
require.Equal(t, passSG, r.PassSG())
|
||||
|
||||
failSG := []*object.ID{objecttest.GenerateID(), objecttest.GenerateID()}
|
||||
failSG := []*object.ID{objecttest.ID(), objecttest.ID()}
|
||||
r.SetFailSG(failSG)
|
||||
require.Equal(t, failSG, r.FailSG())
|
||||
|
||||
|
@ -68,19 +69,7 @@ func TestResult(t *testing.T) {
|
|||
}
|
||||
|
||||
func TestStorageGroupEncoding(t *testing.T) {
|
||||
r := audit.NewResult()
|
||||
r.SetAuditEpoch(13)
|
||||
r.SetContainerID(cidtest.Generate())
|
||||
r.SetPublicKey([]byte{1, 2, 3})
|
||||
r.SetPassSG([]*object.ID{objecttest.GenerateID(), objecttest.GenerateID()})
|
||||
r.SetFailSG([]*object.ID{objecttest.GenerateID(), objecttest.GenerateID()})
|
||||
r.SetRequests(3)
|
||||
r.SetRetries(2)
|
||||
r.SetHit(1)
|
||||
r.SetMiss(2)
|
||||
r.SetFail(3)
|
||||
r.SetPassNodes([][]byte{{1}, {2}})
|
||||
r.SetFailNodes([][]byte{{3}, {4}})
|
||||
r := audittest.Generate()
|
||||
|
||||
t.Run("binary", func(t *testing.T) {
|
||||
data, err := r.Marshal()
|
||||
|
|
37
pkg/audit/test/generate.go
Normal file
37
pkg/audit/test/generate.go
Normal file
|
@ -0,0 +1,37 @@
|
|||
package audittest
|
||||
|
||||
import (
|
||||
"github.com/nspcc-dev/neofs-api-go/pkg/audit"
|
||||
cidtest "github.com/nspcc-dev/neofs-api-go/pkg/container/id/test"
|
||||
"github.com/nspcc-dev/neofs-api-go/pkg/object"
|
||||
objecttest "github.com/nspcc-dev/neofs-api-go/pkg/object/test"
|
||||
refstest "github.com/nspcc-dev/neofs-api-go/pkg/test"
|
||||
)
|
||||
|
||||
// Generate returns random audit.Result.
|
||||
func Generate() *audit.Result {
|
||||
x := audit.NewResult()
|
||||
|
||||
x.SetVersion(refstest.Version())
|
||||
x.SetContainerID(cidtest.Generate())
|
||||
x.SetPublicKey([]byte("key"))
|
||||
x.SetComplete(true)
|
||||
x.SetAuditEpoch(44)
|
||||
x.SetHit(55)
|
||||
x.SetMiss(66)
|
||||
x.SetFail(77)
|
||||
x.SetRetries(88)
|
||||
x.SetRequests(99)
|
||||
x.SetFailNodes([][]byte{
|
||||
[]byte("node1"),
|
||||
[]byte("node2"),
|
||||
})
|
||||
x.SetPassNodes([][]byte{
|
||||
[]byte("node3"),
|
||||
[]byte("node4"),
|
||||
})
|
||||
x.SetPassSG([]*object.ID{objecttest.ID(), objecttest.ID()})
|
||||
x.SetFailSG([]*object.ID{objecttest.ID(), objecttest.ID()})
|
||||
|
||||
return x
|
||||
}
|
Loading…
Reference in a new issue