diff options
Diffstat (limited to 'tokio/tests/io_driver_drop.rs')
-rw-r--r-- | tokio/tests/io_driver_drop.rs | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/tokio/tests/io_driver_drop.rs b/tokio/tests/io_driver_drop.rs index c75631b7..6b923dd3 100644 --- a/tokio/tests/io_driver_drop.rs +++ b/tokio/tests/io_driver_drop.rs @@ -6,7 +6,7 @@ use tokio_test::{assert_err, assert_pending, assert_ready, task}; #[test] fn tcp_doesnt_block() { - let rt = runtime::Builder::new().basic_scheduler().build().unwrap(); + let rt = rt(); let mut listener = rt.enter(|| { let listener = std::net::TcpListener::bind("127.0.0.1:0").unwrap(); @@ -24,7 +24,7 @@ fn tcp_doesnt_block() { #[test] fn drop_wakes() { - let rt = runtime::Builder::new().basic_scheduler().build().unwrap(); + let rt = rt(); let mut listener = rt.enter(|| { let listener = std::net::TcpListener::bind("127.0.0.1:0").unwrap(); @@ -42,3 +42,11 @@ fn drop_wakes() { assert!(task.is_woken()); assert_ready!(task.poll()); } + +fn rt() -> runtime::Runtime { + runtime::Builder::new() + .basic_scheduler() + .enable_all() + .build() + .unwrap() +} |