runtime: add `thread_park_ok` test (#7402)

diff --git a/tokio/tests/rt_handle_block_on.rs b/tokio/tests/rt_handle_block_on.rs
index d62815df9cfeb02a1b6034860b7b89af35779244..b6375ebf0b6973b2925976d10f987596a7b0d654 100644 (file)
--- a/tokio/tests/rt_handle_block_on.rs
+++ b/tokio/tests/rt_handle_block_on.rs
@@ -220,6 +220,24 @@ rt_test! {
         assert_eq!(answer, 42);
     }
 
+    #[test]
+    fn thread_park_ok() {
+        use core::task::Poll;
+        let rt = rt();
+        let mut exit = false;
+        rt.handle().block_on(core::future::poll_fn(move |cx| {
+            if exit {
+                return Poll::Ready(());
+            }
+            cx.waker().wake_by_ref();
+            // Verify that consuming the park token does not result in a hang.
+            std::thread::current().unpark();
+            std::thread::park();
+            exit = true;
+            Poll::Pending
+        }));
+    }
+
     // ==== net ======
 
     #[test]