From 8c34084e24be362a296e3065c909d289953a7c21 Mon Sep 17 00:00:00 2001
From: Dave Collins <davec@conformal.com>
Date: Sun, 19 Jan 2014 01:37:25 -0600
Subject: [PATCH] Add interface tests for new Fetch functions.

---
 interface_test.go | 60 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)

diff --git a/interface_test.go b/interface_test.go
index a19e9978..19f56c79 100644
--- a/interface_test.go
+++ b/interface_test.go
@@ -135,6 +135,54 @@ func testFetchBlockBySha(tc *testContext) bool {
 	return true
 }
 
+// testFetchBlockHeightBySha ensures FetchBlockHeightBySha conforms to the
+// interface contract.
+func testFetchBlockHeightBySha(tc *testContext) bool {
+	// The block height must be fetchable by its hash without any errors.
+	blockHeight, err := tc.db.FetchBlockHeightBySha(tc.blockHash)
+	if err != nil {
+		tc.t.Errorf("FetchBlockHeightBySha (%s): block #%d (%s) err: %v",
+			tc.dbType, tc.blockHeight, tc.blockHash, err)
+		return false
+	}
+
+	// The block height fetched from the database must match the expected
+	// height.
+	if blockHeight != tc.blockHeight {
+		tc.t.Errorf("FetchBlockHeightBySha (%s): block #%d (%s) height "+
+			"does not match expected value - got: %v", tc.dbType,
+			tc.blockHeight, tc.blockHash, blockHeight)
+		return false
+	}
+
+	return true
+}
+
+// testFetchBlockHeaderBySha ensures FetchBlockHeaderBySha conforms to the
+// interface contract.
+func testFetchBlockHeaderBySha(tc *testContext) bool {
+	// The block header must be fetchable by its hash without any errors.
+	blockHeader, err := tc.db.FetchBlockHeaderBySha(tc.blockHash)
+	if err != nil {
+		tc.t.Errorf("FetchBlockHeaderBySha (%s): block #%d (%s) err: %v",
+			tc.dbType, tc.blockHeight, tc.blockHash, err)
+		return false
+	}
+
+	// The block header fetched from the database must give back the same
+	// BlockHeader that was stored.
+	if !reflect.DeepEqual(&tc.block.MsgBlock().Header, blockHeader) {
+		tc.t.Errorf("FetchBlockHeaderBySha (%s): block header #%d (%s) "+
+			" does not match stored block\ngot: %v\nwant: %v",
+			tc.dbType, tc.blockHeight, tc.blockHash,
+			spew.Sdump(blockHeader),
+			spew.Sdump(&tc.block.MsgBlock().Header))
+		return false
+	}
+
+	return true
+}
+
 // testFetchBlockShaByHeight ensures FetchBlockShaByHeight conforms to the
 // interface contract.
 func testFetchBlockShaByHeight(tc *testContext) bool {
@@ -414,6 +462,18 @@ func testIntegrity(tc *testContext) bool {
 		return false
 	}
 
+	// The height returned for the block given its hash must be the
+	// expected value
+	if !testFetchBlockHeightBySha(tc) {
+		return false
+	}
+
+	// Loading the header back from the database must give back
+	// the same BlockHeader that was stored.
+	if !testFetchBlockHeaderBySha(tc) {
+		return false
+	}
+
 	// The hash returned for the block by its height must be the
 	// expected value.
 	if !testFetchBlockShaByHeight(tc) {