diff options
Diffstat (limited to 'core/test/network_test.go')
| -rw-r--r-- | core/test/network_test.go | 20 |
1 files changed, 4 insertions, 16 deletions
diff --git a/core/test/network_test.go b/core/test/network_test.go index 63407d6..993ae70 100644 --- a/core/test/network_test.go +++ b/core/test/network_test.go @@ -251,43 +251,31 @@ func (s *NetworkTestSuite) TestBroadcastToSet() { gov, err := NewGovernance(NewState( 1, pubKeys, time.Second, &common.NullLogger{}, true), 2) req.NoError(err) - req.NoError(gov.State().RequestChange(StateChangeDKGSetSize, uint32(1))) req.NoError(gov.State().RequestChange(StateChangeNotarySetSize, uint32(1))) gov.NotifyRound(round, gov.Configuration(0).RoundLength) networks := s.setupNetworks(pubKeys) cache := utils.NewNodeSetCache(gov) // Cache required set of nodeIDs. - dkgSet, err := cache.GetDKGSet(round) - req.NoError(err) - req.Len(dkgSet, 1) notarySet, err := cache.GetNotarySet(round) req.NoError(err) req.Len(notarySet, 1) var ( // Some node don't belong to any set. - nerd *Network - dkgNode, notaryNode *Network + nerd *Network + notaryNode *Network ) for nID, n := range networks { - if _, exists := dkgSet[nID]; exists { - continue - } if _, exists := notarySet[nID]; exists { continue } nerd = n break } - for nID := range dkgSet { - dkgNode = networks[nID] - break - } for nID := range notarySet { notaryNode = networks[nID] break } req.NotNil(nerd) - req.NotNil(dkgNode) req.NotNil(notaryNode) nerd.AttachNodeSetCache(cache) // Try broadcasting with datum from round 0, and make sure only node belongs @@ -296,9 +284,9 @@ func (s *NetworkTestSuite) TestBroadcastToSet() { Position: types.Position{Round: round}}}) req.IsType(&types.Vote{}, <-notaryNode.ReceiveChan()) nerd.BroadcastDKGPrivateShare(&typesDKG.PrivateShare{Round: round}) - req.IsType(&typesDKG.PrivateShare{}, <-dkgNode.ReceiveChan()) + req.IsType(&typesDKG.PrivateShare{}, <-notaryNode.ReceiveChan()) nerd.BroadcastDKGPartialSignature(&typesDKG.PartialSignature{Round: round}) - req.IsType(&typesDKG.PartialSignature{}, <-dkgNode.ReceiveChan()) + req.IsType(&typesDKG.PartialSignature{}, <-notaryNode.ReceiveChan()) nerd.BroadcastBlock(&types.Block{Position: types.Position{Round: round}}) req.IsType(&types.Block{}, <-notaryNode.ReceiveChan()) } |