mirror of
https://github.com/rust-lang/rust.git
synced 2024-11-23 07:14:28 +00:00
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.
38 lines
877 B
Rust
38 lines
877 B
Rust
#![feature(coverage_attribute)]
|
|
#![feature(custom_inner_attributes)] // for #![rustfmt::skip]
|
|
#![feature(noop_waker)]
|
|
#![rustfmt::skip]
|
|
//@ edition: 2021
|
|
|
|
#[coverage(off)]
|
|
async fn ready() -> u8 { 1 }
|
|
|
|
async fn await_ready() -> u8 {
|
|
// await should be covered even if the function never yields
|
|
ready()
|
|
.await
|
|
}
|
|
|
|
fn main() {
|
|
let mut future = Box::pin(await_ready());
|
|
executor::block_on(future.as_mut());
|
|
}
|
|
|
|
mod executor {
|
|
use core::future::Future;
|
|
use core::pin::pin;
|
|
use core::task::{Context, Poll, Waker};
|
|
|
|
#[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;
|
|
}
|
|
}
|
|
}
|
|
}
|