diff options
Diffstat (limited to 'core/dkg-tsig-protocol_test.go')
| -rw-r--r-- | core/dkg-tsig-protocol_test.go | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/core/dkg-tsig-protocol_test.go b/core/dkg-tsig-protocol_test.go index 89fd105..184460b 100644 --- a/core/dkg-tsig-protocol_test.go +++ b/core/dkg-tsig-protocol_test.go @@ -50,6 +50,7 @@ type testDKGReceiver struct { antiComplaints map[types.NodeID]*typesDKG.PrivateShare ready []*typesDKG.MPKReady final []*typesDKG.Finalize + success []*typesDKG.Success } func newTestDKGReceiver(s *DKGTSIGProtocolTestSuite, @@ -102,6 +103,10 @@ func (r *testDKGReceiver) ProposeDKGFinalize(final *typesDKG.Finalize) { r.final = append(r.final, final) } +func (r *testDKGReceiver) ProposeDKGSuccess(success *typesDKG.Success) { + r.success = append(r.success, success) +} + func (s *DKGTSIGProtocolTestSuite) setupDKGParticipants(n int) { s.nIDs = make(types.NodeIDs, 0, n) s.signers = make(map[types.NodeID]*utils.Signer, n) @@ -878,6 +883,22 @@ func (s *DKGTSIGProtocolTestSuite) TestProposeFinalize() { }, final) } +func (s *DKGTSIGProtocolTestSuite) TestProposeSuccess() { + prvKey, err := ecdsa.NewPrivateKey() + s.Require().NoError(err) + recv := newTestDKGReceiver(s, utils.NewSigner(prvKey)) + nID := types.NewNodeID(prvKey.PublicKey()) + protocol := newDKGProtocol(nID, recv, 1, 3, 2) + protocol.proposeSuccess() + s.Require().Len(recv.success, 1) + success := recv.success[0] + s.Equal(&typesDKG.Success{ + ProposerID: nID, + Round: 1, + Reset: 3, + }, success) +} + func (s *DKGTSIGProtocolTestSuite) TestTSigVerifierCache() { k := 3 n := 10 |