about summary refs log tree commit diff
path: root/tests/coverage/async_block.coverage
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/async_block.coverage
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/async_block.coverage')
-rw-r--r--tests/coverage/async_block.coverage21
1 files changed, 3 insertions, 18 deletions
diff --git a/tests/coverage/async_block.coverage b/tests/coverage/async_block.coverage
index 7fd17bc51c3..d9be8480d80 100644
--- a/tests/coverage/async_block.coverage
+++ b/tests/coverage/async_block.coverage
@@ -2,6 +2,9 @@
    LL|       |#![feature(noop_waker)]
    LL|       |//@ edition: 2021
    LL|       |
+   LL|       |//@ aux-build: executor.rs
+   LL|       |extern crate executor;
+   LL|       |
    LL|      1|fn main() {
    LL|     17|    for i in 0..16 {
                       ^16
@@ -15,22 +18,4 @@
    LL|     16|        executor::block_on(future);
    LL|     16|    }
    LL|      1|}
-   LL|       |
-   LL|       |mod executor {
-   LL|       |    use core::future::Future;
-   LL|       |    use core::pin::pin;
-   LL|       |    use core::task::{Context, Poll, Waker};
-   LL|       |
-   LL|       |    #[coverage(off)]
-   LL|       |    pub fn block_on<F: Future>(mut future: F) -> F::Output {
-   LL|       |        let mut future = pin!(future);
-   LL|       |        let mut context = Context::from_waker(Waker::noop());
-   LL|       |
-   LL|       |        loop {
-   LL|       |            if let Poll::Ready(val) = future.as_mut().poll(&mut context) {
-   LL|       |                break val;
-   LL|       |            }
-   LL|       |        }
-   LL|       |    }
-   LL|       |}