1 use futures::executor::block_on;
2 use futures::future::{Future, FutureExt};
3 use futures::io::{AsyncBufReadExt, Cursor};
4 use futures::stream::{self, StreamExt, TryStreamExt};
5 use futures::task::Poll;
6 use futures_test::io::AsyncReadTestExt;
7 use futures_test::task::noop_context;
8
run<F: Future + Unpin>(mut f: F) -> F::Output9 fn run<F: Future + Unpin>(mut f: F) -> F::Output {
10 let mut cx = noop_context();
11 loop {
12 if let Poll::Ready(x) = f.poll_unpin(&mut cx) {
13 return x;
14 }
15 }
16 }
17
18 macro_rules! block_on_next {
19 ($expr:expr) => {
20 block_on($expr.next()).unwrap().unwrap()
21 };
22 }
23
24 macro_rules! run_next {
25 ($expr:expr) => {
26 run($expr.next()).unwrap().unwrap()
27 };
28 }
29
30 #[test]
lines()31 fn lines() {
32 let buf = Cursor::new(&b"12\r"[..]);
33 let mut s = buf.lines();
34 assert_eq!(block_on_next!(s), "12\r".to_string());
35 assert!(block_on(s.next()).is_none());
36
37 let buf = Cursor::new(&b"12\r\n\n"[..]);
38 let mut s = buf.lines();
39 assert_eq!(block_on_next!(s), "12".to_string());
40 assert_eq!(block_on_next!(s), "".to_string());
41 assert!(block_on(s.next()).is_none());
42 }
43
44 #[test]
maybe_pending()45 fn maybe_pending() {
46 let buf =
47 stream::iter(vec![&b"12"[..], &b"\r"[..]]).map(Ok).into_async_read().interleave_pending();
48 let mut s = buf.lines();
49 assert_eq!(run_next!(s), "12\r".to_string());
50 assert!(run(s.next()).is_none());
51
52 let buf = stream::iter(vec![&b"12"[..], &b"\r\n"[..], &b"\n"[..]])
53 .map(Ok)
54 .into_async_read()
55 .interleave_pending();
56 let mut s = buf.lines();
57 assert_eq!(run_next!(s), "12".to_string());
58 assert_eq!(run_next!(s), "".to_string());
59 assert!(run(s.next()).is_none());
60 }
61