From 39d08a8ae65a37adfb774fdbf6e0f6247e94bda5 Mon Sep 17 00:00:00 2001
From: Wei-Ning Huang <w@dexon.org>
Date: Tue, 27 Nov 2018 13:21:54 +0800
Subject: tests: fix tests

---
 core/evm.go | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'core')

diff --git a/core/evm.go b/core/evm.go
index 80b0eaa7a..04bc173f5 100644
--- a/core/evm.go
+++ b/core/evm.go
@@ -60,7 +60,7 @@ func NewEVMContext(msg Message, header *types.Header, chain ChainContext, author
 		Transfer:       Transfer,
 		GetHash:        GetHashFn(header, chain),
 		StateAtNumber:  StateAtNumberFn(chain),
-		GetRoundHeight: chain.GetRoundHeight,
+		GetRoundHeight: GetRoundHeightFn(chain),
 		Origin:         msg.From(),
 		Coinbase:       beneficiary,
 		BlockNumber:    new(big.Int).Set(header.Number),
@@ -81,6 +81,15 @@ func StateAtNumberFn(chain ChainContext) func(n uint64) (*state.StateDB, error)
 	}
 }
 
+func GetRoundHeightFn(chain ChainContext) func(uint64) (uint64, bool) {
+	if chain != nil {
+		return chain.GetRoundHeight
+	}
+	return func(uint64) (uint64, bool) {
+		return 0, false
+	}
+}
+
 // GetHashFn returns a GetHashFunc which retrieves header hashes by number
 func GetHashFn(ref *types.Header, chain ChainContext) func(n uint64) common.Hash {
 	var cache map[uint64]common.Hash
-- 
cgit v1.2.3