33 lines
826 B
Rust
33 lines
826 B
Rust
|
// run-pass
|
||
|
|
// edition: 2024
|
||
|
|
// compile-flags: -Zunstable-options
|
||
|
|
#![feature(async_iterator, async_iter_from_iter, const_waker, async_for_loop, noop_waker,
|
||
|
|
gen_blocks)]
|
||
|
|
|
||
|
|
use std::future::Future;
|
||
|
|
|
||
|
|
async gen fn async_iter() -> i32 {
|
||
|
|
let iter = core::async_iter::from_iter(0..3);
|
||
|
|
for await i in iter {
|
||
|
|
yield i + 1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// make sure a simple for await loop works
|
||
|
|
async fn real_main() {
|
||
|
|
let mut count = 1;
|
||
|
|
for await i in async_iter() {
|
||
|
|
assert_eq!(i, count);
|
||
|
|
count += 1;
|
||
|
|
}
|
||
|
|
assert_eq!(count, 4);
|
||
|
|
}
|
||
|
|
|
||
|
|
fn main() {
|
||
|
|
let future = real_main();
|
||
|
|
let waker = std::task::Waker::noop();
|
||
|
|
let mut cx = &mut core::task::Context::from_waker(&waker);
|
||
|
|
let mut future = core::pin::pin!(future);
|
||
|
|
while let core::task::Poll::Pending = future.as_mut().poll(&mut cx) {}
|
||
|
|
}
|