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
|
extern crate core;
extern crate futures;
use core::marker;
use futures::prelude::*;
use futures::future::{ok, err};
use futures::stream;
#[test]
fn successful_future() {
let stream_items = vec![17, 19];
let future_of_a_stream = ok::<_, bool>(stream::iter_ok(stream_items));
let stream = future_of_a_stream.flatten_stream();
let mut iter = stream.wait();
assert_eq!(Ok(17), iter.next().unwrap());
assert_eq!(Ok(19), iter.next().unwrap());
assert_eq!(None, iter.next());
}
struct PanickingStream<T, E> {
_marker: marker::PhantomData<(T, E)>
}
impl<T, E> Stream for PanickingStream<T, E> {
type Item = T;
type Error = E;
fn poll(&mut self) -> Poll<Option<Self::Item>, Self::Error> {
panic!()
}
}
#[test]
fn failed_future() {
let future_of_a_stream = err::<PanickingStream<bool, u32>, _>(10);
let stream = future_of_a_stream.flatten_stream();
let mut iter = stream.wait();
assert_eq!(Err(10), iter.next().unwrap());
assert_eq!(None, iter.next());
}
|
