fixup! spawn_mandatory_blocking: fix race with shutdown

tokio-rs · Jan 24, 2022 · f96fa96 · f96fa96
1 parent 75f487f
commit f96fa96
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tokio/src/runtime/tests/loom_blocking.rs b/tokio/src/runtime/tests/loom_blocking.rs
@@ -49,7 +49,7 @@ fn spawn_mandatory_blocking_should_run_even_when_shutting_down_from_other_thread
     use crate::runtime::tests::loom_oneshot;
     loom::model(|| {
         let rt = runtime::Builder::new_current_thread().build().unwrap();
-        let _enter = rt.enter();
+        let handle = rt.handle().clone();
 
         // Drop the runtime in a different thread
         {
@@ -58,6 +58,7 @@ fn spawn_mandatory_blocking_should_run_even_when_shutting_down_from_other_thread
             });
         }
 
+        let _enter = handle.enter();
         let (tx, rx) = loom_oneshot::channel();
         let handle = runtime::spawn_mandatory_blocking(move || {
             let _ = tx.send(());