frostfs-api-go-pogpp/v2/reputation/test/generate.go
Leonard Lyubich 101d14d405 [#265] reputation: Implement test package with message generators
Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
2021-03-25 11:20:34 +03:00

66 lines
1.5 KiB
Go

package reputationtest
import (
"github.com/nspcc-dev/neofs-api-go/v2/reputation"
sessiontest "github.com/nspcc-dev/neofs-api-go/v2/session/test"
)
func GenerateTrust(empty bool) *reputation.Trust {
m := new(reputation.Trust)
if !empty {
m.SetPeer([]byte{1, 2, 3})
m.SetValue(1)
}
return m
}
func GenerateTrusts(empty bool) (res []*reputation.Trust) {
if !empty {
res = append(res,
GenerateTrust(false),
GenerateTrust(false),
)
}
return
}
func GenerateSendLocalTrustRequestBody(empty bool) *reputation.SendLocalTrustRequestBody {
m := new(reputation.SendLocalTrustRequestBody)
if !empty {
m.SetEpoch(13)
}
m.SetTrusts(GenerateTrusts(empty))
return m
}
func GenerateSendLocalTrustRequest(empty bool) *reputation.SendLocalTrustRequest {
m := new(reputation.SendLocalTrustRequest)
m.SetBody(GenerateSendLocalTrustRequestBody(empty))
m.SetMetaHeader(sessiontest.GenerateRequestMetaHeader(empty))
m.SetVerificationHeader(sessiontest.GenerateRequestVerificationHeader(empty))
return m
}
func GenerateSendLocalTrustResponseBody(empty bool) *reputation.SendLocalTrustResponseBody {
m := new(reputation.SendLocalTrustResponseBody)
return m
}
func GenerateSendLocalTrustResponse(empty bool) *reputation.SendLocalTrustResponse {
m := new(reputation.SendLocalTrustResponse)
m.SetBody(GenerateSendLocalTrustResponseBody(empty))
m.SetMetaHeader(sessiontest.GenerateResponseMetaHeader(empty))
m.SetVerificationHeader(sessiontest.GenerateResponseVerificationHeader(empty))
return m
}