core: fast forward should close previous done() channel (#444)

* core: fast forward faster

* core: modify unit test

1 files changed, 12 insertions, 0 deletions

diff --git a/core/agreement_test.go b/core/agreement_test.go
index f49fd7d..4ddff8d 100644
--- a/core/agreement_test.go
+++ b/core/agreement_test.go
@@ -404,6 +404,7 @@ func (s *AgreementTestSuite) TestFastForwardCond2() {
 		return true, nil
 	})
 	a.data.period = 1
+	done := a.done()
 	hash := common.NewRandomHash()
 	for nID := range a.notarySet {
 		vote := s.prepareVote(nID, types.VotePreCom, hash, uint64(2))
@@ -414,6 +415,11 @@ func (s *AgreementTestSuite) TestFastForwardCond2() {
 	}
 
 	select {
+	case <-done:
+	default:
+		s.FailNow("Expecting fast forward for pending done() call.")
+	}
+	select {
 	case <-a.done():
 	default:
 		s.FailNow("Expecting fast forward.")
@@ -443,6 +449,7 @@ func (s *AgreementTestSuite) TestFastForwardCond3() {
 		return true, nil
 	})
 	a.data.period = 1
+	done := a.done()
 	for nID := range a.notarySet {
 		vote := s.prepareVote(nID, types.VoteCom, common.NewRandomHash(), uint64(2))
 		votes = append(votes, vote)
@@ -453,6 +460,11 @@ func (s *AgreementTestSuite) TestFastForwardCond3() {
 	}
 
 	select {
+	case <-done:
+	default:
+		s.FailNow("Expecting fast forward for pending done() call.")
+	}
+	select {
 	case <-a.done():
 	default:
 		s.FailNow("Expecting fast forward.")