1 #![warn(rust_2018_idioms)]
2 #![cfg(feature = "full")]
3
4 use std::io;
5 use tokio::io::AsyncReadExt;
6 use tokio_test::assert_ok;
7 use tokio_test::io::Builder;
8
9 #[tokio::test]
read_to_string()10 async fn read_to_string() {
11 let mut buf = String::new();
12 let mut rd: &[u8] = b"hello world";
13
14 let n = assert_ok!(rd.read_to_string(&mut buf).await);
15 assert_eq!(n, 11);
16 assert_eq!(buf[..], "hello world"[..]);
17 }
18
19 #[tokio::test]
to_string_does_not_truncate_on_utf8_error()20 async fn to_string_does_not_truncate_on_utf8_error() {
21 let data = vec![0xff, 0xff, 0xff];
22
23 let mut s = "abc".to_string();
24
25 match AsyncReadExt::read_to_string(&mut data.as_slice(), &mut s).await {
26 Ok(len) => panic!("Should fail: {} bytes.", len),
27 Err(err) if err.to_string() == "stream did not contain valid UTF-8" => {}
28 Err(err) => panic!("Fail: {}.", err),
29 }
30
31 assert_eq!(s, "abc");
32 }
33
34 #[tokio::test]
to_string_does_not_truncate_on_io_error()35 async fn to_string_does_not_truncate_on_io_error() {
36 let mut mock = Builder::new()
37 .read(b"def")
38 .read_error(io::Error::new(io::ErrorKind::Other, "whoops"))
39 .build();
40 let mut s = "abc".to_string();
41
42 match AsyncReadExt::read_to_string(&mut mock, &mut s).await {
43 Ok(len) => panic!("Should fail: {} bytes.", len),
44 Err(err) if err.to_string() == "whoops" => {}
45 Err(err) => panic!("Fail: {}.", err),
46 }
47
48 assert_eq!(s, "abc");
49 }
50
51 #[tokio::test]
to_string_appends()52 async fn to_string_appends() {
53 let data = b"def".to_vec();
54
55 let mut s = "abc".to_string();
56
57 let len = AsyncReadExt::read_to_string(&mut data.as_slice(), &mut s)
58 .await
59 .unwrap();
60
61 assert_eq!(len, 3);
62 assert_eq!(s, "abcdef");
63 }
64