package accounting import ( "testing" "github.com/nspcc-dev/neofs-api-go/decimal" "github.com/nspcc-dev/neofs-api-go/service" "github.com/nspcc-dev/neofs-crypto/test" "github.com/stretchr/testify/require" ) func TestSignBalanceRequest(t *testing.T) { sk := test.DecodeKey(0) type sigType interface { service.SignedDataWithToken service.SignKeyPairAccumulator service.SignKeyPairSource SetToken(*service.Token) } items := []struct { constructor func() sigType payloadCorrupt []func(sigType) }{ { // BalanceRequest constructor: func() sigType { return new(BalanceRequest) }, payloadCorrupt: []func(sigType){ func(s sigType) { req := s.(*BalanceRequest) owner := req.GetOwnerID() owner[0]++ req.SetOwnerID(owner) }, }, }, { // GetRequest constructor: func() sigType { return new(GetRequest) }, payloadCorrupt: []func(sigType){ func(s sigType) { req := s.(*GetRequest) id, err := NewChequeID() require.NoError(t, err) req.SetID(id) }, func(s sigType) { req := s.(*GetRequest) id := req.GetOwnerID() id[0]++ req.SetOwnerID(id) }, }, }, { // PutRequest constructor: func() sigType { req := new(PutRequest) amount := decimal.New(1) req.SetAmount(amount) return req }, payloadCorrupt: []func(sigType){ func(s sigType) { req := s.(*PutRequest) owner := req.GetOwnerID() owner[0]++ req.SetOwnerID(owner) }, func(s sigType) { req := s.(*PutRequest) mid := req.GetMessageID() mid[0]++ req.SetMessageID(mid) }, func(s sigType) { req := s.(*PutRequest) req.SetHeight(req.GetHeight() + 1) }, func(s sigType) { req := s.(*PutRequest) amount := req.GetAmount() if amount == nil { req.SetAmount(decimal.New(0)) } else { req.SetAmount(amount.Add(decimal.New(amount.GetValue()))) } }, }, }, { // ListRequest constructor: func() sigType { return new(ListRequest) }, payloadCorrupt: []func(sigType){ func(s sigType) { req := s.(*ListRequest) owner := req.GetOwnerID() owner[0]++ req.SetOwnerID(owner) }, }, }, } for _, item := range items { { // token corruptions v := item.constructor() token := new(service.Token) v.SetToken(token) require.NoError(t, service.SignDataWithSessionToken(sk, v)) require.NoError(t, service.VerifyAccumulatedSignaturesWithToken(v)) token.SetSessionKey(append(token.GetSessionKey(), 1)) require.Error(t, service.VerifyAccumulatedSignaturesWithToken(v)) } { // payload corruptions for _, corruption := range item.payloadCorrupt { v := item.constructor() require.NoError(t, service.SignDataWithSessionToken(sk, v)) require.NoError(t, service.VerifyAccumulatedSignaturesWithToken(v)) corruption(v) require.Error(t, service.VerifyAccumulatedSignaturesWithToken(v)) } } } }