From 90d2f61b919eb9ec077b30b99867b5509850916a Mon Sep 17 00:00:00 2001
From: joanestebanr <129153821+joanestebanr@users.noreply.github.com>
Date: Mon, 22 Jul 2024 09:53:24 +0200
Subject: [PATCH] add Set and Get for Signature of types SignedSequence

---
 types/interfaces.go          |  2 ++
 types/sequence.go            | 10 ++++++++++
 types/sequence_test.go       |  7 +++++++
 types/sequencebanana.go      | 10 ++++++++++
 types/sequencebanana_test.go | 14 ++++++++++++++
 5 files changed, 43 insertions(+)
 create mode 100644 types/sequencebanana_test.go

diff --git a/types/interfaces.go b/types/interfaces.go
index 450937ae..00c8f294 100644
--- a/types/interfaces.go
+++ b/types/interfaces.go
@@ -11,4 +11,6 @@ type SignedSequenceInterface interface {
 	Signer() (common.Address, error)
 	OffChainData() []OffChainData
 	Sign(privateKey *ecdsa.PrivateKey) (ArgBytes, error)
+	SetSignature([]byte)
+	GetSignature() []byte
 }
diff --git a/types/sequence.go b/types/sequence.go
index d410b60e..a8ad10a1 100644
--- a/types/sequence.go
+++ b/types/sequence.go
@@ -84,3 +84,13 @@ func (s *SignedSequence) OffChainData() []OffChainData {
 func (s *SignedSequence) Sign(privateKey *ecdsa.PrivateKey) (ArgBytes, error) {
 	return s.Sequence.Sign(privateKey)
 }
+
+// SetSignature set signature
+func (s *SignedSequence) SetSignature(sign []byte) {
+	s.Signature = sign
+}
+
+// GetSignature returns signature
+func (s *SignedSequence) GetSignature() []byte {
+	return s.Signature
+}
diff --git a/types/sequence_test.go b/types/sequence_test.go
index 79c9b13c..1e8eaa66 100644
--- a/types/sequence_test.go
+++ b/types/sequence_test.go
@@ -56,3 +56,10 @@ func TestSigning(t *testing.T) {
 		}
 	}
 }
+
+func TestGetSetSignature(t *testing.T) {
+	sut := SignedSequence{}
+	signature := []byte{1, 2, 3}
+	sut.SetSignature(signature)
+	assert.Equal(t, signature, sut.GetSignature())
+}
diff --git a/types/sequencebanana.go b/types/sequencebanana.go
index 153b2297..605f7cd0 100644
--- a/types/sequencebanana.go
+++ b/types/sequencebanana.go
@@ -114,3 +114,13 @@ func (s *SignedSequenceBanana) OffChainData() []OffChainData {
 func (s *SignedSequenceBanana) Sign(privateKey *ecdsa.PrivateKey) (ArgBytes, error) {
 	return s.Sequence.Sign(privateKey)
 }
+
+// SetSignature set signature
+func (s *SignedSequenceBanana) SetSignature(sign []byte) {
+	s.Signature = sign
+}
+
+// GetSignature returns signature
+func (s *SignedSequenceBanana) GetSignature() []byte {
+	return s.Signature
+}
diff --git a/types/sequencebanana_test.go b/types/sequencebanana_test.go
new file mode 100644
index 00000000..349cb5aa
--- /dev/null
+++ b/types/sequencebanana_test.go
@@ -0,0 +1,14 @@
+package types
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestGetSetSignatureBanana(t *testing.T) {
+	sut := SignedSequenceBanana{}
+	signature := []byte{1, 2, 3}
+	sut.SetSignature(signature)
+	assert.Equal(t, signature, sut.GetSignature())
+}