From daf3bab93c323b173345811adc9a334dad4a7094 Mon Sep 17 00:00:00 2001
From: haoping-ku <haoping.ku@dexon.org>
Date: Thu, 29 Nov 2018 14:30:02 +0800
Subject: core: syncer: add syncer (#346)

---
 core/test/revealer_test.go | 60 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)

(limited to 'core/test/revealer_test.go')

diff --git a/core/test/revealer_test.go b/core/test/revealer_test.go
index d9920c7..5a1bc07 100644
--- a/core/test/revealer_test.go
+++ b/core/test/revealer_test.go
@@ -157,6 +157,66 @@ func (s *RevealerTestSuite) TestRandomTipReveal() {
 	s.baseTest(revealer, 10, checkFunc)
 }
 
+func (s *RevealerTestSuite) TestCompactionChainReveal() {
+	db, err := blockdb.NewMemBackedBlockDB()
+	s.Require().NoError(err)
+	// Put several blocks with finalization field ready.
+	b1 := &types.Block{
+		Hash: common.NewRandomHash(),
+		Finalization: types.FinalizationResult{
+			Height: 1,
+		}}
+	b2 := &types.Block{
+		Hash: common.NewRandomHash(),
+		Finalization: types.FinalizationResult{
+			ParentHash: b1.Hash,
+			Height:     2,
+		}}
+	b3 := &types.Block{
+		Hash: common.NewRandomHash(),
+		Finalization: types.FinalizationResult{
+			ParentHash: b2.Hash,
+			Height:     3,
+		}}
+	s.Require().NoError(db.Put(*b1))
+	s.Require().NoError(db.Put(*b3))
+	iter, err := db.GetAll()
+	s.Require().NoError(err)
+	// The compaction chain is not complete, we can't construct a revealer
+	// instance successfully.
+	r, err := NewCompactionChainRevealer(iter, 0)
+	s.Require().Nil(r)
+	s.Require().IsType(ErrNotValidCompactionChain, err)
+	// Put a block to make the compaction chain complete.
+	s.Require().NoError(db.Put(*b2))
+	// We can construct that revealer now.
+	iter, err = db.GetAll()
+	s.Require().NoError(err)
+	r, err = NewCompactionChainRevealer(iter, 0)
+	s.Require().NotNil(r)
+	s.Require().NoError(err)
+	// The revealing order should be ok.
+	chk := func(h uint64) {
+		b, err := r.Next()
+		s.Require().NoError(err)
+		s.Require().Equal(b.Finalization.Height, h)
+	}
+	chk(1)
+	chk(2)
+	chk(3)
+	// Iteration should be finished
+	_, err = r.Next()
+	s.Require().IsType(blockdb.ErrIterationFinished, err)
+	// Test 'startHeight' parameter.
+	iter, err = db.GetAll()
+	s.Require().NoError(err)
+	r, err = NewCompactionChainRevealer(iter, 2)
+	s.Require().NotNil(r)
+	s.Require().NoError(err)
+	chk(2)
+	chk(3)
+}
+
 func TestRevealer(t *testing.T) {
 	suite.Run(t, new(RevealerTestSuite))
 }
-- 
cgit v1.2.3