core: integrate authenticator (#150)

1 files changed, 2 insertions, 14 deletions

diff --git a/core/leader-selector_test.go b/core/leader-selector_test.go
index 8da7e2a..b5dfe91 100644
--- a/core/leader-selector_test.go
+++ b/core/leader-selector_test.go
@@ -88,7 +88,8 @@ func (s *LeaderSelectorTestSuite) TestLeaderBlockHash() {
 			ProposerID: types.NewNodeID(prv.PublicKey()),
 			Hash:       common.NewRandomHash(),
 		}
-		s.Require().Nil(leader.prepareBlock(block, prv))
+		s.Require().NoError(
+			NewAuthenticator(prv).SignCRS(block, leader.hashCRS))
 		s.Require().Nil(leader.processBlock(block))
 		blocks[block.Hash] = block
 	}
@@ -105,19 +106,6 @@ func (s *LeaderSelectorTestSuite) TestLeaderBlockHash() {
 	}
 }
 
-func (s *LeaderSelectorTestSuite) TestPrepareBlock() {
-	leader := s.newLeader()
-	prv, err := ecdsa.NewPrivateKey()
-	s.Require().Nil(err)
-	block := &types.Block{
-		ProposerID: types.NewNodeID(prv.PublicKey()),
-	}
-	s.Require().Nil(leader.prepareBlock(block, prv))
-	s.Nil(leader.processBlock(block))
-	block.Position.Height++
-	s.Error(ErrIncorrectCRSSignature, leader.processBlock(block))
-}
-
 func TestLeaderSelector(t *testing.T) {
 	suite.Run(t, new(LeaderSelectorTestSuite))
 }