Added block notification after import

paritytech · Oct 19, 2018 · 43943d7 · 43943d7
1 parent df72212
commit 43943d7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/core/network/src/test/mod.rs b/core/network/src/test/mod.rs
@@ -228,8 +228,11 @@ impl<V: 'static + Verifier<Block>> Peer<V> {
 			let mut builder = self.client.new_block().unwrap();
 			edit_block(&mut builder);
 			let block = builder.bake().unwrap();
-			trace!("Generating {}, (#{}, parent={})", block.header.hash(), block.header.number, block.header.parent_hash);
+			let hash = block.header.hash();
+			trace!("Generating {}, (#{}, parent={})", hash, block.header.number, block.header.parent_hash);
+			let header = block.header.clone();
 			self.client.justify_and_import(origin, block).unwrap();
+			self.sync.on_block_imported(&mut TestIo::new(&self.queue, None), hash, &header);
 		}
 	}