core: Add unit test for DKG CRS in consensus (#163)

1 files changed, 13 insertions, 1 deletions

diff --git a/core/test/governance.go b/core/test/governance.go
index 473c331..5c7ae02 100644
--- a/core/test/governance.go
+++ b/core/test/governance.go
@@ -18,6 +18,7 @@
 package test
 
 import (
+	"encoding/json"
 	"fmt"
 	"sync"
 	"time"
@@ -106,11 +107,15 @@ func (g *Governance) Configuration(_ uint64) *types.Config {
 
 // CRS returns the CRS for a given round.
 func (g *Governance) CRS(round uint64) common.Hash {
+	g.lock.RLock()
+	defer g.lock.RUnlock()
 	return g.crs[round]
 }
 
 // ProposeCRS propose a CRS.
 func (g *Governance) ProposeCRS(round uint64, signedCRS []byte) {
+	g.lock.Lock()
+	defer g.lock.Unlock()
 	g.crs[round] = crypto.Keccak256Hash(signedCRS)
 }
 
@@ -165,5 +170,12 @@ func (g *Governance) DKGMasterPublicKeys(
 	if !exist {
 		return []*types.DKGMasterPublicKey{}
 	}
-	return masterPublicKeys
+	mpks := make([]*types.DKGMasterPublicKey, 0, len(masterPublicKeys))
+	for _, mpk := range masterPublicKeys {
+		bytes, _ := json.Marshal(mpk)
+		mpkCopy := types.NewDKGMasterPublicKey()
+		json.Unmarshal(bytes, mpkCopy)
+		mpks = append(mpks, mpkCopy)
+	}
+	return mpks
 }