• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 use futures::executor::block_on;
2 use futures::future::{Future, FutureExt};
3 use futures::io::{AsyncReadExt, 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 
9 #[test]
read_to_string()10 fn read_to_string() {
11     let mut c = Cursor::new(&b""[..]);
12     let mut v = String::new();
13     assert_eq!(block_on(c.read_to_string(&mut v)).unwrap(), 0);
14     assert_eq!(v, "");
15 
16     let mut c = Cursor::new(&b"1"[..]);
17     let mut v = String::new();
18     assert_eq!(block_on(c.read_to_string(&mut v)).unwrap(), 1);
19     assert_eq!(v, "1");
20 
21     let mut c = Cursor::new(&b"\xff"[..]);
22     let mut v = String::new();
23     assert!(block_on(c.read_to_string(&mut v)).is_err());
24 }
25 
26 #[test]
interleave_pending()27 fn interleave_pending() {
28     fn run<F: Future + Unpin>(mut f: F) -> F::Output {
29         let mut cx = noop_context();
30         loop {
31             if let Poll::Ready(x) = f.poll_unpin(&mut cx) {
32                 return x;
33             }
34         }
35     }
36     let mut buf = stream::iter(vec![&b"12"[..], &b"33"[..], &b"3"[..]])
37         .map(Ok)
38         .into_async_read()
39         .interleave_pending();
40 
41     let mut v = String::new();
42     assert_eq!(run(buf.read_to_string(&mut v)).unwrap(), 5);
43     assert_eq!(v, "12333");
44 }
45