Fix shuttle tests

akoshelev · Oct 31, 2023 · f2a0dda · f2a0dda
1 parent 0b05f39
commit f2a0dda
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/helpers/gateway/mod.rs b/src/helpers/gateway/mod.rs
@@ -12,6 +12,8 @@ use std::num::NonZeroUsize;
 
 pub(super) use receive::ReceivingEnd;
 pub(super) use send::SendingEnd;
+#[cfg(all(test, feature = "shuttle"))]
+use shuttle::future as tokio;
 #[cfg(feature = "stall-detection")]
 pub(super) use {observable::Observed, stall_detection::InstrumentedGateway};