25d183057e
Currently `await` is only counted towards coverage if the containing function is suspended and resumed at least once. This is because it expands to code which contains a branch on the discriminant of `Poll`. By treating it like a branching macro (e.g. `assert!`), these implementation details will be hidden from the coverage results.
39 lines
1.4 KiB
Plaintext
39 lines
1.4 KiB
Plaintext
LL| |#![feature(coverage_attribute)]
|
|
LL| |#![feature(custom_inner_attributes)] // for #![rustfmt::skip]
|
|
LL| |#![feature(noop_waker)]
|
|
LL| |#![rustfmt::skip]
|
|
LL| |//@ edition: 2021
|
|
LL| |
|
|
LL| |#[coverage(off)]
|
|
LL| |async fn ready() -> u8 { 1 }
|
|
LL| |
|
|
LL| 1|async fn await_ready() -> u8 {
|
|
LL| 1| // await should be covered even if the function never yields
|
|
LL| 1| ready()
|
|
LL| 1| .await
|
|
LL| 1|}
|
|
LL| |
|
|
LL| 1|fn main() {
|
|
LL| 1| let mut future = Box::pin(await_ready());
|
|
LL| 1| executor::block_on(future.as_mut());
|
|
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| |}
|
|
|