core: fixed head write on block insertion

Due to a rebase this probably got overlooked / ignored. This fixes the
issue of a block insertion never writing the last block.

1 files changed, 13 insertions, 0 deletions

diff --git a/core/blockchain_test.go b/core/blockchain_test.go
index e034417ce..13971ccba 100644
--- a/core/blockchain_test.go
+++ b/core/blockchain_test.go
@@ -153,6 +153,19 @@ func insertChain(done chan bool, blockchain *BlockChain, chain types.Blocks, t *
 	done <- true
 }
 
+func TestLastBlock(t *testing.T) {
+	db, err := ethdb.NewMemDatabase()
+	if err != nil {
+		t.Fatal("Failed to create db:", err)
+	}
+	bchain := theBlockChain(db, t)
+	block := makeChain(bchain.CurrentBlock(), 1, db, 0)[0]
+	bchain.insert(block)
+	if block.Hash() != GetHeadBlockHash(db) {
+		t.Errorf("Write/Get HeadBlockHash failed")
+	}
+}
+
 func TestExtendCanonical(t *testing.T) {
 	CanonicalLength := 5
 	db, err := ethdb.NewMemDatabase()