2023-12-14 23:38:27 -06:00
|
|
|
LL| |#![feature(coverage_attribute)]
|
|
|
|
LL| |#![feature(noop_waker)]
|
|
|
|
LL| |// edition: 2018
|
2023-08-15 21:42:33 -05:00
|
|
|
LL| |
|
|
|
|
LL| 1|fn non_async_func() {
|
|
|
|
LL| 1| println!("non_async_func was covered");
|
|
|
|
LL| 1| let b = true;
|
|
|
|
LL| 1| if b {
|
|
|
|
LL| 1| println!("non_async_func println in block");
|
|
|
|
LL| 1| }
|
2021-05-01 16:56:48 -05:00
|
|
|
^0
|
2023-08-15 21:42:33 -05:00
|
|
|
LL| 1|}
|
|
|
|
LL| |
|
|
|
|
LL| 1|async fn async_func() {
|
|
|
|
LL| 1| println!("async_func was covered");
|
|
|
|
LL| 1| let b = true;
|
|
|
|
LL| 1| if b {
|
|
|
|
LL| 1| println!("async_func println in block");
|
|
|
|
LL| 1| }
|
2021-04-18 17:04:59 -05:00
|
|
|
^0
|
2023-08-15 21:42:33 -05:00
|
|
|
LL| 1|}
|
|
|
|
LL| |
|
|
|
|
LL| 1|async fn async_func_just_println() {
|
|
|
|
LL| 1| println!("async_func_just_println was covered");
|
|
|
|
LL| 1|}
|
|
|
|
LL| |
|
|
|
|
LL| 1|fn main() {
|
|
|
|
LL| 1| println!("codecovsample::main");
|
|
|
|
LL| 1|
|
|
|
|
LL| 1| non_async_func();
|
|
|
|
LL| 1|
|
|
|
|
LL| 1| executor::block_on(async_func());
|
|
|
|
LL| 1| executor::block_on(async_func_just_println());
|
|
|
|
LL| 1|}
|
|
|
|
LL| |
|
|
|
|
LL| |mod executor {
|
2023-12-14 23:38:27 -06:00
|
|
|
LL| | use core::future::Future;
|
|
|
|
LL| | use core::pin::pin;
|
|
|
|
LL| | use core::task::{Context, Poll, Waker};
|
2023-08-15 21:42:33 -05:00
|
|
|
LL| |
|
2023-12-14 23:38:27 -06:00
|
|
|
LL| | #[coverage(off)]
|
|
|
|
LL| | pub fn block_on<F: Future>(mut future: F) -> F::Output {
|
|
|
|
LL| | let mut future = pin!(future);
|
|
|
|
LL| | let waker = Waker::noop();
|
|
|
|
LL| | let mut context = Context::from_waker(&waker);
|
2023-08-15 21:42:33 -05:00
|
|
|
LL| |
|
|
|
|
LL| | loop {
|
2023-12-14 23:38:27 -06:00
|
|
|
LL| | if let Poll::Ready(val) = future.as_mut().poll(&mut context) {
|
|
|
|
LL| | break val;
|
|
|
|
LL| | }
|
2023-08-15 21:42:33 -05:00
|
|
|
LL| | }
|
2023-12-14 23:38:27 -06:00
|
|
|
LL| | }
|
2023-08-15 21:42:33 -05:00
|
|
|
LL| |}
|
2021-04-18 17:04:59 -05:00
|
|
|
|