From 81f537cda88329272b2d1cfe71eb2e9a7fec11d8 Mon Sep 17 00:00:00 2001
From: Leonard Lyubich <leonard@nspcc.ru>
Date: Mon, 11 May 2020 13:20:58 +0300
Subject: [PATCH] accounting: implement SignedDataSource on ListRequest message

---
 accounting/sign.go       | 30 ++++++++++++++++++++++++++++++
 accounting/sign_test.go  | 15 +++++++++++++++
 accounting/types.go      | 10 ++++++++++
 accounting/types_test.go |  9 +++++++++
 4 files changed, 64 insertions(+)

diff --git a/accounting/sign.go b/accounting/sign.go
index 8da8cf6e..dc8f6d77 100644
--- a/accounting/sign.go
+++ b/accounting/sign.go
@@ -122,3 +122,33 @@ func (m PutRequest) ReadSignedData(p []byte) (int, error) {
 
 	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
+}
diff --git a/accounting/sign_test.go b/accounting/sign_test.go
index 1f88dcfc..77ee0147 100644
--- a/accounting/sign_test.go
+++ b/accounting/sign_test.go
@@ -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 {
diff --git a/accounting/types.go b/accounting/types.go
index 3a4b15ea..8ea52e3f 100644
--- a/accounting/types.go
+++ b/accounting/types.go
@@ -411,3 +411,13 @@ func (m *PutRequest) SetAmount(amount *decimal.Decimal) {
 func (m *PutRequest) SetHeight(h uint64) {
 	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
+}
diff --git a/accounting/types_test.go b/accounting/types_test.go
index 844ea70b..cdf56104 100644
--- a/accounting/types_test.go
+++ b/accounting/types_test.go
@@ -152,3 +152,12 @@ func TestPutRequestGettersSetters(t *testing.T) {
 		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())
+}