18 lines
668 B
Rust
18 lines
668 B
Rust
|
use core::async_iter::{self, AsyncIterator, IntoAsyncIterator};
|
||
|
|
use core::pin::pin;
|
||
|
|
use core::task::Poll;
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn into_async_iter() {
|
||
|
|
let async_iter = async_iter::from_iter(0..3);
|
||
|
|
let mut async_iter = pin!(async_iter.into_async_iter());
|
||
|
|
|
||
|
|
let waker = core::task::Waker::noop();
|
||
|
|
let mut cx = &mut core::task::Context::from_waker(&waker);
|
||
|
|
|
||
|
|
assert_eq!(async_iter.as_mut().poll_next(&mut cx), Poll::Ready(Some(0)));
|
||
|
|
assert_eq!(async_iter.as_mut().poll_next(&mut cx), Poll::Ready(Some(1)));
|
||
|
|
assert_eq!(async_iter.as_mut().poll_next(&mut cx), Poll::Ready(Some(2)));
|
||
|
|
assert_eq!(async_iter.as_mut().poll_next(&mut cx), Poll::Ready(None));
|
||
|
|
}
|