From 5ae7700f0b1d36a821489526ee174813aea3cb53 Mon Sep 17 00:00:00 2001
From: Pierre Krieger <pierre.krieger1708@gmail.com>
Date: Fri, 24 Apr 2020 11:45:00 +0200
Subject: [PATCH] Companion to #5763 - network polling

---
 network/test/src/lib.rs | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/network/test/src/lib.rs b/network/test/src/lib.rs
index 36bdda0ba4d6..e565d75f30db 100644
--- a/network/test/src/lib.rs
+++ b/network/test/src/lib.rs
@@ -752,8 +752,12 @@ pub trait TestNetFactory: Sized {
 		self.mut_peers(|peers| {
 			for peer in peers {
 				trace!(target: "sync", "-- Polling {}", peer.id());
-				if let Poll::Ready(res) = Pin::new(&mut peer.network).poll(cx) {
-					res.unwrap();
+				loop {
+					let net_poll_future = peer.network.next_action();
+					futures::pin_mut!(net_poll_future);
+					if let Poll::Pending = net_poll_future.poll(cx) {
+						break;
+					}
 				}
 				trace!(target: "sync", "-- Polling complete {}", peer.id());