diff options
Diffstat (limited to 'core/configuration-chain_test.go')
| -rw-r--r-- | core/configuration-chain_test.go | 13 |
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) |