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.
59 lines
1.7 KiB
59 lines
1.7 KiB
4 months ago
|
mod util {
|
||
|
|
use std::future::Future;
|
||
|
|
use futures::executor::block_on;
|
||
|
|
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>,
|
||
|
|
{
|
||
|
|
let output = block_on(actual_fut());
|
||
|
|
assert_eq!(output, expected);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn collect_collects() {
|
||
|
|
use futures_util::future::{err, ok, try_join_all};
|
||
|
|
|
||
|
|
use util::assert_done;
|
||
|
|
|
||
|
|
assert_done(|| Box::new(try_join_all(vec![ok(1), ok(2)])), Ok::<_, usize>(vec![1, 2]));
|
||
|
|
assert_done(|| Box::new(try_join_all(vec![ok(1), err(2)])), Err(2));
|
||
|
|
assert_done(|| Box::new(try_join_all(vec![ok(1)])), Ok::<_, usize>(vec![1]));
|
||
|
|
// REVIEW: should this be implemented?
|
||
|
|
// assert_done(|| Box::new(try_join_all(Vec::<i32>::new())), Ok(vec![]));
|
||
|
|
|
||
|
|
// TODO: needs more tests
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn try_join_all_iter_lifetime() {
|
||
|
|
use futures_util::future::{ok, try_join_all};
|
||
|
|
use std::future::Future;
|
||
|
|
|
||
|
|
use util::assert_done;
|
||
|
|
|
||
|
|
// In futures-rs version 0.1, this function would fail to typecheck due to an overly
|
||
|
|
// conservative type parameterization of `TryJoinAll`.
|
||
|
|
fn sizes(bufs: Vec<&[u8]>) -> Box<dyn Future<Output = Result<Vec<usize>, ()>> + Unpin> {
|
||
|
|
let iter = bufs.into_iter().map(|b| ok::<usize, ()>(b.len()));
|
||
|
|
Box::new(try_join_all(iter))
|
||
|
|
}
|
||
|
|
|
||
|
|
assert_done(|| sizes(vec![&[1,2,3], &[], &[0]]), Ok(vec![3_usize, 0, 1]));
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn try_join_all_from_iter() {
|
||
|
|
use futures_util::future::{ok, TryJoinAll};
|
||
|
|
|
||
|
|
use util::assert_done;
|
||
|
|
|
||
|
|
assert_done(
|
||
|
|
|| Box::new(vec![ok(1), ok(2)].into_iter().collect::<TryJoinAll<_>>()),
|
||
|
|
Ok::<_, usize>(vec![1, 2]),
|
||
|
|
)
|
||
|
|
}
|