feat: add node discovery and status check (#374)

[bytom/vapor.git] / test / mock / chain.go

diff --git a/test/mock/chain.go b/test/mock/chain.go
index 69ef7c4..82fc7dc 100644 (file)
--- a/test/mock/chain.go
+++ b/test/mock/chain.go
@@ -28,10 +28,11 @@ type Chain struct {
 
 func NewChain(mempool *Mempool) *Chain {
        return &Chain{
-               heightMap:   map[uint64]*types.Block{},
-               blockMap:    map[bc.Hash]*types.Block{},
-               prevOrphans: make(map[bc.Hash]*types.Block),
-               mempool:     mempool,
+               bestBlockHeader: &types.BlockHeader{},
+               heightMap:       map[uint64]*types.Block{},
+               blockMap:        map[bc.Hash]*types.Block{},
+               prevOrphans:     make(map[bc.Hash]*types.Block),
+               mempool:         mempool,
        }
 }
 
@@ -147,6 +148,11 @@ func (c *Chain) ProcessBlock(block *types.Block) (bool, error) {
        return false, nil
 }
 
+// TODO:
+func (c *Chain) ProcessBlockSignature(signature, pubkey []byte, blockHash *bc.Hash) error {
+       return nil
+}
+
 func (c *Chain) SetBestBlockHeader(header *types.BlockHeader) {
        c.bestBlockHeader = header
 }