From ee86901acbc98cef9672008bb30aa71aaab62747 Mon Sep 17 00:00:00 2001
From: Jim Posen <jimpo@coinbase.com>
Date: Mon, 25 Sep 2017 13:21:53 -0700
Subject: [PATCH] shachain: Add two missing test cases from the BOLT spec.

---
 shachain/element_test.go | 16 ++++++++++++++++
 shachain/store_test.go   |  2 ++
 2 files changed, 18 insertions(+)

diff --git a/shachain/element_test.go b/shachain/element_test.go
index c10e49fc..392f676f 100644
--- a/shachain/element_test.go
+++ b/shachain/element_test.go
@@ -187,6 +187,8 @@ func TestDeriveIndex(t *testing.T) {
 	}
 }
 
+// deriveElementTests encodes the test vectors specified in BOLT-03,
+// Appendix D, Generation Tests.
 var deriveElementTests = []struct {
 	name       string
 	index      index
@@ -194,6 +196,20 @@ var deriveElementTests = []struct {
 	seed       string
 	shouldFail bool
 }{
+	{
+		name:       "generate_from_seed 0 final node",
+		seed:       "0000000000000000000000000000000000000000000000000000000000000000",
+		index:      0xffffffffffff,
+		output:     "02a40c85b6f28da08dfdbe0926c53fab2de6d28c10301f8f7c4073d5e42e3148",
+		shouldFail: false,
+	},
+	{
+		name:       "generate_from_seed FF final node",
+		seed:       "FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF",
+		index:      0xffffffffffff,
+		output:     "7cc854b54e3e0dcdb010d7a3fee464a9687be6e8db3be6854c475621e007a5dc",
+		shouldFail: false,
+	},
 	{
 		name:       "generate_from_seed FF alternate bits 1",
 		index:      0xaaaaaaaaaaa,
diff --git a/shachain/store_test.go b/shachain/store_test.go
index 71ae48d9..5e2e319a 100644
--- a/shachain/store_test.go
+++ b/shachain/store_test.go
@@ -13,6 +13,8 @@ type testInsert struct {
 	successful bool
 }
 
+// tests encodes the test vectors specified in BOLT-03, Appendix D,
+// Storage Tests.
 var tests = []struct {
 	name    string
 	inserts []testInsert