core: Add DKGSuccess (#569)

* core: Add DKGSuccess

* core: reset if not enough of dkg success

1 files changed, 13 insertions, 0 deletions

diff --git a/core/configuration-chain_test.go b/core/configuration-chain_test.go
index 61f0906..edaacf6 100644
--- a/core/configuration-chain_test.go
+++ b/core/configuration-chain_test.go
@@ -114,6 +114,12 @@ func (r *testCCGlobalReceiver) ProposeDKGFinalize(final *typesDKG.Finalize) {
 	}
 }
 
+func (r *testCCGlobalReceiver) ProposeDKGSuccess(success *typesDKG.Success) {
+	for _, gov := range r.govs {
+		gov.AddDKGSuccess(test.CloneDKGSuccess(success))
+	}
+}
+
 type testCCReceiver struct {
 	signer *utils.Signer
 	recv   *testCCGlobalReceiver
@@ -176,6 +182,13 @@ func (r *testCCReceiver) ProposeDKGFinalize(final *typesDKG.Finalize) {
 	r.recv.ProposeDKGFinalize(final)
 }
 
+func (r *testCCReceiver) ProposeDKGSuccess(success *typesDKG.Success) {
+	if err := r.signer.SignDKGSuccess(success); err != nil {
+		panic(err)
+	}
+	r.recv.ProposeDKGSuccess(success)
+}
+
 func (s *ConfigurationChainTestSuite) setupNodes(n int) {
 	s.nIDs = make(types.NodeIDs, 0, n)
 	s.signers = make(map[types.NodeID]*utils.Signer, n)