From 5fc0efa940c7663a33d0fc501807a2627d2cb573 Mon Sep 17 00:00:00 2001 From: Mission Liao Date: Fri, 28 Sep 2018 12:32:50 +0800 Subject: core: hide types.NodeID from full node. (#147) * Refine core.Governance interface - Remove types.NodeID from interface declaration. - All parameter should be round based. * Add core.NodeSetCache * Agreement accepts map of nodeID directly. * test.Transport.Peers method return public keys. --- core/agreement_test.go | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'core/agreement_test.go') 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}, -- cgit