Fix get consensus edge case (#147)

[bytom/vapor.git] / protocol / state / blockindex_test.go

diff --git a/protocol/state/blockindex_test.go b/protocol/state/blockindex_test.go
index 28e2281..360c049 100644 (file)
--- a/protocol/state/blockindex_test.go
+++ b/protocol/state/blockindex_test.go
@@ -121,8 +121,9 @@ func TestSetMainChain(t *testing.T) {
 // MockBlockIndex will mock a empty BlockIndex
 func MockBlockIndex() *BlockIndex {
        return &BlockIndex{
-               index:     make(map[bc.Hash]*BlockNode),
-               mainChain: make([]*BlockNode, 0, 2),
+               index:       make(map[bc.Hash]*BlockNode),
+               heightIndex: make(map[uint64][]*BlockNode),
+               mainChain:   make([]*BlockNode, 0, 2),
        }
 }