Fix the build failure

iotexproject · Nov 28, 2018 · 034d8f7 · 034d8f7
1 parent e93e3bd
commit 034d8f7
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 1 deletion.
diff --git a/blocksync/blocksync.go b/blocksync/blocksync.go
@@ -26,6 +26,7 @@ import (
 type BlockSync interface {
 	lifecycle.StartStopper
 
+	TargetHeight() uint64
 	P2P() network.Overlay
 	ProcessSyncRequest(sender string, sync *pb.BlockSync) error
 	ProcessBlock(blk *blockchain.Block) error
@@ -75,6 +76,11 @@ func NewBlockSyncer(
 	return bs, nil
 }
 
+// TargetHeight returns the target height to sync to
+func (bs *blockSyncer) TargetHeight() uint64 {
+	return bs.worker.targetHeight
+}
+
 // P2P returns the network overlay object
 func (bs *blockSyncer) P2P() network.Overlay {
 	return bs.p2p

diff --git a/blocksync/blocksync_test.go b/blocksync/blocksync_test.go
@@ -463,7 +463,7 @@ func TestBlockSyncerChaser(t *testing.T) {
 	require.NoError(chain.Start(ctx))
 	ap, err := actpool.NewActPool(chain, cfg.ActPool)
 	require.NoError(err)
-	bs, err := NewBlockSyncer(cfg, chain, ap, network.NewOverlay(cfg.Network))
+	bs, err := NewBlockSyncer(cfg, chain, ap, network.NewOverlay(&cfg.Network))
 	require.NoError(err)
 	require.NoError(bs.Start(ctx))
 

diff --git a/test/mock/mock_blocksync/mock_blocksync.go b/test/mock/mock_blocksync/mock_blocksync.go