core/tests: Remove unnecessary Arc (#2763)

libp2p · Jul 22, 2022 · 2e2c117 · 2e2c117
1 parent f15a3dc
commit 2e2c117
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 15 deletions.
diff --git a/muxers/mplex/tests/async_write.rs b/muxers/mplex/tests/async_write.rs
@@ -22,7 +22,6 @@ use futures::future::poll_fn;
 use futures::{channel::oneshot, prelude::*};
 use libp2p_core::{upgrade, StreamMuxer, Transport};
 use libp2p_tcp::TcpTransport;
-use std::sync::Arc;
 
 #[test]
 fn async_write() {
@@ -72,7 +71,7 @@ fn async_write() {
         let mut transport = TcpTransport::default()
             .and_then(move |c, e| upgrade::apply(c, mplex, e, upgrade::Version::V1));
 
-        let client = Arc::new(transport.dial(rx.await.unwrap()).unwrap().await.unwrap());
+        let client = transport.dial(rx.await.unwrap()).unwrap().await.unwrap();
         let mut inbound = poll_fn(|cx| client.poll_inbound(cx)).await.unwrap();
         inbound.write_all(b"hello world").await.unwrap();
 

diff --git a/muxers/mplex/tests/two_peers.rs b/muxers/mplex/tests/two_peers.rs
@@ -22,7 +22,6 @@ use futures::future::poll_fn;
 use futures::{channel::oneshot, prelude::*};
 use libp2p_core::{upgrade, StreamMuxer, Transport};
 use libp2p_tcp::TcpTransport;
-use std::sync::Arc;
 
 #[test]
 fn client_to_server_outbound() {
@@ -73,7 +72,7 @@ fn client_to_server_outbound() {
             .and_then(move |c, e| upgrade::apply(c, mplex, e, upgrade::Version::V1))
             .boxed();
 
-        let client = Arc::new(transport.dial(rx.await.unwrap()).unwrap().await.unwrap());
+        let client = transport.dial(rx.await.unwrap()).unwrap().await.unwrap();
         let mut inbound = poll_fn(|cx| client.poll_inbound(cx)).await.unwrap();
         inbound.write_all(b"hello world").await.unwrap();
         inbound.close().await.unwrap();
@@ -108,17 +107,15 @@ fn client_to_server_inbound() {
 
         tx.send(addr).unwrap();
 
-        let client = Arc::new(
-            transport
-                .next()
-                .await
-                .expect("some event")
-                .into_incoming()
-                .unwrap()
-                .0
-                .await
-                .unwrap(),
-        );
+        let client = transport
+            .next()
+            .await
+            .expect("some event")
+            .into_incoming()
+            .unwrap()
+            .0
+            .await
+            .unwrap();
 
         let mut inbound = poll_fn(|cx| client.poll_inbound(cx)).await.unwrap();