From 942bedb8ed92fbb6587a19cbcbc1f79377260080 Mon Sep 17 00:00:00 2001
From: Leonard Lyubich <leonard@nspcc.ru>
Date: Tue, 28 Apr 2020 16:02:40 +0300
Subject: [PATCH] service: implement Token field setter on
 RequestVerificationHeader

After recent changes Token field is presented in RequestVerificationHeader.
There is a need to provide an interface of field getter/setter.

This commit:

  * defines TokenHeader interface of token value container;

  * implements Token field setter on RequestVerificationHeader.
---
 service/verify.go      | 11 +++++++++++
 service/verify_test.go | 14 ++++++++++++++
 2 files changed, 25 insertions(+)

diff --git a/service/verify.go b/service/verify.go
index ade13ef..182685d 100644
--- a/service/verify.go
+++ b/service/verify.go
@@ -27,6 +27,12 @@ type (
 		SetOwner(*ecdsa.PublicKey, []byte)
 		GetLastPeer() (*ecdsa.PublicKey, error)
 	}
+
+	// TokenHeader is an interface of the container of a Token pointer value
+	TokenHeader interface {
+		GetToken() *Token
+		SetToken(*Token)
+	}
 )
 
 const (
@@ -97,6 +103,11 @@ func (m *RequestVerificationHeader) GetLastPeer() (*ecdsa.PublicKey, error) {
 	}
 }
 
+// SetToken is a Token field setter.
+func (m *RequestVerificationHeader) SetToken(token *Token) {
+	m.Token = token
+}
+
 func newSignature(key *ecdsa.PrivateKey, data []byte) (*RequestVerificationHeader_Signature, error) {
 	sign, err := crypto.Sign(key, data)
 	if err != nil {
diff --git a/service/verify_test.go b/service/verify_test.go
index ce333aa..c192f0f 100644
--- a/service/verify_test.go
+++ b/service/verify_test.go
@@ -188,3 +188,17 @@ func TestVerifyAndSignRequestHeaderWithoutCloning(t *testing.T) {
 
 	require.Contains(t, buf.String(), "proto: don't know how to copy")
 }
+
+func TestRequestVerificationHeader_SetToken(t *testing.T) {
+	id, err := refs.NewUUID()
+	require.NoError(t, err)
+
+	token := new(Token)
+	token.ID = id
+
+	h := new(RequestVerificationHeader)
+
+	h.SetToken(token)
+
+	require.Equal(t, token, h.GetToken())
+}