From ac7f1f6719621e9390f1f97bc64a6aaa43cf97d0 Mon Sep 17 00:00:00 2001
From: Robert Habermeier <rphmeier@gmail.com>
Date: Thu, 23 Mar 2017 14:15:13 +0100
Subject: [PATCH] fix header chain tests

---
 ethcore/light/src/client/header_chain.rs | 21 ++++++++++++++-------
 1 file changed, 14 insertions(+), 7 deletions(-)

diff --git a/ethcore/light/src/client/header_chain.rs b/ethcore/light/src/client/header_chain.rs
index 8256595c3..1c218204b 100644
--- a/ethcore/light/src/client/header_chain.rs
+++ b/ethcore/light/src/client/header_chain.rs
@@ -517,8 +517,9 @@ mod tests {
 			parent_hash = header.hash();
 
 			let mut tx = db.transaction();
-			chain.insert(&mut tx, header).unwrap();
+			let pending = chain.insert(&mut tx, header).unwrap();
 			db.write(tx).unwrap();
+			chain.apply_pending(pending);
 
 			rolling_timestamp += 10;
 		}
@@ -548,8 +549,9 @@ mod tests {
 			parent_hash = header.hash();
 
 			let mut tx = db.transaction();
-			chain.insert(&mut tx, header).unwrap();
+			let pending = chain.insert(&mut tx, header).unwrap();
 			db.write(tx).unwrap();
+			chain.apply_pending(pending);
 
 			rolling_timestamp += 10;
 		}
@@ -566,8 +568,9 @@ mod tests {
 				parent_hash = header.hash();
 
 				let mut tx = db.transaction();
-				chain.insert(&mut tx, header).unwrap();
+				let pending = chain.insert(&mut tx, header).unwrap();
 				db.write(tx).unwrap();
+				chain.apply_pending(pending);
 
 				rolling_timestamp += 10;
 			}
@@ -589,8 +592,9 @@ mod tests {
 				parent_hash = header.hash();
 
 				let mut tx = db.transaction();
-				chain.insert(&mut tx, header).unwrap();
+				let pending = chain.insert(&mut tx, header).unwrap();
 				db.write(tx).unwrap();
+				chain.apply_pending(pending);
 
 				rolling_timestamp += 11;
 			}
@@ -640,8 +644,9 @@ mod tests {
 				parent_hash = header.hash();
 
 				let mut tx = db.transaction();
-				chain.insert(&mut tx, header).unwrap();
+				let pending = chain.insert(&mut tx, header).unwrap();
 				db.write(tx).unwrap();
+				chain.apply_pending(pending);
 
 				rolling_timestamp += 10;
 			}
@@ -676,8 +681,9 @@ mod tests {
 				parent_hash = header.hash();
 
 				let mut tx = db.transaction();
-				chain.insert(&mut tx, header).unwrap();
+				let pending = chain.insert(&mut tx, header).unwrap();
 				db.write(tx).unwrap();
+				chain.apply_pending(pending);
 
 				rolling_timestamp += 10;
 			}
@@ -692,8 +698,9 @@ mod tests {
 				parent_hash = header.hash();
 
 				let mut tx = db.transaction();
-				chain.insert(&mut tx, header).unwrap();
+				let pending = chain.insert(&mut tx, header).unwrap();
 				db.write(tx).unwrap();
+				chain.apply_pending(pending);
 
 				rolling_timestamp += 10;
 			}