Apply suggestions from code review

Co-authored-by: Bastian Köcher <[email protected]>
ltfschoen · Feb 13, 2023 · 9fa646d · 9fa646d
1 parent faec2e9
commit 9fa646d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git 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
@@ -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());