about summary refs log tree commit diff
path: root/tests/coverage/auxiliary/executor.rs
diff options
context:
space:
mode:
authorZalathar <Zalathar@users.noreply.github.com>2024-09-06 15:41:10 +1000
committerZalathar <Zalathar@users.noreply.github.com>2024-09-10 16:08:36 +1000
commite96b4e479a3835d43d3c0bfd1028b5ac9d99cfec (patch)
treef995ef9632a946a93bf06b395cbdc69e614d2162 /tests/coverage/auxiliary/executor.rs
parent304b7f801bab31233680879ca4fb6eb294706a59 (diff)
downloadrust-e96b4e479a3835d43d3c0bfd1028b5ac9d99cfec.tar.gz
rust-e96b4e479a3835d43d3c0bfd1028b5ac9d99cfec.zip
coverage: Extract `executor::block_on` from several async coverage tests
By moving `block_on` to an auxiliary crate, we avoid having to keep a separate
copy of it in every async test.

(This also incorporates some small tweaks to the headers in `await_ready.rs`.)
Diffstat (limited to 'tests/coverage/auxiliary/executor.rs')
-rw-r--r--tests/coverage/auxiliary/executor.rs19
1 files changed, 19 insertions, 0 deletions
diff --git a/tests/coverage/auxiliary/executor.rs b/tests/coverage/auxiliary/executor.rs
new file mode 100644
index 00000000000..fb07c8ce304
--- /dev/null
+++ b/tests/coverage/auxiliary/executor.rs
@@ -0,0 +1,19 @@
+#![feature(coverage_attribute, noop_waker)]
+//@ edition: 2021
+
+use core::future::Future;
+use core::pin::pin;
+use core::task::{Context, Poll, Waker};
+
+/// Dummy "executor" that just repeatedly polls a future until it's ready.
+#[coverage(off)]
+pub fn block_on<F: Future>(mut future: F) -> F::Output {
+    let mut future = pin!(future);
+    let mut context = Context::from_waker(Waker::noop());
+
+    loop {
+        if let Poll::Ready(val) = future.as_mut().poll(&mut context) {
+            break val;
+        }
+    }
+}