From 6679c8b051a087bb6b668afd746332a7b08f1522 Mon Sep 17 00:00:00 2001 From: Weiliang Li Date: Tue, 17 Sep 2019 18:16:22 +0900 Subject: [PATCH] Update mod.rs (#3625) Add `import_notification_stream` for test client --- substrate/core/network/src/test/mod.rs | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/substrate/core/network/src/test/mod.rs b/substrate/core/network/src/test/mod.rs index f6c866d76d..b6a6103071 100644 --- a/substrate/core/network/src/test/mod.rs +++ b/substrate/core/network/src/test/mod.rs @@ -28,7 +28,7 @@ use crate::config::build_multiaddr; use log::trace; use crate::chain::FinalityProofProvider; use client::{ - self, ClientInfo, BlockchainEvents, BlockImportNotification, FinalityNotifications, + self, ClientInfo, BlockchainEvents, BlockImportNotification, FinalityNotifications, ImportNotifications, FinalityNotification, LongestChain }; use client::error::Result as ClientResult; @@ -189,6 +189,13 @@ impl PeersClient { } } + pub fn import_notification_stream(&self) -> ImportNotifications{ + match *self { + PeersClient::Full(ref client, ref _backend) => client.import_notification_stream(), + PeersClient::Light(ref client, ref _backend) => client.import_notification_stream(), + } + } + pub fn finalize_block( &self, id: BlockId,