diff options
Diffstat (limited to 'core/agreement_test.go')
-rw-r--r-- | core/agreement_test.go | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/core/agreement_test.go b/core/agreement_test.go index 2c98181..d384dc6 100644 --- a/core/agreement_test.go +++ b/core/agreement_test.go @@ -86,14 +86,15 @@ func (s *AgreementTestSuite) newAgreement(numNotarySet int) *agreement { return s.proposeBlock(agreementIdx) } - notarySet := make(types.NodeIDs, numNotarySet-1) - for i := range notarySet { + notarySet := make(map[types.NodeID]struct{}) + for i := 0; i < numNotarySet-1; i++ { prvKey, err := ecdsa.NewPrivateKey() s.Require().Nil(err) - notarySet[i] = types.NewNodeID(prvKey.PublicKey()) - s.prvKey[notarySet[i]] = prvKey + nID := types.NewNodeID(prvKey.PublicKey()) + notarySet[nID] = struct{}{} + s.prvKey[nID] = prvKey } - notarySet = append(notarySet, s.ID) + notarySet[s.ID] = struct{}{} agreement := newAgreement( s.ID, &agreementTestReceiver{s}, |