From 79be89a6b0b1d24b889e7c9fe0244026af4d49d0 Mon Sep 17 00:00:00 2001
From: Jimmy Hu <jimmy.hu@dexon.org>
Date: Mon, 15 Apr 2019 11:44:04 +0800
Subject: core: Add DKGSuccess (#569)

* core: Add DKGSuccess

* core: reset if not enough of dkg success
---
 core/configuration-chain_test.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

(limited to 'core/configuration-chain_test.go')

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)
-- 
cgit