summaryrefslogtreecommitdiffstats
path: root/tokio/tests/io_driver_drop.rs
diff options
context:
space:
mode:
Diffstat (limited to 'tokio/tests/io_driver_drop.rs')
-rw-r--r--tokio/tests/io_driver_drop.rs12
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()
+}