diff --git a/transports/quic/tests/stream_compliance.rs b/transports/quic/tests/stream_compliance.rs index 13c29f2caa0..47326aa7ebb 100644 --- a/transports/quic/tests/stream_compliance.rs +++ b/transports/quic/tests/stream_compliance.rs @@ -7,14 +7,16 @@ use libp2p_core::{ }; use libp2p_quic as quic; -#[async_std::test] +#[cfg(feature = "tokio")] +#[tokio::test] async fn close_implies_flush() { let (alice, bob) = connected_peers().await; libp2p_muxer_test_harness::close_implies_flush(alice, bob).await; } -#[async_std::test] +#[cfg(feature = "tokio")] +#[tokio::test] async fn read_after_close() { let (alice, bob) = connected_peers().await; @@ -36,10 +38,10 @@ async fn connected_peers() -> (quic::Connection, quic::Connection) { let (dialer_conn_sender, dialer_conn_receiver) = oneshot::channel(); let (listener_conn_sender, listener_conn_receiver) = oneshot::channel(); - async_std::task::spawn(async move { + tokio::spawn(async move { let (upgrade, _) = listener.next().await.unwrap().into_incoming().unwrap(); - async_std::task::spawn(async move { + tokio::spawn(async move { let (_, connection) = upgrade.await.unwrap(); let _ = listener_conn_sender.send(connection); @@ -58,13 +60,13 @@ async fn connected_peers() -> (quic::Connection, quic::Connection) { }, ) .unwrap(); - async_std::task::spawn(async move { + tokio::spawn(async move { let connection = dial_fut.await.unwrap().1; let _ = dialer_conn_sender.send(connection); }); - async_std::task::spawn(async move { + tokio::spawn(async move { loop { dialer.next().await; } @@ -75,10 +77,10 @@ async fn connected_peers() -> (quic::Connection, quic::Connection) { .unwrap() } -fn new_transport() -> quic::async_std::Transport { +fn new_transport() -> quic::tokio::Transport { let keypair = libp2p_identity::Keypair::generate_ed25519(); let mut config = quic::Config::new(&keypair); config.handshake_timeout = Duration::from_secs(1); - quic::async_std::Transport::new(config) + quic::tokio::Transport::new(config) }