From 9fa646d0ed613e5f8623d3d37d1d59ec0a535850 Mon Sep 17 00:00:00 2001 From: Dmitry Markin Date: Mon, 13 Feb 2023 17:53:32 +0300 Subject: [PATCH] Apply suggestions from code review MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Bastian Köcher --- client/network/src/service/tests/mod.rs | 6 +++--- client/network/test/src/lib.rs | 5 ++--- 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/client/network/src/service/tests/mod.rs b/client/network/src/service/tests/mod.rs index 94fbf3208fe96..6f35a19aeb9d3 100644 --- a/client/network/src/service/tests/mod.rs +++ b/client/network/src/service/tests/mod.rs @@ -80,9 +80,9 @@ impl TestNetwork { let service = worker.service().clone(); let event_stream = service.event_stream("test"); - tokio::spawn(async move { - worker.run().await; - }); + tokio::spawn( + worker.run() + ); (service, event_stream) } diff --git a/client/network/test/src/lib.rs b/client/network/test/src/lib.rs index d680b03b3a4cd..77a83d10aeee8 100644 --- a/client/network/test/src/lib.rs +++ b/client/network/test/src/lib.rs @@ -1065,9 +1065,8 @@ where for (i, peer) in peers.iter_mut().enumerate() { trace!(target: "sync", "-- Polling {}: {}", i, peer.id()); { - let net_poll_future = peer.network.run(); - pin_mut!(net_poll_future); - let _ = net_poll_future.poll(cx); + peer.network.next_action().poll(cx); + } trace!(target: "sync", "-- Polling complete {}: {}", i, peer.id());