You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

52 lines
1.5 KiB

mod util {
use std::future::Future;
use std::fmt::Debug;
pub fn assert_done<T, F>(actual_fut: F, expected: T)
where
T: PartialEq + Debug,
F: FnOnce() -> Box<dyn Future<Output = T> + Unpin>,
{
use futures::executor::block_on;
let output = block_on(actual_fut());
assert_eq!(output, expected);
}
}
#[test]
fn collect_collects() {
use futures_util::future::{join_all,ready};
util::assert_done(|| Box::new(join_all(vec![ready(1), ready(2)])), vec![1, 2]);
util::assert_done(|| Box::new(join_all(vec![ready(1)])), vec![1]);
// REVIEW: should this be implemented?
// assert_done(|| Box::new(join_all(Vec::<i32>::new())), vec![]);
// TODO: needs more tests
}
#[test]
fn join_all_iter_lifetime() {
use futures_util::future::{join_all,ready};
use std::future::Future;
// In futures-rs version 0.1, this function would fail to typecheck due to an overly
// conservative type parameterization of `JoinAll`.
fn sizes(bufs: Vec<&[u8]>) -> Box<dyn Future<Output = Vec<usize>> + Unpin> {
let iter = bufs.into_iter().map(|b| ready::<usize>(b.len()));
Box::new(join_all(iter))
}
util::assert_done(|| sizes(vec![&[1,2,3], &[], &[0]]), vec![3_usize, 0, 1]);
}
#[test]
fn join_all_from_iter() {
use futures_util::future::{JoinAll,ready};
util::assert_done(
|| Box::new(vec![ready(1), ready(2)].into_iter().collect::<JoinAll<_>>()),
vec![1, 2],
)
}