diff --git a/types/interfaces.go b/types/interfaces.go index 450937a..00c8f29 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 d410b60..a8ad10a 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 79c9b13..1e8eaa6 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 153b229..605f7cd 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 0000000..349cb5a --- /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()) +}