Fix wait_for_blocks counting blocks that didn't go through consensus (

#6850) * Initial commit Forked at: 74804b5 Parent branch: origin/master * WIP Forked at: 74804b5 Parent branch: origin/master
paritytech · Aug 7, 2020 · 7b00235 · 7b00235
1 parent 74804b5
commit 7b00235
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/test-utils/client/src/lib.rs b/test-utils/client/src/lib.rs
@@ -384,9 +384,11 @@ where
 
 		Box::pin(async move {
 			while let Some(notification) = import_notification_stream.next().await {
-				blocks.insert(notification.hash);
-				if blocks.len() == count {
-					break;
+				if notification.is_new_best {
+					blocks.insert(notification.hash);
+					if blocks.len() == count {
+						break;
+					}
 				}
 			}
 		})