extern crate futures;
mod support;
use futures::stream;
use futures::future::{ok, Either};
use support::*;
#[test]
fn unfold1() {
let mut stream = stream::unfold(0, |state| {
if state <= 2 {
let res = ok::<_, ()>(Some((state * 2, state + 1)));
Either::A(delay_future(res))
} else {
Either::B(ok(None))
}
});
sassert_empty(&mut stream);
sassert_next(&mut stream, 0);
sassert_empty(&mut stream);
sassert_next(&mut stream, 2);
sassert_empty(&mut stream);
sassert_next(&mut stream, 4);
sassert_done(&mut stream);
}
#[test]
fn unfold_err1() {
let mut stream = stream::unfold(0, |state| {
if state <= 2 {
Ok(Some((state * 2, state + 1)))
} else {
Err(-1)
}
});
sassert_next(&mut stream, 0);
sassert_next(&mut stream, 2);
sassert_next(&mut stream, 4);
sassert_err(&mut stream, -1);
sassert_done(&mut stream);
}