diff options
Diffstat (limited to 'core/test/governance.go')
-rw-r--r-- | core/test/governance.go | 10 |
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) } } |