From 9ddebfdb8aa4dd835a06daf7e3cab1e45a736a73 Mon Sep 17 00:00:00 2001
From: Evgenii Stratonikov <evgeniy@nspcc.ru>
Date: Mon, 2 Dec 2019 17:00:17 +0300
Subject: [PATCH] consensus: add tests for recovery message

---
 pkg/consensus/recovery_message_test.go | 80 ++++++++++++++++++++++++++
 1 file changed, 80 insertions(+)
 create mode 100644 pkg/consensus/recovery_message_test.go

diff --git a/pkg/consensus/recovery_message_test.go b/pkg/consensus/recovery_message_test.go
new file mode 100644
index 000000000..438235d3d
--- /dev/null
+++ b/pkg/consensus/recovery_message_test.go
@@ -0,0 +1,80 @@
+package consensus
+
+import (
+	"testing"
+
+	"github.com/CityOfZion/neo-go/pkg/util"
+	"github.com/nspcc-dev/dbft/payload"
+	"github.com/stretchr/testify/require"
+)
+
+func TestRecoveryMessage_Setters(t *testing.T) {
+	r := &recoveryMessage{}
+	p := new(Payload)
+	p.SetType(payload.RecoveryMessageType)
+	p.SetPayload(r)
+
+	req := &prepareRequest{
+		timestamp:         87,
+		nonce:             321,
+		transactionHashes: []util.Uint256{{1}},
+		minerTx:           *newMinerTx(123),
+		nextConsensus:     util.Uint160{1, 2},
+	}
+	p1 := new(Payload)
+	p1.SetType(payload.PrepareRequestType)
+	p1.SetPayload(req)
+
+	t.Run("prepare response is added", func(t *testing.T) {
+		p2 := new(Payload)
+		p2.SetType(payload.PrepareResponseType)
+		p2.SetPayload(&prepareResponse{
+			preparationHash: p1.Hash(),
+		})
+
+		r.AddPayload(p2)
+		require.NotNil(t, r.PreparationHash())
+		require.Equal(t, p1.Hash(), *r.PreparationHash())
+
+		ps := r.GetPrepareResponses(p)
+		require.Len(t, ps, 1)
+		require.Equal(t, p2, ps[0])
+	})
+
+	t.Run("prepare request is added", func(t *testing.T) {
+		pr := r.GetPrepareRequest(p)
+		require.Nil(t, pr)
+
+		r.AddPayload(p1)
+		pr = r.GetPrepareRequest(p)
+		require.NotNil(t, pr)
+		require.Equal(t, p1, pr)
+	})
+
+	t.Run("change view is added", func(t *testing.T) {
+		p3 := new(Payload)
+		p3.SetType(payload.ChangeViewType)
+		p3.SetPayload(&changeView{
+			newViewNumber: 1,
+			timestamp:     12345,
+		})
+
+		r.AddPayload(p3)
+
+		ps := r.GetChangeViews(p)
+		require.Len(t, ps, 1)
+		require.Equal(t, p3, ps[0])
+	})
+
+	t.Run("commit is added", func(t *testing.T) {
+		p4 := new(Payload)
+		p4.SetType(payload.CommitType)
+		p4.SetPayload(randomMessage(t, commitType))
+
+		r.AddPayload(p4)
+
+		ps := r.GetCommits(p)
+		require.Len(t, ps, 1)
+		require.Equal(t, p4, ps[0])
+	})
+}