1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
|
#![feature(test)]
#![allow(deprecated)]
use futures_lite::future::block_on;
use futures_micro::{poll_fn, prelude::*};
use std::task::Poll;
fn ready<T>(x: T) -> impl Future<Output = T> {
let mut x = Some(x);
poll_fn(move |_| Poll::Ready(x.take().unwrap()))
}
fn pending<T>() -> impl Future<Output = T> {
poll_fn(|_| Poll::Pending)
}
#[test]
fn waker_sleep_test() {
assert!(block_on(async {
let waker = waker().await;
waker.wake();
sleep().await;
true
}));
}
#[test]
fn yield_once_test() {
assert_eq!(
true,
block_on(async {
yield_once().await;
true
})
);
assert_eq!(
false,
block_on(or!(
async {
yield_once().await;
true
},
ready(false)
))
);
}
#[test]
fn or_test() {
assert_eq!(false, block_on(or(pending::<bool>(), ready(false))));
assert_eq!(1, block_on(or!(ready(1), ready(2), ready(3))));
assert_eq!(2, block_on(or!(pending(), ready(2), ready(3))));
assert_eq!(3, block_on(or!(pending(), pending(), ready(3))));
}
#[test]
fn zip_test() {
assert_eq!((true, false), block_on(zip(ready(true), ready(false))));
assert_eq!((1, 2, 3), block_on(zip!(ready(1), ready(2), ready(3))));
assert_eq!(
(1, false, 3),
block_on(zip!(ready(1), ready(false), ready(3)))
);
}
|