aboutsummaryrefslogtreecommitdiffstats
path: root/core/test/governance.go
diff options
context:
space:
mode:
Diffstat (limited to 'core/test/governance.go')
-rw-r--r--core/test/governance.go10
1 files changed, 10 insertions, 0 deletions
diff --git a/core/test/governance.go b/core/test/governance.go
index e6b6232..f27139f 100644
--- a/core/test/governance.go
+++ b/core/test/governance.go
@@ -232,6 +232,16 @@ func (g *Governance) CatchUpWithRound(round uint64) {
config, nodeSet := g.stateModule.Snapshot()
g.configs = append(g.configs, config)
g.nodeSets = append(g.nodeSets, nodeSet)
+ if g.networkModule == nil {
+ continue
+ }
+ // Notify network module for new notary set.
+ round := uint64(len(g.configs)) - 1
+ notarySet := make(map[types.NodeID]struct{})
+ for _, k := range g.nodeSets[round] {
+ notarySet[types.NewNodeID(k)] = struct{}{}
+ }
+ g.networkModule.appendRoundSetting(round, notarySet)
}
}