core: Add DKG final message. (#181)

1 files changed, 19 insertions, 0 deletions

diff --git a/core/dkg-tsig-protocol_test.go b/core/dkg-tsig-protocol_test.go
index a3a0acc..9d66222 100644
--- a/core/dkg-tsig-protocol_test.go
+++ b/core/dkg-tsig-protocol_test.go
@@ -46,6 +46,7 @@ type testDKGReceiver struct {
 	mpk            *types.DKGMasterPublicKey
 	prvShare       map[types.NodeID]*types.DKGPrivateShare
 	antiComplaints map[types.NodeID]*types.DKGPrivateShare
+	final          []*types.DKGFinalize
 }
 
 func newTestDKGReceiver(
@@ -90,6 +91,10 @@ func (r *testDKGReceiver) ProposeDKGAntiNackComplaint(
 	r.antiComplaints[prv.ReceiverID] = prv
 }
 
+func (r *testDKGReceiver) ProposeDKGFinalize(final *types.DKGFinalize) {
+	r.final = append(r.final, final)
+}
+
 func (s *DKGTSIGProtocolTestSuite) setupDKGParticipants(n int) {
 	s.nIDs = make(types.NodeIDs, 0, n)
 	s.prvKeys = make(map[types.NodeID]crypto.PrivateKey, n)
@@ -616,6 +621,20 @@ func (s *DKGTSIGProtocolTestSuite) TestPartialSignature() {
 	s.True(gpk.VerifySignature(msgHash, sig))
 }
 
+func (s *DKGTSIGProtocolTestSuite) TestProposeFinalize() {
+	prvKey, err := ecdsa.NewPrivateKey()
+	s.Require().NoError(err)
+	recv := newTestDKGReceiver(s, prvKey)
+	nID := types.NewNodeID(prvKey.PublicKey())
+	protocol := newDKGProtocol(nID, recv, 1, 2)
+	protocol.proposeFinalize()
+	s.Require().Len(recv.final, 1)
+	final := recv.final[0]
+	s.Equal(&types.DKGFinalize{
+		ProposerID: nID,
+		Round:      1,
+	}, final)
+}
 func TestDKGTSIGProtocol(t *testing.T) {
 	suite.Run(t, new(DKGTSIGProtocolTestSuite))
 }