accounting: implement SignedDataSource on ListRequest message

This commit is contained in:
Leonard Lyubich 2020-05-11 13:20:58 +03:00
parent 8c492a7712
commit 81f537cda8
4 changed files with 64 additions and 0 deletions

View file

@ -122,3 +122,33 @@ func (m PutRequest) ReadSignedData(p []byte) (int, error) {
return off, nil return off, nil
} }
// SignedData returns payload bytes of the request.
func (m ListRequest) SignedData() ([]byte, error) {
data := make([]byte, m.SignedDataSize())
if _, err := m.ReadSignedData(data); err != nil {
return nil, err
}
return data, nil
}
// SignedDataSize returns payload size of the request.
func (m ListRequest) SignedDataSize() int {
return m.GetOwnerID().Size()
}
// ReadSignedData copies payload bytes to passed buffer.
//
// If the buffer size is insufficient, io.ErrUnexpectedEOF returns.
func (m ListRequest) ReadSignedData(p []byte) (int, error) {
sz := m.SignedDataSize()
if len(p) < sz {
return 0, io.ErrUnexpectedEOF
}
copy(p, m.GetOwnerID().Bytes())
return sz, nil
}

View file

@ -104,6 +104,21 @@ func TestSignBalanceRequest(t *testing.T) {
}, },
}, },
}, },
{ // 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 { for _, item := range items {

View file

@ -411,3 +411,13 @@ func (m *PutRequest) SetAmount(amount *decimal.Decimal) {
func (m *PutRequest) SetHeight(h uint64) { func (m *PutRequest) SetHeight(h uint64) {
m.Height = h m.Height = h
} }
// GetOwnerID is an OwnerID field getter.
func (m ListRequest) GetOwnerID() OwnerID {
return m.OwnerID
}
// SetOwnerID is an OwnerID field setter.
func (m *ListRequest) SetOwnerID(id OwnerID) {
m.OwnerID = id
}

View file

@ -152,3 +152,12 @@ func TestPutRequestGettersSetters(t *testing.T) {
require.Equal(t, h, m.GetHeight()) require.Equal(t, h, m.GetHeight())
}) })
} }
func TestListRequestGettersSetters(t *testing.T) {
ownerID := OwnerID{1, 2, 3}
m := new(ListRequest)
m.SetOwnerID(ownerID)
require.Equal(t, ownerID, m.GetOwnerID())
}